| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 76 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
77 void AddNodeAttr(StringPiece name, AttrValue&& value, NodeDef* node_def);
78 void AddNodeAttr(StringPiece name, StringPiece value, NodeDef* node_def);
79 void AddNodeAttr(StringPiece name, const char* value, NodeDef* node_def);
80 void AddNodeAttr(StringPiece name, int32 value, NodeDef* node_def);
81 void AddNodeAttr(StringPiece name, int64 value, NodeDef* node_def);
82 void AddNodeAttr(StringPiece name, float value, NodeDef* node_def);
83 void AddNodeAttr(StringPiece name, double value, NodeDef* node_def);
84 void AddNodeAttr(StringPiece name, bool value, NodeDef* node_def);
85 void AddNodeAttr(StringPiece name, DataType value, NodeDef* node_def);
[all …]
|
| D | node_def_util_test.cc | 98 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()
608 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
629 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
|
| D | node_def_util.cc | 696 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def); in AddDefaultsToNodeDef()
830 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def) { in AddNodeAttr() function
835 void AddNodeAttr(StringPiece name, AttrValue&& value, NodeDef* node_def) { in AddNodeAttr() function
840 void AddNodeAttr(StringPiece name, T value, NodeDef* node_def) { \
843 AddNodeAttr(name, attr_value, node_def); \
|
| D | resource_mgr_test.cc | 174 AddNodeAttr("container", attr_container, &ndef); in ComputePolicy()
177 AddNodeAttr("shared_name", attr_shared_name, &ndef); in ComputePolicy()
|
| D | node_def_builder.cc | 279 AddNodeAttr(name, value, &node_def_); in Attr()
286 AddNodeAttr(name, std::move(value), &node_def_); in Attr()
|
| D | function_test.cc | 1283 AddNodeAttr("annotation", true, &ndef); in TEST()
1335 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
1341 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | equal_graph_def_test.cc | 257 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/ |
| D | rename_attribute_test.cc | 46 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/ |
| D | gradients.cc | 73 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()
323 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef); in SumGradients()
324 AddNodeAttr("T", dtype, &ndef); in SumGradients()
|
| D | graph_partition.cc | 1024 AddNodeAttr("_start_time", start_time, dst_def); in Partition()
1134 AddNodeAttr("_start_time", send_start_time, dummy); in Partition()
1170 AddNodeAttr("_start_time", recv_start_time, recv); in Partition()
1172 AddNodeAttr("_start_time", recv_start_time, real_recv); in Partition()
1233 AddNodeAttr("_start_time", it.second.start_time, it.second.recv); in Partition()
1235 AddNodeAttr("_start_time", it.second.start_time, it.second.real_recv); in Partition()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_xla_computations_pass.cc | 183 AddNodeAttr(EncapsulateXlaComputationsPass::kXlaClusterAttr, call_def->name(), in RewriteSubgraph()
185 AddNodeAttr("_variable_start_index", variable_start_index, call_def); in RewriteSubgraph()
332 AddNodeAttr("Tconstants", DataTypeVector{}, &def); in BuildXlaLaunchOps()
333 AddNodeAttr("Targs", arg_types, &def); in BuildXlaLaunchOps()
334 AddNodeAttr("Nresources", num_variables, &def); in BuildXlaLaunchOps()
335 AddNodeAttr("Tresults", output_types, &def); in BuildXlaLaunchOps()
338 AddNodeAttr("function", function, &def); in BuildXlaLaunchOps()
|
| D | shape_inference.cc | 125 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()
|
| D | build_xla_ops_pass_test.cc | 94 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()
|
| D | extract_outside_compilation_pass.cc | 2232 AddNodeAttr("_outside_compilation_subgraph", old_name, node_def); in operator ()()
2235 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2236 AddNodeAttr("shapes", *shapes, node_def); in operator ()()
2242 AddNodeAttr("shape_inference_graph", shape_inference_graph, node_def); in operator ()()
2243 AddNodeAttr("shapes", std::vector<TensorShapeProto>{}, node_def); in operator ()()
2245 AddNodeAttr("ancestors", std::vector<string>{}, node_def); in operator ()()
2246 AddNodeAttr("Tinputs", recv_at_host_dtypes, node_def); in operator ()()
2247 AddNodeAttr("Toutputs", send_from_host_dtypes, node_def); in operator ()()
2248 AddNodeAttr("key", absl::StrCat("host_compute_channel_", new_name), node_def); in operator ()()
|
| D | encapsulate_subgraphs_pass.cc | 1299 AddNodeAttr(kXlaCompiledKernelAttr, true, node); in Run()
1300 AddNodeAttr(kXlaNumConstantArgsAttr, num_consts, node); in Run()
1301 AddNodeAttr(kXlaNumResourceArgsAttr, num_resources, node); in Run()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | map_vectorization.cc | 99 AddNodeAttr("Targuments", t_args, map_defun_node); in CreateMapDefunWrapper()
100 AddNodeAttr("Tcaptured", t_captured, map_defun_node); in CreateMapDefunWrapper()
428 AddNodeAttr("Targuments", t_arguments, &choose_fastest_node); in AddNewChooseFastestNode()
429 AddNodeAttr("num_elements_per_branch", 100, &choose_fastest_node); in AddNewChooseFastestNode()
430 AddNodeAttr("branches", branches, &choose_fastest_node); in AddNewChooseFastestNode()
431 AddNodeAttr("other_arguments_lengths", other_arguments_lengths, in AddNewChooseFastestNode()
|
| D | map_and_filter_fusion.cc | 100 AddNodeAttr("Targuments", std::vector<DataType>({}), &filter_node); in MakeFilterNode()
136 AddNodeAttr("Targuments", std::vector<DataType>({}), &map_node); in MakeMapNode()
|
| D | slack.cc | 86 AddNodeAttr("slack_period", slack_period_, dataset_node); in RecursivelyHandleOp()
|
| D | vectorization_utils.cc | 86 AddNodeAttr("T", type, &ret_node_def); in AddMapDefunOutput()
87 AddNodeAttr("index", index, &ret_node_def); in AddMapDefunOutput()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_util.cc | 47 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
48 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | scoped_allocator_optimizer_test.cc | 247 AddNodeAttr("_output_shapes", {shape_proto}, &n); in SetShapes()
430 AddNodeAttr("_scoped_allocator", {0, 2}, &nd2); in TEST_F()
431 AddNodeAttr("_scoped_allocator", {6, 7}, &nd2); in TEST_F()
432 AddNodeAttr("_scoped_allocator", {2, 3}, &nd2); in TEST_F()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops_test.cc | 176 AddNodeAttr("incompatible_shape_error", true, &op.node_def); in TEST()
181 AddNodeAttr("incompatible_shape_error", false, &op.node_def); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 369 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()
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | functions_test.cc | 590 AddNodeAttr("Tag", "const_input_x", &const_input_x); in TEST_F()
594 AddNodeAttr("Tag", "const_input_y", &const_input_y); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 114 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()
|