Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/framework/
Dattr_value_util.h54 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);
60 void SetAttrValue(double value, AttrValue* out);
61 void SetAttrValue(bool value, AttrValue* out);
62 void SetAttrValue(DataType value, AttrValue* out);
63 void SetAttrValue(const TensorShape& value, AttrValue* out);
[all …]
Dattr_value_util.cc432 void SetAttrValue(const AttrValue& value, AttrValue* out) { *out = value; } in SetAttrValue() function
435 void SetAttrValue(ARG_TYPE value, AttrValue* out) { out->set_##FIELD(value); }
438 void SetAttrValue(ARG_TYPE value, AttrValue* out) { \
461 void SetAttrValue(StringPiece value, AttrValue* out) { in DEFINE_SET_ATTR_VALUE_ONE()
465 void SetAttrValue(const gtl::ArraySlice<StringPiece> value, AttrValue* out) { in SetAttrValue() function
472 void SetAttrValue(const TensorShape& value, AttrValue* out) { in SetAttrValue() function
476 void SetAttrValue(const TensorShapeProto& value, AttrValue* out) { in SetAttrValue() function
480 void SetAttrValue(const PartialTensorShape& value, AttrValue* out) { in SetAttrValue() function
484 void SetAttrValue(const gtl::ArraySlice<TensorShape> value, AttrValue* out) { in SetAttrValue() function
491 void SetAttrValue(gtl::ArraySlice<TensorShapeProto> 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.cc1292 void SetAttrValue(FunctionDef* fdef, const string& attr, const T& value) { in SetAttrValue() function
1294 SetAttrValue(value, &attr_value); in SetAttrValue()
1302 SetAttrValue(fdef, "annotation", true); in TEST()
1303 SetAttrValue(fdef, "options", "some string data"); in TEST()
1323 SetAttrValue(fdef, "annotation", true); in TEST()
1325 SetAttrValue(fdef, "annotation", false); in TEST()
1465 SetAttrValue(&fdef2, "ExtraAttr", true); in TEST()
1472 SetAttrValue(&fdef2, "Foo", true); in TEST()
1473 SetAttrValue(&fdef3, "Foo", true); in TEST()
1474 SetAttrValue(&fdef2, "Bar", 123); in TEST()
[all …]
Dnode_def_builder.cc276 SetAttrValue(value, &attr_value); \
Dnode_def_util.cc730 SetAttrValue(value, &attr_value); \
768 SetAttrValue(value, &attr_value); \
Dfunction.h95 SetAttrValue(val, &proto); in AttrValueWrapper()
Ddataset.h193 SetAttrValue(value, attr); in BuildAttrValue()
Dfunction.cc100 SetAttrValue(val, &((*ndef->mutable_attr())[name])); in AddAttr()
1501 SetAttrValue(val, &proto); in InitFromString()
/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()
136 SetAttrValue("f", &f_attr); in TEST()
139 SetAttrValue("Targuments", &args_attr); in TEST()
156 SetAttrValue("output_shapes", &shapes_attr); in TEST()
159 SetAttrValue("output_types", &types_attr); in TEST()
223 SetAttrValue("f", &f_attr); in TEST()
226 SetAttrValue("Targuments", &args_attr); in TEST()
[all …]
Dshuffle_and_repeat_fusion_test.cc34 SetAttrValue("output_shapes", &shapes_attr); in TEST()
37 SetAttrValue("output_types", &types_attr); in TEST()
102 SetAttrValue("output_shapes", &shapes_attr); in TEST()
105 SetAttrValue("output_types", &types_attr); in TEST()
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.cc107 SetAttrValue(dtype, &dtype_attr); in BuildConstNode()
157 SetAttrValue(t, &value_attr); in BuildConstNode()
406 SetAttrValue(dtype, &dtype_attr); in NodeWithTensorContent()
455 SetAttrValue(t, &value_attr); in NodeWithTensorContent()
522 SetAttrValue(dtype, &dtype_attr); in BuildUnaryNode()
621 SetAttrValue(2, &value_attr); in TEST()
632 SetAttrValue(types, &value_attr); in TEST()
644 SetAttrValue(types, &value_attr); in TEST()
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc45 SetAttrValue(input_tensor->dtype(), &typeAttr); in TestBitcastOp()
48 SetAttrValue(out_type, &outTypeAttr); in TestBitcastOp()
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator_test.cc207 SetAttrValue(data_format, &(*attr_data_format)["data_format"]); in DescribeFusedConv2DBiasActivation()
209 SetAttrValue(filter_format, &(*attr_filter_format)["filter_format"]); in DescribeFusedConv2DBiasActivation()
399 SetAttrValue(data_format, &(*attr)["data_format"]); in DescribePoolingOp()
400 SetAttrValue(padding, &(*attr)["padding"]); in DescribePoolingOp()
401 SetAttrValue(strides, &(*attr)["strides"]); in DescribePoolingOp()
402 SetAttrValue(ksize, &(*attr)["ksize"]); in DescribePoolingOp()
445 SetAttrValue(is_training, &(*attr)["is_training"]); in DescribeFusedBatchNorm()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.h132 SetAttrValue(value, &attr_value); in SetInAttrValueMap()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_test.cc38 SetAttrValue(type, &attr_value); in TypeAttrValue()
Dxla_jit_compiled_cpu_function_test.cc36 SetAttrValue(type, &attr_value); in TypeAttrValue()
/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.h132 SetAttrValue(std::forward<T>(value), &attrs_.back().second); in WithAttrImpl()
Dgraph.h184 SetAttrValue(val, AddAttrHelper(name)); in AddAttr()
Dgraph_partition.cc935 SetAttrValue(incarnation, in SetIncarnation()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h78 SetAttrValue(value, &attr_value); in SetNodeAttr()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc482 SetAttrValue(fused_ops, &(*attr)["fused_ops"]); in SetFusedConv2DAttributes()
483 SetAttrValue(num_args, &(*attr)["num_args"]); in SetFusedConv2DAttributes()
484 SetAttrValue(epsilon, &(*attr)["epsilon"]); // required only for BatchNorm in SetFusedConv2DAttributes()
/external/tensorflow/tensorflow/core/kernels/
Dcollective_ops.cc160 SetAttrValue(col_params_.instance.data_type, in CollectiveReduceOpKernel()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc381 SetAttrValue(value, &attr_value); in SetNodeAttr()

12