Home
last modified time | relevance | path

Searched refs:AttrValueMap (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.h76 const AttrValueMap& attrs, in FunctionAttr()
83 const AttrValueMap& attrs) { in FunctionCall()
91 const AttrValueMap& attrs) { in SymbolicGradient()
103 const AttrValueMap& attrs() const { return attrs_; } in attrs()
107 const AttrValueMap& attrs, const string& attr_name) in AssociatedFunctionInfo()
116 AttrValueMap attrs_;
Dtf2xla_util.cc598 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
602 AttrValueMap attrs(node.attrs().begin(), node.attrs().end()); in GetAssociatedFunctions()
Dtf2xla_util_test.cc305 AttrValueMap attrs; in TEST()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.h106 void FillAttrValueMap(AttrValueMap* m) const { FillAttrValueMap(m, true); } in FillAttrValueMap()
122 void FillAttrValueMap(AttrValueMap* m, bool include_those_in_node_def) const;
125 void SetInAttrValueMap(AttrValueMap* m, const string& attr_name, in SetInAttrValueMap()
133 m->insert(AttrValueMap::value_type(attr_name, attr_value)); in SetInAttrValueMap()
Dattr_builder.cc145 void AttrBuilder::FillAttrValueMap(AttrValueMap* m, in FillAttrValueMap()
160 for (AttrValueMap::const_iterator it = node_def_->attr().begin(); in FillAttrValueMap()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h66 typedef protobuf::Map<string, AttrValue> AttrValueMap; typedef
123 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map);
124 void AddAttr(StringPiece name, bool value, AttrValueMap* map);
131 explicit AttrSlice(const AttrValueMap* a);
167 AttrValueMap::const_iterator begin() const { return attrs_->begin(); } in begin()
168 AttrValueMap::const_iterator end() const { return attrs_->end(); } in end()
172 const AttrValueMap* attrs_;
Dnode_def_util.cc44 static const AttrValueMap* const kEmptyAttrValueMap = new AttrValueMap; in AttrSlice()
51 AttrSlice::AttrSlice(const AttrValueMap* a) : ndef_(nullptr), attrs_(a) {} in AttrSlice()
724 AttrValueMap::value_type(string(name), value)); in AddNodeAttr()
761 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map) { in ADD_NODE_ATTR()
762 map->insert(AttrValueMap::value_type(string(name), value)); in ADD_NODE_ATTR()
766 void AddAttr(StringPiece name, T value, AttrValueMap* map) { \
Dfunction_testlib.h46 AttrValueMap map_;
Dfunction.cc665 AttrValueMap* attrs) { in AddDefaultAttrs()
714 std::vector<AttrValueMap> node_attrs; in InstantiateFunction()
Dop_kernel_test.cc418 AttrValueMap::value_type(parts[0], attr_value)); in CreateNodeDef()
Dfunction_test.cc47 AttrValueMap map_;
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc1647 tensorflow::AttrValueMap attr_values; in TEST()
1649 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1687 tensorflow::AttrValueMap attr_values; in TEST()
1689 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1728 tensorflow::AttrValueMap attr_values; in TEST()
1730 tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T"); in TEST()
1772 tensorflow::AttrValueMap attr_values; in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.cc318 AttrValueMap attr_values = func_.attr(); in ComputeAsync()
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc197 const AttrValueMap attr_value_map = node->def().attr(); in MatchAndExplain()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc412 AttrValueMap FixAttrs(const AttrSlice& attrs);
/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc9069 tensorflow::AttrValueMap attrs; in TFE_AddEagerOpToGraph()