| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.h | 57 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def);
58 void AddNodeAttr(StringPiece name, StringPiece value, NodeDef* node_def);
59 void AddNodeAttr(StringPiece name, const char* value, NodeDef* node_def);
60 void AddNodeAttr(StringPiece name, int32 value, NodeDef* node_def);
61 void AddNodeAttr(StringPiece name, int64 value, NodeDef* node_def);
62 void AddNodeAttr(StringPiece name, float value, NodeDef* node_def);
63 void AddNodeAttr(StringPiece name, double value, NodeDef* node_def);
64 void AddNodeAttr(StringPiece name, bool value, NodeDef* node_def);
65 void AddNodeAttr(StringPiece name, DataType value, NodeDef* node_def);
66 void AddNodeAttr(StringPiece name, const PartialTensorShape& value,
[all …]
|
| D | node_def_util_test.cc | 95 AddNodeAttr("EXTRA", 17, &bad); in TEST()
101 AddNodeAttr("T", 17, &bad); in TEST()
151 AddNodeAttr("T", DT_STRING, &bad); in TEST()
170 AddNodeAttr("e", "orange", &good); in TEST()
176 AddNodeAttr("e", "foo", &bad); in TEST()
|
| D | node_def_util.cc | 539 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def); in AddDefaultsToNodeDef()
639 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def) { in AddNodeAttr() function
645 void AddNodeAttr(StringPiece name, T value, NodeDef* node_def) { \
648 AddNodeAttr(name, attr_value, node_def); \
|
| D | resource_mgr_test.cc | 146 AddNodeAttr("container", attr_container, &ndef); in ComputePolicy()
149 AddNodeAttr("shared_name", attr_shared_name, &ndef); in ComputePolicy()
|
| D | function_test.cc | 1216 AddNodeAttr("annotation", true, &ndef); in TEST()
1268 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
1274 AddNodeAttr(FunctionLibraryDefinition::kFuncAttr, nal, &ndef); in TEST()
|
| D | node_def_builder.cc | 259 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/compiler/jit/ |
| D | create_xla_launch_op.cc | 74 AddNodeAttr("Tconstants", DataTypeVector{}, &launch_def); in CreateXlaLaunchOp()
75 AddNodeAttr("Nresources", 0, &launch_def); in CreateXlaLaunchOp()
76 AddNodeAttr("Targs", fbody->arg_types, &launch_def); in CreateXlaLaunchOp()
77 AddNodeAttr("Tresults", fbody->ret_types, &launch_def); in CreateXlaLaunchOp()
81 AddNodeAttr("function", func, &launch_def); in CreateXlaLaunchOp()
|
| D | build_xla_launch_ops_pass.cc | 45 AddNodeAttr("Tconstants", constant_dtypes, &def); in BuildLaunchNode()
46 AddNodeAttr("Targs", arg_dtypes, &def); in BuildLaunchNode()
47 AddNodeAttr("Nresources", num_resources, &def); in BuildLaunchNode()
48 AddNodeAttr("Tresults", result_dtypes, &def); in BuildLaunchNode()
52 AddNodeAttr("function", function, &def); in BuildLaunchNode()
|
| D | encapsulate_subgraphs_pass.cc | 2108 AddNodeAttr(kXlaCompiledKernelAttr, true, node); in Run()
2109 AddNodeAttr(kXlaNumConstantArgsAttr, num_consts, node); in Run()
2110 AddNodeAttr(kXlaNumResourceArgsAttr, num_resources, node); in Run()
|
| /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 | 72 AddNodeAttr("T", input.dtype(), &ndef); in AddZerosLike()
106 AddNodeAttr("Tin", in_types, &ndef); in AddSymGrad()
110 AddNodeAttr("Tout", n->input_types(), &ndef); in AddSymGrad()
116 AddNodeAttr("f", func, &ndef); in AddSymGrad()
287 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef); in SumGradients()
288 AddNodeAttr("T", dtype, &ndef); in SumGradients()
|
| D | graph_partition.cc | 982 AddNodeAttr("_start_time", start_time, dst_def); in Partition()
1092 AddNodeAttr("_start_time", send_start_time, dummy); in Partition()
1128 AddNodeAttr("_start_time", recv_start_time, recv); in Partition()
1130 AddNodeAttr("_start_time", recv_start_time, real_recv); in Partition()
1190 AddNodeAttr("_start_time", it.second.start_time, it.second.recv); in Partition()
1192 AddNodeAttr("_start_time", it.second.start_time, it.second.real_recv); in Partition()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function.cc | 104 AddNodeAttr("T", BaseType(input.dtype()), &ndef); in AddIdentity()
118 AddNodeAttr("T", dtype, &ndef); in AddArg()
119 AddNodeAttr("index", index, &ndef); in AddArg()
133 AddNodeAttr("T", input.dtype(), &ndef); in AddRet()
134 AddNodeAttr("index", index, &ndef); in AddRet()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_utils.cc | 367 AddNodeAttr(ATTR_OUTPUT_DATA_TYPES, data_types, node_def); in AddOutputTensorShapeType()
368 AddNodeAttr(ATTR_OUTPUT_SHAPES, shapes, node_def); in AddOutputTensorShapeType()
1169 AddNodeAttr(ATTR_NODE_TYPE, attr_str, &node_def); in PlaceRemoteGraphArguments()
1450 AddNodeAttr("dtype", type, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
1451 AddNodeAttr("shape", shape, &placeholder_node); in ReplaceInputNodeByPlaceHolder()
|
| D | constant_op.cc | 63 AddNodeAttr("dtype", ctx->output_type(0), ret); in StripTensorDataFromNodeDef()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow.cc | 182 AddNodeAttr("T", type, &ret_def); in BuildRetvalNode()
183 AddNodeAttr("index", index, &ret_def); in BuildRetvalNode()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops_test.cc | 261 AddNodeAttr("index_type", DT_INT32, &op.node_def); in TEST()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 172 AddNodeAttr(attr_name, std::forward<T>(value), &node_def_); in Attr()
199 AddNodeAttr(kXlaCompileAttr, true, test_def); in BuildGraph()
|