| /external/tensorflow/tensorflow/core/framework/ |
| 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...] |
| node_def_util_test.cc | 95 AddNodeAttr("EXTRA", 17, &bad);
101 AddNodeAttr("T", 17, &bad);
151 AddNodeAttr("T", DT_STRING, &bad);
170 AddNodeAttr("e", "orange", &good);
176 AddNodeAttr("e", "foo", &bad);
|
| resource_mgr_test.cc | 146 AddNodeAttr("container", attr_container, &ndef);
149 AddNodeAttr("shared_name", attr_shared_name, &ndef);
|
| node_def_util.cc | 539 AddNodeAttr(attr_def.name(), attr_def.default_value(), node_def);
639 void AddNodeAttr(StringPiece name, const AttrValue& value, NodeDef* node_def) {
645 void AddNodeAttr(StringPiece name, T value, NodeDef* node_def) { \
648 AddNodeAttr(name, attr_value, node_def); \
|
| node_def_builder.cc | 259 AddNodeAttr(name, value, &node_def_);
|
| function_test.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| create_xla_launch_op.cc | 74 AddNodeAttr("Tconstants", DataTypeVector{}, &launch_def);
75 AddNodeAttr("Nresources", 0, &launch_def);
76 AddNodeAttr("Targs", fbody->arg_types, &launch_def);
77 AddNodeAttr("Tresults", fbody->ret_types, &launch_def);
81 AddNodeAttr("function", func, &launch_def);
|
| build_xla_launch_ops_pass.cc | 45 AddNodeAttr("Tconstants", constant_dtypes, &def);
46 AddNodeAttr("Targs", arg_dtypes, &def);
47 AddNodeAttr("Nresources", num_resources, &def);
48 AddNodeAttr("Tresults", result_dtypes, &def);
52 AddNodeAttr("function", function, &def);
|
| encapsulate_subgraphs_pass.cc | [all...] |
| /external/tensorflow/tensorflow/core/util/ |
| equal_graph_def_test.cc | 257 AddNodeAttr("foo", "bar", &same);
264 AddNodeAttr("foo", "bar", &actual);
273 AddNodeAttr("foo", "bar", &expected);
282 AddNodeAttr("foo", "bar", &actual);
283 AddNodeAttr("baz", 42, &actual);
286 AddNodeAttr("baz", 42, &expected);
287 AddNodeAttr("foo", "bar", &expected);
295 AddNodeAttr("foo", "bar", &actual);
296 AddNodeAttr("baz", 5, &actual);
299 AddNodeAttr("baz", 42, &expected)
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| rename_attribute_test.cc | 46 AddNodeAttr("foo", 23, mul_node1);
47 AddNodeAttr("bar", "something", mul_node1);
54 AddNodeAttr("foo", 46, add_node2);
55 AddNodeAttr("bob", 23, add_node2);
56 AddNodeAttr("bar", "something else", add_node2);
|
| /external/tensorflow/tensorflow/core/graph/ |
| gradients.cc | 72 AddNodeAttr("T", input.dtype(), &ndef);
106 AddNodeAttr("Tin", in_types, &ndef);
110 AddNodeAttr("Tout", n->input_types(), &ndef);
116 AddNodeAttr("f", func, &ndef);
287 AddNodeAttr("N", static_cast<int64>(grads.size()), &ndef);
288 AddNodeAttr("T", dtype, &ndef);
|
| graph_partition.cc | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| function.cc | 104 AddNodeAttr("T", BaseType(input.dtype()), &ndef);
118 AddNodeAttr("T", dtype, &ndef);
119 AddNodeAttr("index", index, &ndef);
133 AddNodeAttr("T", input.dtype(), &ndef);
134 AddNodeAttr("index", index, &ndef);
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils.cc | 367 AddNodeAttr(ATTR_OUTPUT_DATA_TYPES, data_types, node_def);
368 AddNodeAttr(ATTR_OUTPUT_SHAPES, shapes, node_def);
[all...] |
| constant_op.cc | 63 AddNodeAttr("dtype", ctx->output_type(0), ret);
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| functionalize_control_flow.cc | 182 AddNodeAttr("T", type, &ret_def);
183 AddNodeAttr("index", index, &ret_def);
[all...] |
| /external/tensorflow/tensorflow/core/ops/ |
| array_ops_test.cc | 261 AddNodeAttr("index_type", DT_INT32, &op.node_def);
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| randomized_tests.cc | 172 AddNodeAttr(attr_name, std::forward<T>(value), &node_def_);
199 AddNodeAttr(kXlaCompileAttr, true, test_def);
[all...] |
