Home
last modified time | relevance | path

Searched refs:SetAttrValue (Results 1 – 25 of 36) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h54 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);
60 void SetAttrValue(float value, AttrValue* out);
61 void SetAttrValue(double value, AttrValue* out);
62 void SetAttrValue(bool value, AttrValue* out);
63 void SetAttrValue(DataType value, AttrValue* out);
[all …]
Dattr_value_util.cc455 void SetAttrValue(const AttrValue& value, AttrValue* out) { *out = value; } in SetAttrValue() function
458 void SetAttrValue(ARG_TYPE value, AttrValue* out) { out->set_##FIELD(value); }
461 void SetAttrValue(ARG_TYPE value, AttrValue* out) { \
484 void SetAttrValue(const tstring& value, AttrValue* out) { in DEFINE_SET_ATTR_VALUE_ONE()
488 void SetAttrValue(gtl::ArraySlice<tstring> value, AttrValue* out) { in SetAttrValue() function
495 void SetAttrValue(StringPiece value, AttrValue* out) { in SetAttrValue() function
499 void SetAttrValue(const gtl::ArraySlice<StringPiece> value, AttrValue* out) { in SetAttrValue() function
513 void SetAttrValue(const TensorShape& value, AttrValue* out) { in SetAttrValue() function
517 void SetAttrValue(const TensorShapeProto& value, AttrValue* out) { in SetAttrValue() function
521 void SetAttrValue(const PartialTensorShape& value, AttrValue* out) { in SetAttrValue() function
[all …]
Dattr_value_util_test.cc32 SetAttrValue(value, &ret); in V()
142 SetAttrValue(string(40, '-'), &attr_value); in TEST()
149 SetAttrValue(string(80, '-'), &attr_value); in TEST()
158 SetAttrValue(alist, &attr_value); in TEST()
167 SetAttrValue(alist, &attr_value); in TEST()
Dfunction_test.cc1298 void SetAttrValue(FunctionDef* fdef, const string& attr, const T& value) { in SetAttrValue() function
1300 SetAttrValue(value, &attr_value); in SetAttrValue()
1308 SetAttrValue(fdef, "annotation", true); in TEST()
1309 SetAttrValue(fdef, "options", "some string data"); in TEST()
1329 SetAttrValue(fdef, "annotation", true); in TEST()
1331 SetAttrValue(fdef, "annotation", false); in TEST()
1471 SetAttrValue(&fdef2, "ExtraAttr", true); in TEST()
1478 SetAttrValue(&fdef2, "Foo", true); in TEST()
1479 SetAttrValue(&fdef3, "Foo", true); in TEST()
1480 SetAttrValue(&fdef2, "Bar", 123); in TEST()
[all …]
Dnode_def_builder.cc294 SetAttrValue(value, &attr_value); \
Dnode_def_util.cc875 SetAttrValue(value, &attr_value); \
913 SetAttrValue(value, &attr_value); \
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion_test.cc53 SetAttrValue("f", &f_attr); in TEST()
56 SetAttrValue("Targuments", &args_attr); in TEST()
70 SetAttrValue("output_shapes", &shapes_attr); in TEST()
73 SetAttrValue("output_types", &types_attr); in TEST()
135 SetAttrValue("f", &f_attr); in TEST()
138 SetAttrValue("Targuments", &args_attr); in TEST()
155 SetAttrValue("output_shapes", &shapes_attr); in TEST()
158 SetAttrValue("output_types", &types_attr); in TEST()
221 SetAttrValue("f", &f_attr); in TEST()
224 SetAttrValue("Targuments", &args_attr); in TEST()
[all …]
Dshuffle_and_repeat_fusion_test.cc38 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
41 SetAttrValue(kOutputTypes, &types_attr); in TEST()
106 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
109 SetAttrValue(kOutputTypes, &types_attr); in TEST()
171 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
174 SetAttrValue(kOutputTypes, &types_attr); in TEST()
243 SetAttrValue(kOutputShapes, &shapes_attr); in TEST()
246 SetAttrValue(kOutputTypes, &types_attr); in TEST()
Dslack_test.cc36 SetAttrValue(std::vector<TensorShape>({{}}), &shapes_attr); in SetupGrapplerItem()
39 SetAttrValue(std::vector<DataType>({DT_INT64}), &types_attr); in SetupGrapplerItem()
126 SetAttrValue(std::vector<TensorShape>({{}}), &shapes_attr); in TEST()
128 SetAttrValue(std::vector<DataType>({DT_INT64}), &types_attr); in TEST()
142 SetAttrValue(DT_VARIANT, &variant_type_attr); in TEST()
Dautotune_buffer_sizes_test.cc179 SetAttrValue("value", &attr_val); in TEST_P()
194 SetAttrValue(legacy_autotune, &legacy_autotune_attr); in TEST_P()
196 SetAttrValue(0, &buffer_size_min_attr); in TEST_P()
Dnoop_elimination_test.cc30 SetAttrValue("output_shapes", &shapes_attr); in GetCommonAttributes()
31 SetAttrValue("output_types", &types_attr); in GetCommonAttributes()
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow_test.cc109 SetAttrValue(dtype, &dtype_attr); in BuildConstNode()
164 SetAttrValue(t, &value_attr); in BuildConstNode()
203 SetAttrValue(0.5, &dtype_attr); in TEST()
450 SetAttrValue(dtype, &dtype_attr); in NodeWithTensorContent()
499 SetAttrValue(t, &value_attr); in NodeWithTensorContent()
566 SetAttrValue(dtype, &dtype_attr); in BuildUnaryNode()
665 SetAttrValue(2, &value_attr); in TEST()
676 SetAttrValue(types, &value_attr); in TEST()
688 SetAttrValue(types, &value_attr); in TEST()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc44 SetAttrValue(input_tensor->dtype(), &typeAttr); in TestBitcastOp()
47 SetAttrValue(out_type, &outTypeAttr); in TestBitcastOp()
Dsummary_op_test.cc64 SetAttrValue(values->dtype(), &valuesTypeAttr); in TestScalarSummaryOp()
/external/tensorflow/tensorflow/security/fuzzing/
DAreAttrValuesEqual_fuzz.cc32 SetAttrValue(value, &ret); in createAttrValue()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc1023 SetAttrValue(src_attr.at("U"), &(*attr)["U"]); in IsCpuCompatibleDataType()
1026 SetAttrValue(src_attr.at("T"), &(*attr)["U"]); in IsCpuCompatibleDataType()
1028 SetAttrValue(DT_FLOAT, &(*attr)["U"]); in IsCpuCompatibleDataType()
1054 SetAttrValue(fused_ops, &(*attr)["fused_ops"]); in IsCpuCompatibleDataType()
1055 SetAttrValue(num_args, &(*attr)["num_args"]); in IsCpuCompatibleDataType()
1056 SetAttrValue(epsilon, &(*attr)["epsilon"]); // required only for BatchNorm in IsCpuCompatibleDataType()
1409 SetAttrValue(activation.op(), &(*attrs)["activation_mode"]); in IsCpuCompatibleDataType()
1412 SetAttrValue(1, &(*attrs)["num_side_inputs"]); in IsCpuCompatibleDataType()
1416 SetAttrValue(0, &(*attrs)["num_side_inputs"]); in IsCpuCompatibleDataType()
/external/tensorflow/tensorflow/c/
Dkernels_test.cc289 SetAttrValue(gtl::ArraySlice<std::string>(attr_in, 3), &v); in TEST_F()
339 SetAttrValue(gtl::ArraySlice<int64>(attr_in, 4), &v); in TEST_F()
389 SetAttrValue(gtl::ArraySlice<float>(attr_in, 3), &v); in TEST_F()
439 SetAttrValue(gtl::ArraySlice<bool>(attr_in, 4), &v); in TEST_F()
489 SetAttrValue(gtl::ArraySlice<DataType>(attr_in, 4), &v); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.h132 SetAttrValue(std::forward<T>(value), &attrs_.back().second); in WithAttrImpl()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.h120 SetAttrValue(value, &attr_tmp_); in Set()
Deager_operation.h84 Status SetAttrValue(const char* attr_name, const AttrValue& value);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc38 SetAttrValue(type, &attr_value); in TypeAttrValue()
Dxla_jit_compiled_cpu_function_test.cc49 SetAttrValue(type, &attr_value); in TypeAttrValue()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h78 SetAttrValue(value, &attr_value); in SetNodeAttr()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc252 SetAttrValue(data_format, &(*attr_data_format)["data_format"]); in DescribeFusedConv2DBiasActivation()
254 SetAttrValue(filter_format, &(*attr_filter_format)["filter_format"]); in DescribeFusedConv2DBiasActivation()
459 SetAttrValue(data_format, &(*attr)["data_format"]); in DescribePoolingOp()
460 SetAttrValue(padding, &(*attr)["padding"]); in DescribePoolingOp()
461 SetAttrValue(strides, &(*attr)["strides"]); in DescribePoolingOp()
462 SetAttrValue(ksize, &(*attr)["ksize"]); in DescribePoolingOp()
505 SetAttrValue(is_training, &(*attr)["is_training"]); in DescribeFusedBatchNorm()
/external/tensorflow/tensorflow/core/kernels/
Dragged_tensor_to_tensor_op_test.cc514 SetAttrValue(row_partition_types, in SetAttributes()

12