| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 78 const AttrValueMap& attrs, in FunctionAttr()
85 const AttrValueMap& attrs) { in FunctionCall()
93 const AttrValueMap& attrs) { in SymbolicGradient()
105 const AttrValueMap& attrs() const { return attrs_; } in attrs()
109 const AttrValueMap& attrs, const string& attr_name) in AssociatedFunctionInfo()
118 AttrValueMap attrs_;
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 84 typedef protobuf::Map<string, AttrValue> AttrValueMap; typedef
142 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map);
143 void AddAttr(StringPiece name, bool value, AttrValueMap* map);
150 explicit AttrSlice(const AttrValueMap* a);
189 AttrValueMap::const_iterator begin() const { return attrs()->begin(); } in begin()
190 AttrValueMap::const_iterator end() const { return attrs()->end(); } in end()
195 const AttrValueMap* attrs() const { in attrs()
202 const AttrValueMap* attrs_;
|
| D | node_def_util.cc | 55 static const AttrValueMap* const kEmptyAttrValueMap = new AttrValueMap; in AttrSlice()
64 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice()
924 AttrValueMap::value_type(string(name), value)); in AddNodeAttr()
965 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map) { in ADD_NODE_ATTR()
966 map->insert(AttrValueMap::value_type(string(name), value)); in ADD_NODE_ATTR()
970 void AddAttr(StringPiece name, T value, AttrValueMap* map) { \
|
| D | function_testlib.h | 54 AttrValueMap map_;
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | host_training_loop_optimization_util.h | 43 absl::optional<AttrValueMap> encapsulating_function_attrs;
67 const AttrValueMap* current_function_attr,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 48 AttrValueMap* values) { in SetTypeAttribute()
71 AttrValueMap* values) { in SetShapeAttribute()
168 AttrValueMap* attributes) { in PopulateDerivedAttributes()
201 bool ignore_unregistered_attrs, AttrValueMap* attributes) { in GetAttrValuesFromOperation()
|
| D | export_tf_dialect_op.h | 36 bool ignore_unregistered_attrs, AttrValueMap* attributes);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.h | 63 bool remove_ref_type, AttrValueMap* values);
84 AttrValueMap* values);
|
| D | export_utils.cc | 372 bool remove_ref_type, AttrValueMap* values) { in ConvertAttributes()
373 AttrValueMap func_call_attrs; in ConvertAttributes()
444 AttrValueMap* values) { in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.h | 155 void FillAttrValueMap(AttrValueMap* m) const;
162 void FillAttrValueMapWithoutDefaults(AttrValueMap* m) const;
190 void SetInAttrValueMap(AttrValueMap* m, const string& attr_name, in SetInAttrValueMap()
|
| D | attr_builder_test.cc | 88 string ToString(const AttrValueMap& m) { in ToString()
101 AttrValueMap m; in TEST()
113 AttrValueMap m; in TEST()
|
| D | attr_builder.cc | 182 void AttrBuilder::FillAttrValueMap(AttrValueMap* m) const { in FillAttrValueMap()
185 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMap()
222 void AttrBuilder::FillAttrValueMapWithoutDefaults(AttrValueMap* m) const { in FillAttrValueMapWithoutDefaults()
231 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMapWithoutDefaults()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | convert_attributes.h | 39 tensorflow::AttrValueMap* values);
46 tensorflow::AttrValueMap* values);
|
| D | convert_attributes.cc | 40 using tensorflow::AttrValueMap;
233 bool remove_ref_type, AttrValueMap* values) { in ConvertAttributes()
236 AttrValueMap func_call_attrs; in ConvertAttributes()
281 AttrValueMap* values) { in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/util/ |
| D | attr_util.h | 90 AttrValueMap* attr_value_map);
95 tensorflow::AttrValueMap* attr_value_map);
|
| /external/tensorflow/tensorflow/core/tfrt/fallback/ |
| D | op_kernel_runner.cc | 43 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder) { in BuildNodeDef()
83 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder, in Create()
104 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder, in Create()
|
| D | op_kernel_runner.h | 44 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder,
51 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder,
|
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | op_kernel_runner.cc | 43 const std::function<llvm::Error(tensorflow::AttrValueMap*)>& attr_builder) { in BuildNodeDef()
85 const std::function<llvm::Error(tensorflow::AttrValueMap*)>& attr_builder, in Create()
172 const std::function<llvm::Error(tensorflow::AttrValueMap*)>& attr_builder, in GetOrCreate()
|
| D | op_kernel_runner.h | 70 const std::function<llvm::Error(tensorflow::AttrValueMap*)>& attr_builder,
177 const std::function<llvm::Error(tensorflow::AttrValueMap*)>& attr_builder,
|
| D | kernel_fallback_op_handler.cc | 178 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder,
192 const std::function<Status(tensorflow::AttrValueMap*)>& attr_builder, in GetOrCreate()
290 tensorflow::AttrValueMap* attr_value_map) { in MakeOp()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | saved_model_utils.cc | 54 gtl::FlatMap<StringPiece, const AttrValueMap*, StringPieceHasher>;
73 const AttrValueMap* attrs = node_name_and_attrs->second; in ConstantFromSavedConstant()
364 gtl::FlatMap<StringPiece, const AttrValueMap*, StringPieceHasher> NodeToAttrMap( in NodeToAttrMap()
366 gtl::FlatMap<StringPiece, const AttrValueMap*, StringPieceHasher> result; in NodeToAttrMap()
|
| D | saved_model_utils.h | 89 gtl::FlatMap<StringPiece, const AttrValueMap*, StringPieceHasher> NodeToAttrMap(
|
| /external/tensorflow/tensorflow/core/ir/utils/ |
| D | shape_inference_utils_test.cc | 221 bool, tensorflow::AttrValueMap *) { return ::tensorflow::OkStatus(); }; in TEST_F() argument
241 tensorflow::AttrValueMap *) { in TEST_F() argument
|
| D | shape_inference_utils.h | 56 tensorflow::AttrValueMap*)>;
|
| /external/tensorflow/tensorflow/core/data/ |
| D | hash_utils.cc | 524 Status HashFunction(const std::string& name, const AttrValueMap& attrs, in HashFunction()
575 const AttrValueMap& this_attrs, GraphHasher* that, in CheckFunctionsEqual()
577 const AttrValueMap& that_attrs) { in CheckFunctionsEqual()
588 const AttrValueMap& this_attrs, in CheckFunctionsEqualHelper()
591 const AttrValueMap& that_attrs) { in CheckFunctionsEqualHelper()
|