Home
last modified time | relevance | path

Searched refs:AddNodeAttr (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h70 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
71 void AddNodeAttr(StringPiece name, StringPiece value, NodeDef* node_def);
72 void AddNodeAttr(StringPiece name, const char* value, NodeDef* node_def);
73 void AddNodeAttr(StringPiece name, int32 value, NodeDef* node_def);
74 void AddNodeAttr(StringPiece name, int64 value, NodeDef* node_def);
75 void AddNodeAttr(StringPiece name, float value, NodeDef* node_def);
76 void AddNodeAttr(StringPiece name, double value, NodeDef* node_def);
77 void AddNodeAttr(StringPiece name, bool value, NodeDef* node_def);
78 void AddNodeAttr(StringPiece name, DataType value, NodeDef* node_def);
79 void AddNodeAttr(StringPiece name, const PartialTensorShape& value,
[all …]
Dnode_def_util_test.cc98 AddNodeAttr("EXTRA", 17, &bad); in TEST()
104 AddNodeAttr("T", 17, &bad); in TEST()
154 AddNodeAttr("T", DT_STRING, &bad); in TEST()
173 AddNodeAttr("e", "orange", &good); in TEST()
179 AddNodeAttr("e", "foo", &bad); in TEST()
551 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
572 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
Dresource_mgr_test.cc172 AddNodeAttr("container", attr_container, &ndef); in ComputePolicy()
175 AddNodeAttr("shared_name", attr_shared_name, &ndef); in ComputePolicy()
Dnode_def_util.cc614 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def); in AddDefaultsToNodeDef()
722 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def) { in AddNodeAttr() function
728 void AddNodeAttr(StringPiece name, T value, NodeDef* node_def) { \
731 AddNodeAttr(name, attr_value, node_def); \
Dfunction_test.cc1287 AddNodeAttr("annotation", true, &ndef); in TEST()
1339 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
1345 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
Dnode_def_builder.cc268 AddNodeAttr(name, value, &node_def_); in Attr()
/external/tensorflow/tensorflow/core/util/
Dequal_graph_def_test.cc257 AddNodeAttr("foo", "bar", &same); in TEST_F()
264 AddNodeAttr("foo", "bar", &actual); in TEST_F()
273 AddNodeAttr("foo", "bar", &expected); in TEST_F()
282 AddNodeAttr("foo", "bar", &actual); in TEST_F()
283 AddNodeAttr("baz", 42, &actual); in TEST_F()
286 AddNodeAttr("baz", 42, &expected); in TEST_F()
287 AddNodeAttr("foo", "bar", &expected); in TEST_F()
295 AddNodeAttr("foo", "bar", &actual); in TEST_F()
296 AddNodeAttr("baz", 5, &actual); in TEST_F()
299 AddNodeAttr("baz", 42, &expected); in TEST_F()
[all …]
/external/tensorflow/tensorflow/tools/graph_transforms/
Drename_attribute_test.cc46 AddNodeAttr("foo", 23, mul_node1); in TestRenameAttribute()
47 AddNodeAttr("bar", "something", mul_node1); in TestRenameAttribute()
54 AddNodeAttr("foo", 46, add_node2); in TestRenameAttribute()
55 AddNodeAttr("bob", 23, add_node2); in TestRenameAttribute()
56 AddNodeAttr("bar", "something else", add_node2); in TestRenameAttribute()
/external/tensorflow/tensorflow/core/graph/
Dgradients.cc73 AddNodeAttr("dtype", DT_FLOAT, &read_def); in AddZerosLike()
82 AddNodeAttr("T", DT_FLOAT, &ndef); in AddZerosLike()
92 AddNodeAttr("T", input.dtype(), &ndef); in AddZerosLike()
127 AddNodeAttr("Tin", in_types, &ndef); in AddSymGrad()
138 AddNodeAttr("Tout", out_types, &ndef); in AddSymGrad()
144 AddNodeAttr("f", func, &ndef); in AddSymGrad()
315 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef); in SumGradients()
316 AddNodeAttr("T", dtype, &ndef); in SumGradients()
Dgraph_partition.cc999 AddNodeAttr("_start_time", start_time, dst_def); in Partition()
1109 AddNodeAttr("_start_time", send_start_time, dummy); in Partition()
1145 AddNodeAttr("_start_time", recv_start_time, recv); in Partition()
1147 AddNodeAttr("_start_time", recv_start_time, real_recv); in Partition()
1208 AddNodeAttr("_start_time", it.second.start_time, it.second.recv); in Partition()
1210 AddNodeAttr("_start_time", it.second.start_time, it.second.real_recv); in Partition()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc182 AddNodeAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, call_def->name(), in RewriteSubgraph()
184 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
331 AddNodeAttr("Tconstants", DataTypeVector{}, &def); in BuildXlaLaunchOps()
332 AddNodeAttr("Targs", arg_types, &def); in BuildXlaLaunchOps()
333 AddNodeAttr("Nresources", num_variables, &def); in BuildXlaLaunchOps()
334 AddNodeAttr("Tresults", output_types, &def); in BuildXlaLaunchOps()
337 AddNodeAttr("function", function, &def); in BuildXlaLaunchOps()
Dbuild_xla_ops_pass_test.cc93 AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node); in MakeXlaCompiledKernel()
94 AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node); in MakeXlaCompiledKernel()
95 AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node); in MakeXlaCompiledKernel()
Dextract_outside_compilation_pass.cc1563 AddNodeAttr("_outside_compilation_subgraph", old_name, node_def); in operator ()()
1566 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
1567 AddNodeAttr("shapes", *shapes, node_def); in operator ()()
1573 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
1574 AddNodeAttr("shapes", std::vector<TensorShapeProto>{}, node_def); in operator ()()
1576 AddNodeAttr("ancestors", std::vector<string>{}, node_def); in operator ()()
1577 AddNodeAttr("Tinputs", recv_at_host_dtypes, node_def); in operator ()()
1578 AddNodeAttr("Toutputs", send_from_host_dtypes, node_def); in operator ()()
1579 AddNodeAttr("key", absl::StrCat("host_compute_channel_", new_name), node_def); in operator ()()
Dencapsulate_subgraphs_pass.cc2658 AddNodeAttr(kXlaCompiledKernelAttr, true, node); in Run()
2659 AddNodeAttr(kXlaNumConstantArgsAttr, num_consts, node); in Run()
2660 AddNodeAttr(kXlaNumResourceArgsAttr, num_resources, node); in Run()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc104 AddNodeAttr("_output_shapes", {shape_proto}, &n); in SetShapes()
236 AddNodeAttr("_scoped_allocator", {0, 2}, &nd2); in TEST_F()
237 AddNodeAttr("_scoped_allocator", {6, 7}, &nd2); in TEST_F()
238 AddNodeAttr("_scoped_allocator", {2, 3}, &nd2); in TEST_F()
Dscoped_allocator_optimizer.cc212 AddNodeAttr(name, values, node_def); in ExtendNodeAttr()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc46 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
47 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
Dtf2xla_util.cc761 AddNodeAttr("T", dtype, &ndef); in BuildIdentityNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc98 AddNodeAttr("Targuments", t_args, map_defun_node); in CreateMapDefunWrapper()
99 AddNodeAttr("Tcaptured", t_captured, map_defun_node); in CreateMapDefunWrapper()
427 AddNodeAttr("Targuments", t_arguments, &choose_fastest_node); in AddNewChooseFastestNode()
428 AddNodeAttr("num_elements_per_branch", 10, &choose_fastest_node); in AddNewChooseFastestNode()
429 AddNodeAttr("branches", branches, &choose_fastest_node); in AddNewChooseFastestNode()
430 AddNodeAttr("other_arguments_lengths", other_arguments_lengths, in AddNewChooseFastestNode()
Drebatch.cc105 AddNodeAttr("SrcT", src_t, &cast_node); in AddCastNode()
106 AddNodeAttr("DstT", dst_t, &cast_node); in AddCastNode()
119 AddNodeAttr("T", type, &node); in AddBinaryNode()
Dvectorization_utils.cc86 AddNodeAttr("T", type, &ret_node_def); in AddMapDefunOutput()
87 AddNodeAttr("index", index, &ret_node_def); in AddMapDefunOutput()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc369 AddNodeAttr(ATTR_OUTPUT_DATA_TYPES, data_types, node_def); in AddOutputTensorShapeType()
370 AddNodeAttr(ATTR_OUTPUT_SHAPES, shapes, node_def); in AddOutputTensorShapeType()
1166 AddNodeAttr(ATTR_NODE_TYPE, attr_str, &node_def); in PlaceRemoteGraphArguments()
1447 AddNodeAttr("dtype", type, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
1448 AddNodeAttr("shape", shape, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
Dconstant_op.cc63 AddNodeAttr("dtype", ctx->output_type(0), ret); in StripTensorDataFromNodeDef()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc114 AddNodeAttr("T", BaseType(input.dtype()), &ndef); in AddIdentity()
128 AddNodeAttr("T", dtype, &ndef); in AddArg()
129 AddNodeAttr("index", index, &ndef); in AddArg()
143 AddNodeAttr("T", input.dtype(), &ndef); in AddRet()
144 AddNodeAttr("index", index, &ndef); in AddRet()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions_test.cc763 AddNodeAttr("Tag", "const_input_x", &const_input_x); in TEST_F()
767 AddNodeAttr("Tag", "const_input_y", &const_input_y); in TEST_F()

12