| /external/tensorflow/tensorflow/core/common_runtime/ |
| direct_session_with_tracking_alloc_test.cc | 51 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
52 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
53 Node* a = test::graph::Constant(&graph, a_tensor);
153 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
154 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
155 Node* a = test::graph::Constant(&graph, a_tensor);
244 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
245 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
246 Node* a = test::graph::Constant(&graph, a_tensor);
306 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}))
[all...] |
| direct_session_test.cc | 62 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
63 test::FillValues<float>(&a_tensor, a_values);
64 Node* a = test::graph::Constant(&graph, a_tensor);
234 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
235 a_tensor.flat<float>().setRandom();
236 Node* a = test::graph::Constant(&graph, a_tensor);
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| list_kernels.h | 203 const Tensor& a_tensor = a.tensors[i]; local
205 if (a_tensor.dtype() == DT_INVALID) {
210 out->tensors.push_back(a_tensor);
213 if (a_tensor.shape() != b_tensor.shape()) {
218 a_tensor.shape().DebugString(), " and the other is ",
223 c->allocate_temp(a_tensor.dtype(), a_tensor.shape(), &out_tensor));
229 a_tensor.flat<dtype>() + b_tensor.flat<dtype>(); \
|
| /external/tensorflow/tensorflow/core/debug/ |
| grpc_session_debug_test.cc | 55 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2}));
56 test::FillValues<float>(&a_tensor, {1.0, 2.0});
57 Node* a = test::graph::Constant(&graph, a_tensor);
232 Tensor a_tensor(DT_STRING, TensorShape({2, 2}));
234 a_tensor.flat<string>()(i) = "hello, world";
236 Node* a = test::graph::Constant(&graph, a_tensor);
|
| debug_gateway_test.cc | 64 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
65 test::FillValues<float>(&a_tensor, a_values);
66 Node* a = test::graph::Constant(&graph, a_tensor);
522 Tensor a_tensor(DT_FLOAT, TensorShape({1, 1}));
523 test::FillValues<float>(&a_tensor, {42.0});
524 Node* a = test::graph::Constant(&graph, a_tensor);
527 Node* c = test::graph::Constant(&graph, a_tensor);
[all...] |
| /external/tensorflow/tensorflow/core/platform/ |
| device_tracer_test.cc | 58 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
59 test::FillValues<float>(&a_tensor, a_values);
60 Node* a = test::graph::Constant(&graph, a_tensor);
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_session_test.cc | 47 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2}));
48 test::FillValues<float>(&a_tensor, {1, 2});
49 Node* a = test::graph::Constant(&graph, a_tensor);
220 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
222 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
223 Node* a = test::graph::Constant(&graph, a_tensor);
308 Tensor a_tensor(DT_FLOAT, TensorShape({1, kSize}));
311 a_tensor.flat<float>()(i) = 2;
314 Node* a = test::graph::Constant(&graph, a_tensor);
426 Tensor a_tensor(DT_STRING, TensorShape({2, 2}))
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| quantize_nodes_test.cc | 213 Tensor a_tensor(DT_FLOAT, TensorShape({m, k}));
214 test::FillValues<float>(&a_tensor, a_values);
215 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor));
376 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2, 3}));
377 test::FillValues<float>(&a_tensor, {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12});
378 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor));
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_test.cc | 295 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
296 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
297 test::graph::Constant(&graph_0, a_tensor, "A");
367 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2}));
369 test::FillValues<float>(&a_tensor, {3, 2, -1, 0});
370 Node* a_node = test::graph::Constant(&graph, a_tensor);
|
| message_wrappers_test.cc | 29 Tensor a_tensor(DT_INT32, TensorShape({2, 2}));
30 test::FillValues<int32>(&a_tensor, {3, 2, -1, 0});
31 return a_tensor;
|
| /external/tensorflow/tensorflow/python/training/ |
| monitored_session_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| ops.py | 851 a_tensor = array_ops.reshape(a.tensor, (1, -1))
855 a_tensor = a.tensor
867 a_tensor, b_tensor, transpose_a=transpose_a, transpose_b=transpose_b)
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| monitors_test.py | [all...] |
