| /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 | 591 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
595 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
|
| D | tf2xla_util_test.cc | 307 AttrValueMap attrs; in TEST()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.h | 60 AttrValueMap* values);
65 AttrValueMap* values);
70 AttrValueMap* values);
75 AttrValueMap* values);
|
| D | export_utils.cc | 325 AttrValueMap* values) { in ConvertAttributes()
326 AttrValueMap func_call_attrs; in ConvertAttributes()
414 AttrValueMap* values) { in SetTypeAttribute()
434 AttrValueMap* values) { in SetShapeAttribute()
462 AttrValueMap* values) { in SetSizeAttribute()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.h | 135 void FillAttrValueMap(AttrValueMap* m) const;
142 void FillAttrValueMapWithoutDefaults(AttrValueMap* m) const;
153 void SetInAttrValueMap(AttrValueMap* m, const string& attr_name, in SetInAttrValueMap()
160 m->insert(AttrValueMap::value_type(attr_name, attr_tmp_)); in SetInAttrValueMap()
|
| 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.cc | 152 void AttrBuilder::FillAttrValueMap(AttrValueMap* m) const { in FillAttrValueMap()
155 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMap()
192 void AttrBuilder::FillAttrValueMapWithoutDefaults(AttrValueMap* m) const { in FillAttrValueMapWithoutDefaults()
201 m->insert(AttrValueMap::value_type(entry.first, attr_tmp_)); in FillAttrValueMapWithoutDefaults()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 72 typedef protobuf::Map<string, AttrValue> AttrValueMap; typedef
130 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map);
131 void AddAttr(StringPiece name, bool value, AttrValueMap* map);
138 explicit AttrSlice(const AttrValueMap* a);
174 AttrValueMap::const_iterator begin() const { return attrs_->begin(); } in begin()
175 AttrValueMap::const_iterator end() const { return attrs_->end(); } in end()
181 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()
832 AttrValueMap::value_type(string(name), value)); in AddNodeAttr()
873 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map) { in ADD_NODE_ATTR()
874 map->insert(AttrValueMap::value_type(string(name), value)); in ADD_NODE_ATTR()
878 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()
|
| D | function_test.cc | 50 AttrValueMap map_;
|
| D | op_kernel_test.cc | 550 AttrValueMap::value_type(parts[0], attr_value)); in CreateNodeDef()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | export_tf_dialect_op.cc | 47 AttrValueMap* values) { in SetTypeAttribute()
70 AttrValueMap* values) { in SetShapeAttribute()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 1247 tensorflow::AttrValueMap attr_values; in TEST()
1249 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1287 tensorflow::AttrValueMap attr_values; in TEST()
1289 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1328 tensorflow::AttrValueMap attr_values; in TEST()
1330 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1373 tensorflow::AttrValueMap attr_values; in TEST()
|
| D | c_api.cc | 1446 tensorflow::AttrValueMap attrs; in TFE_OpGetInputLength()
1470 tensorflow::AttrValueMap attrs; in TFE_OpGetOutputLength()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | function_ops.cc | 326 AttrValueMap attr_values = func_.attr(); in ComputeAsync()
|
| /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 | 394 absl::optional<AttrValueMap> processed_attrs;
|
| D | graph_view.cc | 812 AttrValueMap(node->attr().begin(), node->attr().end()); in CheckKernelRegisteredForNodes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 440 AttrValueMap FixAttrs(const AttrSlice& attrs);
|