Searched refs:derived_attrs (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 95 llvm::StringMap<Attribute>* derived_attrs) const;
103 llvm::StringMap<Attribute>* derived_attrs) const;
116 const llvm::StringMap<Attribute>& derived_attrs) const;
185 llvm::StringMap<Attribute>* derived_attrs) const { in AddDerivedAttrs()
193 derived_attrs->insert({attr_names[0].getValue(), input_types[0]}); in AddDerivedAttrs()
206 derived_attrs->insert({attr_names[0].getValue(), in AddDerivedAttrs()
210 derived_attrs->insert({attr_names[1].getValue(), input_types[0]}); in AddDerivedAttrs()
216 derived_attrs->insert( in AddDerivedAttrs()
228 llvm::StringMap<Attribute>* derived_attrs) const { in CollectInputsAndAttributes()
257 derived_attrs))) { in CollectInputsAndAttributes()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 121 mlir::Operation* inst, mlir::DictionaryAttr derived_attrs, in GetAttributesToIgnore() argument
129 if (derived_attrs) { in GetAttributesToIgnore()
130 for (auto derived_attr : derived_attrs) { in GetAttributesToIgnore()
167 mlir::DictionaryAttr derived_attrs, in PopulateDerivedAttributes() argument
170 if (derived_attrs) { in PopulateDerivedAttributes()
172 ConvertAttributes(derived_attrs.getValue(), /*attrs_to_ignore=*/{}, in PopulateDerivedAttributes()
204 mlir::DictionaryAttr derived_attrs = nullptr; in GetAttrValuesFromOperation() local
206 derived_attrs = interface.materializeDerivedAttributes(); in GetAttrValuesFromOperation()
208 GetAttributesToIgnore(inst, derived_attrs, op_reg_data, in GetAttrValuesFromOperation()
215 inst, name, derived_attrs, ignore_unregistered_attrs, attributes)); in GetAttrValuesFromOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/python/ |
| D | composite.py | 44 derived_attrs=None, argument
49 self._derived_attrs = derived_attrs
|
| D | op_reg_gen_test.py | 37 'TestNoOp', derived_attrs=['T: numbertype'], outputs=['o1: T'])
46 derived_attrs=['T: numbertype'],
|
| D | tfr_gen.py | 195 derived_attrs = set()
198 derived_attrs.add(arg.type_attr)
200 derived_attrs.add(arg.number_attr)
202 derived_attrs.add(arg.type_list_attr)
208 derived_attrs.add('f32_')
210 derived_attrs.add('i32_')
212 derived_attrs.add('i64_')
214 derived_attrs.add('i1_')
215 return derived_attrs
272 derived_attrs = _collect_derived_attrs_from_proto(op_def)
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| D | ops_defs.py | 55 derived_attrs=['T: {float, int8}'],
124 derived_attrs=['T: {float, int8}'],
174 derived_attrs=['T: {float, int8}'],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/examples/pad/ |
| D | ops_defs.py | 50 derived_attrs=['T: type', 'Tpaddings: {int32, int64} = DT_INT32'],
100 derived_attrs=['T: type', 'Tpaddings: {int32, int64} = DT_INT32'],
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/ |
| D | define_op_template.py | 43 @Composite('TestRandom', derived_attrs=['T: numbertype'], outputs=['o: T'])
|
| D | README.md | 78 derived_attrs=['T: {float, int8}'],
|