Home
last modified time | relevance | path

Searched refs:AttrValue (Results 1 – 25 of 283) sorted by relevance

12345678910>>...12

/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h34 class AttrValue; variable
39 std::string SummarizeAttrValue(const AttrValue& attr_value);
42 Status AttrValueHasType(const AttrValue& attr_value, StringPiece type);
51 bool ParseAttrValue(StringPiece type, StringPiece text, AttrValue* out);
54 void SetAttrValue(const std::string& value, AttrValue* out);
55 void SetAttrValue(const tstring& value, AttrValue* out);
56 void SetAttrValue(const char* value, AttrValue* out);
57 void SetAttrValue(StringPiece value, AttrValue* out);
58 void SetAttrValue(int64 value, AttrValue* out);
59 void SetAttrValue(int32 value, AttrValue* out);
[all …]
Dattr_value_util.cc133 uint64 AttrValueHash(const AttrValue& a, const TensorProtoHasher& tensor_hash) { in AttrValueHash()
139 std::map<string, AttrValue> map(func.attr().begin(), func.attr().end()); in AttrValueHash()
152 bool AreAttrValuesEqual(const AttrValue& a, const AttrValue& b, in AreAttrValuesEqual()
174 std::unordered_map<string, AttrValue> am(af.attr().begin(), in AreAttrValuesEqual()
229 string SummarizeAttrValue(const AttrValue& attr_value) { in SummarizeAttrValue()
231 case AttrValue::kS: in SummarizeAttrValue()
233 case AttrValue::kI: in SummarizeAttrValue()
235 case AttrValue::kF: in SummarizeAttrValue()
237 case AttrValue::kB: in SummarizeAttrValue()
239 case AttrValue::kType: in SummarizeAttrValue()
[all …]
Dattr_value_util_test.cc30 AttrValue V(T value) { in V()
31 AttrValue ret; in V()
36 AttrValue P(const string& p) { in P()
37 AttrValue ret; in P()
42 AttrValue F(const string& name, in F()
43 std::vector<std::pair<string, AttrValue>> pairs) { in F()
44 AttrValue ret; in F()
50 AttrValue Fs( in Fs()
51 std::vector<std::pair<string, std::vector<std::pair<string, AttrValue>>>> in Fs()
53 AttrValue ret; in Fs()
[all …]
Dkernel_def_util.cc28 bool InTypeList(DataType dt, const AttrValue& type_list) { in InTypeList()
40 auto constraint_value_case = AttrValue::VALUE_NOT_SET; in KernelAttrsMatch()
43 constraint_value_case = AttrValue::kType; in KernelAttrsMatch()
47 constraint_value_case = AttrValue::kS; in KernelAttrsMatch()
51 constraint_value_case = AttrValue::kI; in KernelAttrsMatch()
55 constraint_value_case = AttrValue::kB; in KernelAttrsMatch()
74 const AttrValue* attr_value = attrs.Find(constraint.name()); in KernelAttrsMatch()
84 if (constraint_value_case == AttrValue::oneof_case) { \ in KernelAttrsMatch()
114 if (constraint_value_case != AttrValue::kType) { in KernelAttrsMatch()
Dnode_def_util.cc96 const AttrValue& attr_value = it.second; in DebugString()
144 const AttrValue* AttrSlice::Find(StringPiece attr_name) const { in Find()
165 const AttrValue* AttrSlice::FindByString(const string& attr_name) const { in FindByString()
175 const AttrValue** attr_value) const { in Find()
211 const AttrValue* attr_value; \
221 const AttrValue* attr_value; \
235 const AttrValue* attr_value = attrs.Find(attr_name); \
250 const AttrValue* attr_value = attrs.Find(attr_name); \
333 const AttrValue* attr_value = attrs.Find(attr_name); in GetNodeAttrString()
346 const AttrValue* attr_value = attrs.Find(attr_name); in TryGetNodeAttr()
[all …]
Dnode_def_util.h45 class AttrValue; variable
79 typedef protobuf::Map<string, AttrValue> AttrValueMap;
83 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
84 void AddNodeAttr(StringPiece name, AttrValue&& value, NodeDef* node_def);
137 void AddAttr(StringPiece name, const AttrValue& value, AttrValueMap* map);
151 const AttrValue* Find(StringPiece attr_name) const;
152 const AttrValue* FindByString(const std::string& attr_name) const;
156 Status Find(StringPiece attr_name, const AttrValue** attr_value) const;
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc31 StatusOr<mlir::Attribute> ConvertNonFuncAttributeValue(const AttrValue& value, in ConvertNonFuncAttributeValue()
34 case AttrValue::kI: in ConvertNonFuncAttributeValue()
36 case AttrValue::kS: in ConvertNonFuncAttributeValue()
38 case AttrValue::kF: in ConvertNonFuncAttributeValue()
40 case AttrValue::kB: in ConvertNonFuncAttributeValue()
42 case AttrValue::kType: { in ConvertNonFuncAttributeValue()
47 case AttrValue::kShape: in ConvertNonFuncAttributeValue()
49 case AttrValue::kTensor: in ConvertNonFuncAttributeValue()
51 case AttrValue::kList: { in ConvertNonFuncAttributeValue()
82 case AttrValue::VALUE_NOT_SET: in ConvertNonFuncAttributeValue()
[all …]
Dexport_utils.cc92 Status ConvertAttribute(const mlir::BoolAttr& attr, AttrValue* value) { in ConvertAttribute()
97 Status ConvertAttribute(const mlir::IntegerAttr& attr, AttrValue* value) { in ConvertAttribute()
102 Status ConvertAttribute(const mlir::FloatAttr& attr, AttrValue* value) { in ConvertAttribute()
107 Status ConvertAttribute(const mlir::ElementsAttr& attr, AttrValue* value) { in ConvertAttribute()
111 Status ConvertAttribute(const mlir::TF::ShapeAttr& attr, AttrValue* value) { in ConvertAttribute()
124 Status ConvertAttribute(const mlir::FlatSymbolRefAttr& attr, AttrValue* value) { in ConvertAttribute()
130 AttrValue* value) { in ConvertAttribute()
139 Status ConvertAttribute(const mlir::StringAttr& attr, AttrValue* value) { in ConvertAttribute()
163 AttrValue* value) { in ConvertAttribute()
172 AttrValue* value) { in ConvertAttribute()
[all …]
Dconvert_attr.h30 StatusOr<mlir::Attribute> ConvertNonFuncAttributeValue(const AttrValue& value,
34 StatusOr<mlir::Attribute> ConvertAttributeValue(const AttrValue& value,
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion_test.cc40 std::vector<std::pair<string, AttrValue>> range_attrs; in TEST()
51 std::vector<std::pair<string, AttrValue>> map_attrs(2); in TEST()
52 AttrValue f_attr; in TEST()
55 AttrValue args_attr; in TEST()
68 std::vector<std::pair<string, AttrValue>> batch_attrs(2); in TEST()
69 AttrValue shapes_attr; in TEST()
72 AttrValue types_attr; in TEST()
122 std::vector<std::pair<string, AttrValue>> range_attrs; in TEST()
133 std::vector<std::pair<string, AttrValue>> map_attrs(2); in TEST()
134 AttrValue f_attr; in TEST()
[all …]
Dshuffle_and_repeat_fusion_test.cc36 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
37 AttrValue shapes_attr; in TEST()
40 AttrValue types_attr; in TEST()
104 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
105 AttrValue shapes_attr; in TEST()
108 AttrValue types_attr; in TEST()
169 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
170 AttrValue shapes_attr; in TEST()
173 AttrValue types_attr; in TEST()
241 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
[all …]
Dslack_test.cc34 std::vector<std::pair<string, AttrValue>> common_attrs(2); in SetupGrapplerItem()
35 AttrValue shapes_attr; in SetupGrapplerItem()
38 AttrValue types_attr; in SetupGrapplerItem()
125 AttrValue shapes_attr; in TEST()
127 AttrValue types_attr; in TEST()
141 AttrValue variant_type_attr; in TEST()
Dautotune_buffer_sizes_test.cc169 std::vector<std::pair<string, AttrValue>> range_attrs; in TEST_P()
177 std::vector<std::pair<string, AttrValue>> map_attrs(4); in TEST_P()
178 AttrValue attr_val; in TEST_P()
192 std::vector<std::pair<string, AttrValue>> prefetch_attrs(4); in TEST_P()
193 AttrValue legacy_autotune_attr; in TEST_P()
195 AttrValue buffer_size_min_attr; in TEST_P()
/external/tensorflow/tensorflow/lite/toco/
Dtensorflow_util.cc37 using tensorflow::AttrValue;
75 if (attr.second.value_case() == AttrValue::kFunc) { in LogDumpGraphDef()
77 } else if (attr.second.value_case() == AttrValue::kPlaceholder) { in LogDumpGraphDef()
80 } else if (attr.second.value_case() == AttrValue::kS) { in LogDumpGraphDef()
92 } else if (attr.second.value_case() == AttrValue::kI) { in LogDumpGraphDef()
94 } else if (attr.second.value_case() == AttrValue::kF) { in LogDumpGraphDef()
96 } else if (attr.second.value_case() == AttrValue::kB) { in LogDumpGraphDef()
99 } else if (attr.second.value_case() == AttrValue::kType) { in LogDumpGraphDef()
102 } else if (attr.second.value_case() == AttrValue::kShape) { in LogDumpGraphDef()
109 } else if (attr.second.value_case() == AttrValue::kTensor) { in LogDumpGraphDef()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/python/
Dtransform_graph_test.py40 const_op1.attr["dtype"].CopyFrom(attr_value_pb2.AttrValue(
43 attr_value_pb2.AttrValue(tensor=tensor_util.make_tensor_proto(
49 const_op2.attr["dtype"].CopyFrom(attr_value_pb2.AttrValue(
52 attr_value_pb2.AttrValue(tensor=tensor_util.make_tensor_proto(
58 add_op.attr["T"].CopyFrom(attr_value_pb2.AttrValue(
66 relu_op.attr["T"].CopyFrom(attr_value_pb2.AttrValue(
/external/tensorflow/tensorflow/security/fuzzing/
DAreAttrValuesEqual_fuzz.cc30 tensorflow::AttrValue createAttrValue(T value) { in createAttrValue()
31 tensorflow::AttrValue ret; in createAttrValue()
39 const tensorflow::AttrValue proto = createAttrValue(value); in compareValues()
40 const tensorflow::AttrValue proto_same = createAttrValue(value); in compareValues()
41 const tensorflow::AttrValue proto2 = createAttrValue(value_2); in compareValues()
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py124 return attr_value_pb2.AttrValue(b=compile_ops(node_def))
126 xla_compile = attr_value_pb2.AttrValue(b=compile_ops)
132 attr_value_pb2.AttrValue(b=bool(separate_compiled_gradients))
149 attrs["_XlaScope"] = attr_value_pb2.AttrValue(
/external/tensorflow/tensorflow/python/framework/
Dop_def_util.cc258 Safe_PyObjectPtr AttrValueListToPyObject(AttrValue::ListValue list) { in AttrValueListToPyObject()
339 Safe_PyObjectPtr AttrValueToPyObject(const AttrValue& attr_value) { in AttrValueToPyObject()
341 case tensorflow::AttrValue::kS: in AttrValueToPyObject()
343 case tensorflow::AttrValue::kI: in AttrValueToPyObject()
345 case tensorflow::AttrValue::kF: in AttrValueToPyObject()
347 case tensorflow::AttrValue::kB: in AttrValueToPyObject()
349 case tensorflow::AttrValue::kType: in AttrValueToPyObject()
351 case tensorflow::AttrValue::kShape: in AttrValueToPyObject()
353 case tensorflow::AttrValue::kList: in AttrValueToPyObject()
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc288 [](const std::pair<string, absl::optional<AttrValue>>& attr_kv_pair) { in DescribeTo()
337 std::map<string, absl::optional<AttrValue>> attrs;
475 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
477 AttrValue attr_value; in AttrLiteralHelper()
482 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
484 AttrValue attr_value; in AttrLiteralHelper()
485 AttrValue::ListValue* list = attr_value.mutable_list(); in AttrLiteralHelper()
492 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
494 AttrValue attr_value; in AttrLiteralHelper()
495 AttrValue::ListValue* list = attr_value.mutable_list(); in AttrLiteralHelper()
[all …]
Dnode_matchers.h87 using AttrKeyValuePair = std::pair<string, absl::optional<AttrValue>>;
165 impl::NodeMatcherProperties Attr(std::pair<string, AttrValue> attrs);
168 std::pair<string, AttrValue> AttrLiteralHelper(
171 std::pair<string, AttrValue> AttrLiteralHelper(
174 std::pair<string, AttrValue> AttrLiteralHelper(
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc81 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
100 absl::flat_hash_map<string, AttrValue> body_parameters; in TEST_F()
126 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
207 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
303 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
368 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
419 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
459 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
486 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
534 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/tpu/ops/
Dhost_compute_ops.cc63 const AttrValue* graph; in __anonc8f95cfc0202()
66 const AttrValue* shapes; in __anonc8f95cfc0202()
104 const AttrValue* shape_attr; in __anonc8f95cfc0402()
/external/tensorflow/tensorflow/c/
Dkernels.cc96 const tensorflow::AttrValue* GetAttrValue(TF_OpKernelConstruction* ctx, in GetAttrValue()
100 const tensorflow::AttrValue* attr = in GetAttrValue()
282 const tensorflow::AttrValue* attr = GetAttrValue(ctx, attr_name, status); in TF_OpKernelConstruction_GetAttrSize()
290 case tensorflow::AttrValue::kK: \ in TF_OpKernelConstruction_GetAttrSize()
305 case tensorflow::AttrValue::kList: in TF_OpKernelConstruction_GetAttrSize()
334 case tensorflow::AttrValue::kPlaceholder: in TF_OpKernelConstruction_GetAttrSize()
339 case tensorflow::AttrValue::kFunc: in TF_OpKernelConstruction_GetAttrSize()
344 case tensorflow::AttrValue::VALUE_NOT_SET: in TF_OpKernelConstruction_GetAttrSize()
368 const tensorflow::AttrValue* attr = GetAttrValue(ctx, attr_name, status); \
370 if (attr->value_case() != tensorflow::AttrValue::kList) { \
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dannotate_max_batch_sizes_test.py106 attr_value_pb2.AttrValue(i=self.max_batch_sizes[1])
112 attr_value_pb2.AttrValue(i=self.max_batch_sizes[2])
118 attr_value_pb2.AttrValue(i=self.max_batch_sizes[3])
/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py983 input_attr = attr_value_pb2.AttrValue()
987 output_attr = attr_value_pb2.AttrValue()
1113 input_attr = attr_value_pb2.AttrValue()
1117 output_attr = attr_value_pb2.AttrValue()
1190 tin_attr = attr_value_pb2.AttrValue(
1191 list=attr_value_pb2.AttrValue.ListValue(
1193 tout_attr = attr_value_pb2.AttrValue(
1194 list=attr_value_pb2.AttrValue.ListValue(type=tout))
1195 func_attr = attr_value_pb2.AttrValue(
1197 executor_type_attr = attr_value_pb2.AttrValue(
[all …]

12345678910>>...12