| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_with_tracking_alloc_test.cc | 51 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
52 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST()
53 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
169 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TestHWAccelerator() local
170 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TestHWAccelerator()
171 Node* a = test::graph::Constant(&graph, a_tensor); in TestHWAccelerator()
260 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
261 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST()
262 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
322 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
[all …]
|
| D | direct_session_test.cc | 87 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
88 test::FillValues<float>(&a_tensor, a_values); in Initialize()
89 Node* a = test::graph::Constant(&graph, a_tensor); in Initialize()
531 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
532 a_tensor.flat<float>().setRandom(); in TEST_F()
533 Node* a = test::graph::Constant(&graph, a_tensor); in TEST_F()
854 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
855 test::FillValues<float>(&a_tensor, {1.0, 2.0, 3.0, 4.0}); in TEST()
856 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 55 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
56 test::FillValues<float>(&a_tensor, {1.0, 2.0}); in CreateGraphDef()
57 Node* a = test::graph::Constant(&graph, a_tensor); in CreateGraphDef()
232 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST_F() local
234 a_tensor.flat<string>()(i) = "hello, world"; in TEST_F()
236 Node* a = test::graph::Constant(&graph, a_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session_test.cc | 47 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
48 test::FillValues<float>(&a_tensor, {1, 2}); in CreateGraphDef()
49 Node* a = test::graph::Constant(&graph, a_tensor); in CreateGraphDef()
296 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in FindMaxEigen() local
298 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in FindMaxEigen()
299 Node* a = test::graph::Constant(&graph, a_tensor); in FindMaxEigen()
384 Tensor a_tensor(DT_FLOAT, TensorShape({1, kSize})); in TEST() local
387 a_tensor.flat<float>()(i) = 2; in TEST()
390 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
502 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 295 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
296 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST_F()
297 test::graph::Constant(&graph_0, a_tensor, "A"); in TEST_F()
367 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
369 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST_F()
370 Node* a_node = test::graph::Constant(&graph, a_tensor); in TEST_F()
|
| D | message_wrappers_test.cc | 29 Tensor a_tensor(DT_INT32, TensorShape({2, 2})); in TensorA() local
30 test::FillValues<int32>(&a_tensor, {3, 2, -1, 0}); in TensorA()
31 return a_tensor; in TensorA()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 213 Tensor a_tensor(DT_FLOAT, TensorShape({m, k})); in TestQuantizeMatMul() local
214 test::FillValues<float>(&a_tensor, a_values); in TestQuantizeMatMul()
215 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); in TestQuantizeMatMul()
376 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2, 3})); in TestQuantizeConcat() local
377 test::FillValues<float>(&a_tensor, {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12}); in TestQuantizeConcat()
378 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); in TestQuantizeConcat()
1235 Tensor a_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicateConsts() local
1236 test::FillIota<float>(&a_tensor, 1.0f); in TestMergeDuplicateConsts()
1237 Output a_op = Const(root.WithOpName("a_op"), Input::Initializer(a_tensor)); in TestMergeDuplicateConsts()
1280 Tensor a_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesNested() local
[all …]
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | device_tracer_test.cc | 58 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
59 test::FillValues<float>(&a_tensor, a_values); in Initialize()
60 Node* a = test::graph::Constant(&graph, a_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1225 a_tensor = constant_op.constant([0], name='a_tensor')
1227 output = mon_sess.run(fetches=a_tensor,
1244 a_tensor = constant_op.constant([0], name='a_tensor')
1246 mon_sess.run(a_tensor)
1256 session_run_hook.SessionRunArgs(a_tensor))
1285 a_tensor = constant_op.constant([0], name='a_tensor')
1292 output = mon_sess.run(fetches=a_tensor)
1303 a_tensor = constant_op.constant([0], name='a_tensor')
1305 add_tensor = a_tensor + b_tensor
1307 None, feed_dict={a_tensor: [5]})
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.h | 144 const Tensor& a_tensor = a.get_values()[i]; in OptionalBinaryAdd() local
148 BinaryAddTensors<Device>(ctx, a_tensor, b_tensor, &out_tensor)); in OptionalBinaryAdd()
|
| /external/tensorflow/tensorflow/contrib/labeled_tensor/python/ops/ |
| D | ops.py | 853 a_tensor = array_ops.reshape(a.tensor, (1, -1))
857 a_tensor = a.tensor
869 a_tensor, b_tensor, transpose_a=transpose_a, transpose_b=transpose_b)
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/ |
| D | monitors_test.py | 837 a_tensor = constant_op.constant([0], name='a_tensor')
844 output = mon_sess.run(a_tensor)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 885 const Tensor& a_tensor = a.tensors[i]; in TensorListBinaryAdd() local
889 BinaryAddTensors<Device>(c, a_tensor, b_tensor, &out_tensor)); in TensorListBinaryAdd()
|