| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.h | 77 const AttrValueMap& attrs, in FunctionAttr()
84 const AttrValueMap& attrs) { in FunctionCall()
92 const AttrValueMap& attrs) { in SymbolicGradient()
104 const AttrValueMap& attrs() const { return attrs_; } in attrs()
108 const AttrValueMap& attrs, const string& attr_name) in AssociatedFunctionInfo()
117 AttrValueMap attrs_;
|
| D | tf2xla_util.cc | 597 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
601 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
|
| D | tf2xla_util_test.cc | 310 AttrValueMap attrs; in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder_test.cc | 86 string ToString(const AttrValueMap& m) { in ToString()
99 AttrValueMap m; in TEST()
111 AttrValueMap m; in TEST()
|
| D | attr_builder.h | 155 void FillAttrValueMap(AttrValueMap* m) const;
162 void FillAttrValueMapWithoutDefaults(AttrValueMap* m) const;
187 void SetInAttrValueMap(AttrValueMap* m, const string& attr_name, in SetInAttrValueMap()
|
| D | attr_builder.cc | 165 void AttrBuilder::FillAttrValueMap(AttrValueMap* m) const { in FillAttrValueMap()
168 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMap()
205 void AttrBuilder::FillAttrValueMapWithoutDefaults(AttrValueMap* m) const { in FillAttrValueMapWithoutDefaults()
214 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMapWithoutDefaults()
|
| D | eager_operation.cc | 227 AttrValueMap attrs; in InputLength()
254 AttrValueMap attrs; in OutputLength()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.h | 63 bool remove_ref_type, AttrValueMap* values);
68 AttrValueMap* values);
|
| D | export_utils.cc | 344 bool remove_ref_type, AttrValueMap* values) { in ConvertAttributes()
345 AttrValueMap func_call_attrs; in ConvertAttributes()
415 AttrValueMap* values) { in SetShapeAttribute()
|
| D | shape_inference_utils.cc | 210 tensorflow::AttrValueMap attributes; in InferReturnTypeComponentsForTFOp()
|
| /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,
|
| D | host_training_loop_optimization_util.cc | 222 const AttrValueMap* current_function_attr, in FindTPUCompileNodes()
390 const AttrValueMap* current_function_attr, in DetectHostTrainingLoop()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 79 typedef protobuf::Map<string, AttrValue> AttrValueMap; typedef
137 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map);
138 void AddAttr(StringPiece name, bool value, AttrValueMap* map);
145 explicit AttrSlice(const AttrValueMap* a);
182 AttrValueMap::const_iterator begin() const { return attrs_->begin(); } in begin()
183 AttrValueMap::const_iterator end() const { return attrs_->end(); } in end()
189 const AttrValueMap* attrs_;
|
| D | node_def_util.cc | 49 static const AttrValueMap* const kEmptyAttrValueMap = new AttrValueMap; in AttrSlice()
56 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice()
865 AttrValueMap::value_type(string(name), value)); in AddNodeAttr()
906 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map) { in ADD_NODE_ATTR()
907 map->insert(AttrValueMap::value_type(string(name), value)); in ADD_NODE_ATTR()
911 void AddAttr(StringPiece name, T value, AttrValueMap* map) { \
|
| D | function_testlib.h | 54 AttrValueMap map_;
|
| D | graph_def_util.cc | 186 AttrValueMap* attrs = node->mutable_attr(); in StripDefaultAttributes()
|
| D | function.cc | 692 AttrValueMap* attrs) { in AddDefaultAttrs()
761 std::vector<AttrValueMap> node_attrs; in InstantiateFunction()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 48 AttrValueMap* values) { in SetTypeAttribute()
71 AttrValueMap* values) { in SetShapeAttribute()
169 AttrValueMap* attributes) { in PopulateDerivedAttributes()
203 bool ignore_unregistered_attrs, AttrValueMap* attributes) { in GetAttrValuesFromOperation()
|
| D | export_tf_dialect_op.h | 36 bool ignore_unregistered_attrs, AttrValueMap* attributes);
|
| /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/kernels/data/ |
| D | hash_utils.cc | 465 Status HashFunction(const std::string& name, const AttrValueMap& attrs, in HashFunction()
504 const AttrValueMap& this_attrs, GraphHasher* that, in CheckFunctionsEqual()
506 const AttrValueMap& that_attrs) { in CheckFunctionsEqual()
517 const AttrValueMap& this_attrs, in CheckFunctionsEqualHelper()
520 const AttrValueMap& that_attrs) { in CheckFunctionsEqualHelper()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 1432 tensorflow::AttrValueMap ExtractAttrs(TFE_Op* op) { in ExtractAttrs()
1433 tensorflow::AttrValueMap attr_values; in ExtractAttrs()
1456 tensorflow::AttrValueMap attr_values = ExtractAttrs(minOp); in TEST()
1457 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1495 tensorflow::AttrValueMap attr_values = ExtractAttrs(concatOp); in TEST()
1496 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1535 tensorflow::AttrValueMap attr_values = ExtractAttrs(assertOp); in TEST()
1536 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1579 tensorflow::AttrValueMap attr_values = ExtractAttrs(concatOp); in TEST()
1689 tensorflow::AttrValueMap attr_values; in TEST()
[all …]
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers.cc | 201 const AttrValueMap attr_value_map = node->def().attr(); in MatchAndExplain()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | graph_view_internal.h | 398 absl::optional<AttrValueMap> processed_attrs;
|