Searched refs:getDictionaryAttr (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util_test.cc | 149 builder.getDictionaryAttr(builder.getNamedAttr( in TEST()
150 "bad_device", builder.getDictionaryAttr({})))); in TEST()
162 auto device_dict = builder.getDictionaryAttr( in TEST()
164 builder.getDictionaryAttr({}))}); in TEST()
192 (*module_ref)->setAttr("tf.devices", builder.getDictionaryAttr(metadata)); in TEST()
|
| D | device_util.cc | 59 return builder->getDictionaryAttr({}); in ParseGpuDeviceMetadata()
137 auto metadata = builder.getDictionaryAttr({}); in AddDevicesToOp()
142 op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); in AddDevicesToOp()
|
| D | translate_utils.cc | 34 b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>( in PopulateTfVersions()
|
| D | convert_attr.cc | 104 auto func_attrs = builder->getDictionaryAttr(attrs); in ConvertAttributeValue()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | FunctionImplementation.cpp | 151 builder.getDictionaryAttr(argAttrs[i])); in addArgAndResultAttrs()
157 builder.getDictionaryAttr(resultAttrs[i])); in addArgAndResultAttrs()
|
| D | Builders.cpp | 94 DictionaryAttr Builder::getDictionaryAttr(ArrayRef<NamedAttribute> value) { in getDictionaryAttr() function in Builder
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_dynamic_padding_mapper.cc | 177 builder.getDictionaryAttr({shape_indices, padding_arg_indices})); in AnnotateFunctionArgumentsWithPaddings()
|
| D | tpu_rewrite_pass.cc | 445 replicate->setAttr(kDevicesAttr, builder->getDictionaryAttr(device_attrs)); in AssignDevicesToReplicate()
|
| D | resource_op_lifting.cc | 448 DictionaryAttr empty_attrs = builder.getDictionaryAttr({}); in GenerateHoistedReads()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | mlir_graph_optimization_pass_test.cc | 65 b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>( in Run()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 91 DictionaryAttr getDictionaryAttr(ArrayRef<NamedAttribute> value);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device.cc | 573 devices_attr.emplace(builder.getDictionaryAttr(device_list)); in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1191 auto func_attrs = builder_.getDictionaryAttr(attrs); in ConvertAttributeValue()
2260 b.getDictionaryAttr({inputs, outputs, control_outputs}))); in Convert()
2293 b.getDictionaryAttr({inputs, outputs, control_outputs}))); in Convert()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 1045 func->setAttr("tf.entry_function", builder.getDictionaryAttr(attributes)); in ConvertSubgraph()
|