/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.h | 70 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 …]
|
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() 551 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST() 572 AddNodeAttr("frame_name", "test_frame", &node_def); in TEST()
|
D | resource_mgr_test.cc | 172 AddNodeAttr("container", attr_container, &ndef); in ComputePolicy() 175 AddNodeAttr("shared_name", attr_shared_name, &ndef); in ComputePolicy()
|
D | node_def_util.cc | 614 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); \
|
D | function_test.cc | 1287 AddNodeAttr("annotation", true, &ndef); in TEST() 1339 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST() 1345 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
|
D | node_def_builder.cc | 268 AddNodeAttr(name, value, &node_def_); in Attr()
|
/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() 315 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef); in SumGradients() 316 AddNodeAttr("T", dtype, &ndef); in SumGradients()
|
D | graph_partition.cc | 999 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/ |
D | encapsulate_xla_computations_pass.cc | 182 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()
|
D | build_xla_ops_pass_test.cc | 93 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()
|
D | extract_outside_compilation_pass.cc | 1563 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 ()()
|
D | encapsulate_subgraphs_pass.cc | 2658 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/ |
D | scoped_allocator_optimizer_test.cc | 104 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()
|
D | scoped_allocator_optimizer.cc | 212 AddNodeAttr(name, values, node_def); in ExtendNodeAttr()
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | functionalize_control_flow_util.cc | 46 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode() 47 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
|
D | tf2xla_util.cc | 761 AddNodeAttr("T", dtype, &ndef); in BuildIdentityNode()
|
/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
D | map_vectorization.cc | 98 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()
|
D | rebatch.cc | 105 AddNodeAttr("SrcT", src_t, &cast_node); in AddCastNode() 106 AddNodeAttr("DstT", dst_t, &cast_node); in AddCastNode() 119 AddNodeAttr("T", type, &node); in AddBinaryNode()
|
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/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()
|
D | constant_op.cc | 63 AddNodeAttr("dtype", ctx->output_type(0), ret); in StripTensorDataFromNodeDef()
|
/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()
|
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | functions_test.cc | 763 AddNodeAttr("Tag", "const_input_x", &const_input_x); in TEST_F() 767 AddNodeAttr("Tag", "const_input_y", &const_input_y); in TEST_F()
|