Searched refs:derived_attrs (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 98 llvm::StringMap<Attribute>* derived_attrs) const;
106 llvm::StringMap<Attribute>* derived_attrs) const;
119 const llvm::StringMap<Attribute>& derived_attrs) const;
188 llvm::StringMap<Attribute>* derived_attrs) const { in AddDerivedAttrs()
196 derived_attrs->insert({attr_names[0].getValue(), input_types[0]}); in AddDerivedAttrs()
209 derived_attrs->insert({attr_names[0].getValue(), in AddDerivedAttrs()
213 derived_attrs->insert({attr_names[1].getValue(), input_types[0]}); in AddDerivedAttrs()
219 derived_attrs->insert( in AddDerivedAttrs()
231 llvm::StringMap<Attribute>* derived_attrs) const { in CollectInputsAndAttributes()
262 derived_attrs))) { in CollectInputsAndAttributes()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 120 mlir::Operation* inst, mlir::DictionaryAttr derived_attrs, in GetAttributesToIgnore() argument
128 if (derived_attrs) { in GetAttributesToIgnore()
129 for (auto derived_attr : derived_attrs) { in GetAttributesToIgnore()
166 mlir::DictionaryAttr derived_attrs, in PopulateDerivedAttributes() argument
169 if (derived_attrs) { in PopulateDerivedAttributes()
171 ConvertAttributes(derived_attrs.getValue(), /*attrs_to_ignore=*/{}, in PopulateDerivedAttributes()
202 mlir::DictionaryAttr derived_attrs = nullptr; in GetAttrValuesFromOperation() local
204 derived_attrs = interface.materializeDerivedAttributes(); in GetAttrValuesFromOperation()
206 GetAttributesToIgnore(inst, derived_attrs, op_reg_data, in GetAttrValuesFromOperation()
213 inst, name, derived_attrs, ignore_unregistered_attrs, attributes)); in GetAttrValuesFromOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | composite.py | 40 derived_attrs=None, argument
45 self._derived_attrs = derived_attrs
|
| D | op_reg_gen_test.py | 33 'TestNoOp', derived_attrs=['T: numbertype'], outputs=['o1: T'])
42 derived_attrs=['T: numbertype'],
|
| D | tfr_gen.py | 203 derived_attrs = set()
206 derived_attrs.add(arg.type_attr)
208 derived_attrs.add(arg.number_attr)
210 derived_attrs.add(arg.type_list_attr)
216 derived_attrs.add('f32_')
218 derived_attrs.add('i32_')
220 derived_attrs.add('i64_')
222 derived_attrs.add('i1_')
223 return derived_attrs
280 derived_attrs = _collect_derived_attrs_from_proto(op_def)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/ |
| D | derived_attrs.mlir | 3 // CHECK-LABEL: func @derived_attrs
4 func.func @derived_attrs(
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 51 derived_attrs=['T: {float, int8}'],
120 derived_attrs=['T: {float, int8}'],
170 derived_attrs=['T: {float, int8}'],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 46 derived_attrs=['T: type', 'Tpaddings: {int32, int64} = DT_INT32'],
96 derived_attrs=['T: type', 'Tpaddings: {int32, int64} = DT_INT32'],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ |
| D | define_op_template.py | 39 @Composite('TestRandom', derived_attrs=['T: numbertype'], outputs=['o: T'])
|
| D | README.md | 80 derived_attrs=['T: {float, int8}'],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | corert_converter.cc | 59 auto derived_attrs = interface.materializeDerivedAttributes(); in MaterializeDerivedAttributes() local
60 for (auto named_attr : derived_attrs) { in MaterializeDerivedAttributes()
|