| /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()
251 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
252 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST()
253 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
313 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
[all …]
|
| D | direct_session_test.cc | 97 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
98 test::FillValues<float>(&a_tensor, a_values); in Initialize()
99 Node* a = test::graph::Constant(&graph, a_tensor); in Initialize()
678 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
679 a_tensor.flat<float>().setRandom(); in TEST_F()
680 Node* a = test::graph::Constant(&graph, a_tensor); in TEST_F()
1001 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
1002 test::FillValues<float>(&a_tensor, {1.0, 2.0, 3.0, 4.0}); in TEST()
1003 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<tstring>()(i) = "hello, world"; in TEST_F()
236 Node* a = test::graph::Constant(&graph, a_tensor); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | min_max_builder.cc | 28 const auto& a_tensor = context->tensors[a_tensor_id]; in PopulateSubGraph() local
34 TF_LITE_ENSURE_STATUS(ComputeAndAddMinAndMax(context, a_tensor)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 296 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
297 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST_F()
298 test::graph::Constant(&graph_0, a_tensor, "A"); in TEST_F()
368 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
370 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in TEST_F()
371 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/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()
331 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in FindMaxEigen() local
333 test::FillValues<float>(&a_tensor, {3, 2, -1, 0}); in FindMaxEigen()
334 Node* a = test::graph::Constant(&graph, a_tensor); in FindMaxEigen()
419 Tensor a_tensor(DT_FLOAT, TensorShape({1, kSize})); in TEST() local
422 a_tensor.flat<float>()(i) = 2; in TEST()
425 Node* a = test::graph::Constant(&graph, a_tensor); in TEST()
537 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/python/training/ |
| D | monitored_session_test.py | 1271 a_tensor = constant_op.constant([0], name='a_tensor')
1273 output = mon_sess.run(fetches=a_tensor,
1290 a_tensor = constant_op.constant([0], name='a_tensor')
1292 mon_sess.run(a_tensor)
1302 session_run_hook.SessionRunArgs(a_tensor))
1331 a_tensor = constant_op.constant([0], name='a_tensor')
1338 output = mon_sess.run(fetches=a_tensor)
1349 a_tensor = constant_op.constant([0], name='a_tensor')
1351 add_tensor = a_tensor + b_tensor
1353 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/core/profiler/internal/gpu/ |
| D | device_tracer_test.cc | 80 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
81 test::FillValues<float>(&a_tensor, a_values); in Initialize()
82 Node* a = test::graph::HostConstant(&graph, a_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | run_handler_test.cc | 601 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
602 test::FillValues<float>(&a_tensor, a_values); in Initialize()
603 Node* a = test::graph::Constant(&graph, a_tensor); in Initialize()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.h | 862 const Tensor& a_tensor = a.tensors()[i]; in TensorListBinaryAdd() local
866 BinaryAddTensors<Device>(c, a_tensor, b_tensor, &out_tensor)); in TensorListBinaryAdd()
|