Home
last modified time | relevance | path

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

123456789

/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h34 class AttrValue; variable
39 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 string& value, AttrValue* out);
55 void SetAttrValue(const char* value, AttrValue* out);
56 void SetAttrValue(StringPiece value, AttrValue* out);
57 void SetAttrValue(int64 value, AttrValue* out);
58 void SetAttrValue(int32 value, AttrValue* out);
59 void SetAttrValue(float value, AttrValue* out);
[all …]
Dattr_value_util.cc117 uint64 AttrValueHash(const AttrValue& a, const TensorProtoHasher& tensor_hash) { in AttrValueHash()
123 std::map<string, AttrValue> map(func.attr().begin(), func.attr().end()); in AttrValueHash()
135 bool AreAttrValuesEqual(const AttrValue& a, const AttrValue& b, in AreAttrValuesEqual()
151 std::unordered_map<string, AttrValue> am(af.attr().begin(), in AreAttrValuesEqual()
206 string SummarizeAttrValue(const AttrValue& attr_value) { in SummarizeAttrValue()
208 case AttrValue::kS: in SummarizeAttrValue()
210 case AttrValue::kI: in SummarizeAttrValue()
212 case AttrValue::kF: in SummarizeAttrValue()
214 case AttrValue::kB: in SummarizeAttrValue()
216 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 …]
Dnode_def_util.cc155 const AttrValue* AttrSlice::Find(StringPiece attr_name) const { in Find()
177 const AttrValue** attr_value) const { in Find()
213 const AttrValue* attr_value; \
223 const AttrValue* attr_value; \
236 const AttrValue* attr_value = attrs.Find(attr_name); \
251 const AttrValue* attr_value = attrs.Find(attr_name); \
305 const AttrValue* attr_value = attrs.Find(attr_name); in GetNodeAttrString()
318 const AttrValue* attr_value; in GetNodeAttr()
329 const AttrValue* attr_value; in GetNodeAttr()
338 const AttrValue* attr_value; in GetNodeAttr()
[all …]
/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()
123 std::vector<std::pair<string, AttrValue>> range_attrs; in TEST()
134 std::vector<std::pair<string, AttrValue>> map_attrs(2); in TEST()
135 AttrValue f_attr; in TEST()
[all …]
Dshuffle_and_repeat_fusion_test.cc32 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
33 AttrValue shapes_attr; in TEST()
36 AttrValue types_attr; in TEST()
100 std::vector<std::pair<string, AttrValue>> common_attrs(2); in TEST()
101 AttrValue shapes_attr; in TEST()
104 AttrValue types_attr; in TEST()
/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 …]
Dimport_tensorflow_test.cc31 using tensorflow::AttrValue;
85 AttrValue::ListValue* shapes = in BuildNode()
106 AttrValue dtype_attr; in BuildConstNode()
156 AttrValue value_attr; in BuildConstNode()
405 AttrValue dtype_attr; in NodeWithTensorContent()
454 AttrValue value_attr; in NodeWithTensorContent()
521 AttrValue dtype_attr; in BuildUnaryNode()
620 AttrValue value_attr; in TEST()
628 AttrValue value_attr; in TEST()
639 AttrValue value_attr; in TEST()
/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/contrib/compiler/
Djit.py82 return attr_value_pb2.AttrValue(b=compile_ops(node_def))
84 xla_compile = attr_value_pb2.AttrValue(b=compile_ops)
90 attr_value_pb2.AttrValue(b=bool(separate_compiled_gradients))
107 attrs["_XlaScope"] = attr_value_pb2.AttrValue(
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl_test.cc99 const std::unordered_map<string, AttrValue>& attrs, const string& device, in AddOperationToEnqueueRequest()
170 std::unordered_map<string, AttrValue> const_attrs; in TEST_F()
171 AttrValue val; in TEST_F()
182 std::unordered_map<string, AttrValue> attrs; in TEST_F()
254 std::unordered_map<string, AttrValue> const_attrs; in TEST_F()
255 AttrValue val; in TEST_F()
267 2, "MatMulFunction", {{1, 0}}, std::unordered_map<string, AttrValue>(), in TEST_F()
322 std::unordered_map<string, AttrValue> attrs; in TEST_F()
323 AttrValue val; in TEST_F()
/external/tensorflow/tensorflow/compiler/jit/
Dnode_matchers.cc284 [](const std::pair<string, absl::optional<AttrValue>>& attr_kv_pair) { in DescribeTo()
333 std::map<string, absl::optional<AttrValue>> attrs;
471 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
473 AttrValue attr_value; in AttrLiteralHelper()
478 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
480 AttrValue attr_value; in AttrLiteralHelper()
481 AttrValue::ListValue* list = attr_value.mutable_list(); in AttrLiteralHelper()
488 std::pair<string, AttrValue> impl::AttrLiteralHelper( in AttrLiteralHelper()
490 AttrValue attr_value; in AttrLiteralHelper()
491 AttrValue::ListValue* list = attr_value.mutable_list(); in AttrLiteralHelper()
[all …]
Dextract_outside_compilation_pass_test.cc290 protobuf::Map<string, tensorflow::AttrValue> attrs; in TEST_F()
347 AttrValue device_ordinal_temp_value; in TEST_F()
349 protobuf::Map<string, AttrValue> host_func_attrs; in TEST_F()
418 protobuf::Map<string, tensorflow::AttrValue> attrs; in TEST_F()
432 AttrValue device_ordinal_temp_value; in TEST_F()
434 protobuf::Map<string, AttrValue> host_func_attrs; in TEST_F()
466 protobuf::Map<string, tensorflow::AttrValue> attrs; in TEST_F()
493 AttrValue device_ordinal_temp_value; in TEST_F()
495 protobuf::Map<string, AttrValue> host_func_attrs; in TEST_F()
589 protobuf::Map<string, tensorflow::AttrValue> attrs; in TEST_F()
[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/python/ops/
Dfunctional_ops.py638 input_attr = attr_value_pb2.AttrValue()
642 output_attr = attr_value_pb2.AttrValue()
768 input_attr = attr_value_pb2.AttrValue()
772 output_attr = attr_value_pb2.AttrValue()
833 tin_attr = attr_value_pb2.AttrValue(
834 list=attr_value_pb2.AttrValue.ListValue(
836 tout_attr = attr_value_pb2.AttrValue(
837 list=attr_value_pb2.AttrValue.ListValue(type=tout))
838 func_attr = attr_value_pb2.AttrValue(
840 executor_type_attr = attr_value_pb2.AttrValue(
[all …]
/external/swiftshader/third_party/llvm-7.0/llvm/lib/DebugInfo/DWARF/
DDWARFDie.cpp582 : Die(D), AttrValue(0), Index(0) { in attribute_iterator()
590 AttrValue.Offset = D.getOffset() + AbbrDecl->getCodeByteSize(); in attribute_iterator()
601 AttrValue.Attr = AbbrDecl.getAttrByIndex(Index); in updateForIndex()
603 AttrValue.Offset += AttrValue.ByteSize; in updateForIndex()
604 AttrValue.Value.setForm(AbbrDecl.getFormByIndex(Index)); in updateForIndex()
605 uint32_t ParseOffset = AttrValue.Offset; in updateForIndex()
608 bool b = AttrValue.Value.extractValue(U->getDebugInfoExtractor(), in updateForIndex()
612 AttrValue.ByteSize = ParseOffset - AttrValue.Offset; in updateForIndex()
615 AttrValue.clear(); in updateForIndex()
DDWARFVerifier.cpp181 for (auto AttrValue : Die.attributes()) { in verifyUnitContents() local
182 NumUnitErrors += verifyDebugInfoAttribute(Die, AttrValue); in verifyUnitContents()
183 NumUnitErrors += verifyDebugInfoForm(Die, AttrValue); in verifyUnitContents()
386 DWARFAttribute &AttrValue) { in verifyDebugInfoAttribute() argument
396 const auto Attr = AttrValue.Attr; in verifyDebugInfoAttribute()
400 if (auto SectionOffset = AttrValue.Value.getAsSectionOffset()) { in verifyDebugInfoAttribute()
409 if (auto SectionOffset = AttrValue.Value.getAsSectionOffset()) { in verifyDebugInfoAttribute()
429 if (Optional<ArrayRef<uint8_t>> Expr = AttrValue.Value.getAsBlock()) { in verifyDebugInfoAttribute()
432 } else if (auto LocOffset = AttrValue.Value.getAsSectionOffset()) { in verifyDebugInfoAttribute()
449 DWARFAttribute &AttrValue) { in verifyDebugInfoForm() argument
[all …]
/external/tensorflow/tensorflow/contrib/util/
Dconvert_graphdef_memmapped_format_lib.cc44 const AttrValue& value = node_def->attr().at("value"); in ConvertConstantsToImmutable()
91 AttrValue attr_value; in ConvertConstantsToImmutable()
96 AttrValue attr_value; in ConvertConstantsToImmutable()
101 AttrValue attr_value; in ConvertConstantsToImmutable()
/external/tensorflow/tensorflow/python/eager/
Dgraph_only_ops.py34 dtype_value = attr_value_pb2.AttrValue(type=dtype.as_datatype_enum)
44 dtype_value = attr_value_pb2.AttrValue(type=dtype.as_datatype_enum)
47 shape = attr_value_pb2.AttrValue(shape=shape.as_proto())
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc76 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
91 absl::flat_hash_map<string, AttrValue> body_parameters; in TEST_F()
242 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
324 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
416 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
479 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
528 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
592 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
632 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
659 protobuf::Map<string, AttrValue> func_instantiation_attr; in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dgroup_by_reducer_dataset_op.cc143 AttrValue key_func; in AsGraphDefInternal()
145 AttrValue init_func; in AsGraphDefInternal()
147 AttrValue reduce_func; in AsGraphDefInternal()
149 AttrValue finalize_func; in AsGraphDefInternal()
152 AttrValue key_func_other_arguments_types_attr; in AsGraphDefInternal()
155 AttrValue init_func_other_arguments_types_attr; in AsGraphDefInternal()
158 AttrValue reduce_func_other_arguments_types_attr; in AsGraphDefInternal()
161 AttrValue finalize_func_other_arguments_types_attr; in AsGraphDefInternal()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc60 AttrValue cond_func; in TEST()
62 AttrValue body_func; in TEST()
163 AttrValue cond_func; in TEST()
165 AttrValue body_func; in TEST()
265 AttrValue cond_func; in TEST()
267 AttrValue body_func; in TEST()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-attr-value.pbtxt1 path: "tensorflow.AttrValue"
4 name: "AttrValue"
62 type_name: ".tensorflow.AttrValue.ListValue"
/external/tensorflow/tensorflow/core/kernels/data/
Dparallel_map_dataset_op.cc169 AttrValue f_attr; in AsGraphDefInternal()
173 AttrValue other_arguments_types_attr; in AsGraphDefInternal()
177 AttrValue use_inter_op_parallelism_attr; in AsGraphDefInternal()
182 AttrValue sloppy_attr; in AsGraphDefInternal()
186 AttrValue preserve_cardinality_attr; in AsGraphDefInternal()
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc701 tensorflow::AttrValue attr_value; in TFE_OpSetAttrFunction()
710 tensorflow::AttrValue attr_value; in TFE_OpSetAttrFunctionName()
919 const tensorflow::AttrValue& default_value, in SetOpAttrValueScalar()
922 case tensorflow::AttrValue::kS: { in SetOpAttrValueScalar()
927 case tensorflow::AttrValue::kI: in SetOpAttrValueScalar()
930 case tensorflow::AttrValue::kF: in SetOpAttrValueScalar()
933 case tensorflow::AttrValue::kB: in SetOpAttrValueScalar()
936 case tensorflow::AttrValue::kType: in SetOpAttrValueScalar()
940 case tensorflow::AttrValue::kShape: { in SetOpAttrValueScalar()
953 case tensorflow::AttrValue::kFunc: { in SetOpAttrValueScalar()
[all …]

123456789