Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h83 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
84 void AddNodeAttr(StringPiece name, AttrValue&& value, NodeDef* node_def);
85 void AddNodeAttr(StringPiece name, StringPiece value, NodeDef* node_def);
86 void AddNodeAttr(StringPiece name, const char* value, NodeDef* node_def);
87 void AddNodeAttr(StringPiece name, int32 value, NodeDef* node_def);
88 void AddNodeAttr(StringPiece name, int64 value, NodeDef* node_def);
89 void AddNodeAttr(StringPiece name, float value, NodeDef* node_def);
90 void AddNodeAttr(StringPiece name, double value, NodeDef* node_def);
91 void AddNodeAttr(StringPiece name, bool value, NodeDef* node_def);
92 void AddNodeAttr(StringPiece name, DataType value, NodeDef* node_def);
[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()
622 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
636 AddNodeAttr(kColocationAttrName, in TEST()
676 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
Dmemory_types_test.cc67 AddNodeAttr("_input_hostmem", {0}, &node_def); in TEST()
68 AddNodeAttr("_output_hostmem", {6, 7}, &node_def); in TEST()
Dnode_def_util.cc727 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def); in AddDefaultsToNodeDef()
863 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def) { in AddNodeAttr() function
868 void AddNodeAttr(StringPiece name, AttrValue&& value, NodeDef* node_def) { in AddNodeAttr() function
873 void AddNodeAttr(StringPiece name, T value, NodeDef* node_def) { \
876 AddNodeAttr(name, attr_value, node_def); \
Dresource_mgr_test.cc176 AddNodeAttr("container", attr_container, &ndef); in ComputePolicy()
179 AddNodeAttr("shared_name", attr_shared_name, &ndef); in ComputePolicy()
Dnode_def_builder.cc279 AddNodeAttr(name, value, &node_def_); in Attr()
286 AddNodeAttr(name, std::move(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/common_runtime/
Dgradients.cc74 AddNodeAttr("dtype", DT_FLOAT, &read_def); in AddZerosLike()
83 AddNodeAttr("T", DT_FLOAT, &ndef); in AddZerosLike()
93 AddNodeAttr("T", input.dtype(), &ndef); in AddZerosLike()
128 AddNodeAttr("Tin", in_types, &ndef); in AddSymGrad()
139 AddNodeAttr("Tout", out_types, &ndef); in AddSymGrad()
145 AddNodeAttr("f", func, &ndef); in AddSymGrad()
324 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef); in SumGradients()
325 AddNodeAttr("T", dtype, &ndef); in SumGradients()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_xla_computations_pass.cc178 AddNodeAttr(kXlaClusterIdAttr, call_def->name(), call_def); in RewriteSubgraph()
179 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
326 AddNodeAttr("Tconstants", DataTypeVector{}, &def); in BuildXlaLaunchOps()
327 AddNodeAttr("Targs", arg_types, &def); in BuildXlaLaunchOps()
328 AddNodeAttr("Nresources", num_variables, &def); in BuildXlaLaunchOps()
329 AddNodeAttr("Tresults", output_types, &def); in BuildXlaLaunchOps()
332 AddNodeAttr("function", function, &def); in BuildXlaLaunchOps()
Dshape_inference.cc125 AddNodeAttr("dtype", dtype, &const_def); in PropagateShapes()
137 AddNodeAttr("value", value, &const_def); in PropagateShapes()
140 AddNodeAttr(attr.first, attr.second, &const_def); in PropagateShapes()
Dbuild_xla_ops_pass_test.cc94 AddNodeAttr(kXlaCompiledKernelAttr, true, &call_node); in MakeXlaCompiledKernel()
95 AddNodeAttr(kXlaNumConstantArgsAttr, num_constant_args, &call_node); in MakeXlaCompiledKernel()
96 AddNodeAttr(kXlaNumResourceArgsAttr, num_resource_args, &call_node); in MakeXlaCompiledKernel()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_configuration_rewrite_pass.cc63 AddNodeAttr("N", number_of_hosts, &config_def); in AddConfigurationNode()
64 AddNodeAttr("enable_whole_mesh_compilations", enable_whole_mesh_compilations, in AddConfigurationNode()
85 AddNodeAttr("enable_whole_mesh_compilations", enable_whole_mesh_compilations, in AddHostConfigNode()
106 AddNodeAttr("N", static_cast<int32>(host_configuration_nodes.size()), in AddWaitNode()
108 AddNodeAttr("startup_timeout_sec", kDefaultStartupTimeout, &wait_def); in AddWaitNode()
155 AddNodeAttr("T", DT_STRING, &sync_def); in AddSynchronizationNode()
Dincomplete_nodedef_builder.cc33 AddNodeAttr(attr, type, &nodedef_); in AddAttr()
39 AddNodeAttr(attr, val, &nodedef_); in AddAttr()
Dhost_training_loop_optimization_util.cc331 AddNodeAttr("T", dtype, &at_loop_iteration_nodedef); in GetOrCreateBeforeEachIterationNode()
369 AddNodeAttr("T", dtype, &after_last_loop_iteration); in AddNoOpAfterLastIteration()
505 AddNodeAttr("dtype", DT_STRING, &default_sharding); in AddReshardOp()
536 AddNodeAttr("N", static_cast<int>(info.var_inputs.size()), in AddReshardOp()
569 AddNodeAttr("dtype", DT_STRING, &var_handle_def); in AddReshardOp()
570 AddNodeAttr("shape", TensorShape({}), &var_handle_def); in AddReshardOp()
590 AddNodeAttr("N", static_cast<int>(info.var_inputs.size()), in AddReshardOp()
Ddistributed_tpu_rewrite_pass.cc672 AddNodeAttr("dtype", DT_INT32, &paddings_def); in CreatePadNode()
686 AddNodeAttr("value", sizes_tensor_proto, &paddings_def); in CreatePadNode()
696 AddNodeAttr("T", dtype, &pad_def); in CreatePadNode()
697 AddNodeAttr("Tpaddings", DT_INT32, &pad_def); in CreatePadNode()
736 AddNodeAttr("dtype", DT_INT32, &split_dim_def); in CreateSplitNode()
742 AddNodeAttr("value", tensor_proto, &split_dim_def); in CreateSplitNode()
751 AddNodeAttr("num_split", num_splits, &split_def); in CreateSplitNode()
752 AddNodeAttr("T", dtype, &split_def); in CreateSplitNode()
915 AddNodeAttr("dtype", DT_INT32, &concat_dim_def); in CreateConcatNode()
922 AddNodeAttr("value", tensor_proto, &concat_dim_def); in CreateConcatNode()
[all …]
Dvariable_merger_pass.cc109 AddNodeAttr("N", num_reads, &node_def); in MergeReadVariableOps()
110 AddNodeAttr("dtypes", dtypes, &node_def); in MergeReadVariableOps()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_vectorization.cc101 AddNodeAttr("Targuments", t_args, map_defun_node); in CreateMapDefunWrapper()
102 AddNodeAttr("Tcaptured", t_captured, map_defun_node); in CreateMapDefunWrapper()
448 AddNodeAttr("Targuments", t_arguments, &choose_fastest_node); in AddNewChooseFastestNode()
449 AddNodeAttr("num_elements_per_branch", 100, &choose_fastest_node); in AddNewChooseFastestNode()
450 AddNodeAttr("branches", branches, &choose_fastest_node); in AddNewChooseFastestNode()
451 AddNodeAttr("other_arguments_lengths", other_arguments_lengths, in AddNewChooseFastestNode()
Dmap_and_filter_fusion.cc100 AddNodeAttr("Targuments", std::vector<DataType>({}), &filter_node); in MakeFilterNode()
136 AddNodeAttr("Targuments", std::vector<DataType>({}), &map_node); in MakeMapNode()
Dslack.cc86 AddNodeAttr("slack_period", slack_period_, dataset_node); in RecursivelyHandleOp()
Dmap_parallelization.cc50 AddNodeAttr("deterministic", "true", &parallel_map); in MakeParallelMap()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_util.cc47 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
48 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dscoped_allocator_optimizer_test.cc277 AddNodeAttr("_output_shapes", {shape_proto}, &n); in SetShapes()
459 AddNodeAttr("_scoped_allocator", {0, 2}, &nd2); in TEST_F()
460 AddNodeAttr("_scoped_allocator", {6, 7}, &nd2); in TEST_F()
461 AddNodeAttr("_scoped_allocator", {2, 3}, &nd2); in TEST_F()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops_test.cc176 AddNodeAttr("incompatible_shape_error", true, &op.node_def); in TEST()
181 AddNodeAttr("incompatible_shape_error", false, &op.node_def); in TEST()
610 AddNodeAttr("incompatible_shape_error", true, &op.node_def); in TEST()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc1026 AddNodeAttr("_start_time", start_time, dst_def); in Partition()
1136 AddNodeAttr("_start_time", send_start_time, dummy); in Partition()
1172 AddNodeAttr("_start_time", recv_start_time, recv); in Partition()
1174 AddNodeAttr("_start_time", recv_start_time, real_recv); in Partition()
1235 AddNodeAttr("_start_time", it.second.start_time, it.second.recv); in Partition()
1237 AddNodeAttr("_start_time", it.second.start_time, it.second.real_recv); in Partition()

12