Home
last modified time | relevance | path

Searched defs:a_tensor (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_with_tracking_alloc_test.cc51 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
169 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TestHWAccelerator() local
260 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
322 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
Ddirect_session_test.cc87 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
531 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
854 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
/external/tensorflow/tensorflow/core/debug/
Dgrpc_session_debug_test.cc55 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
232 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST_F() local
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc47 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
296 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in FindMaxEigen() local
384 Tensor a_tensor(DT_FLOAT, TensorShape({1, kSize})); in TEST() local
502 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST() local
560 Tensor a_tensor(DT_FLOAT, TensorShape({1, 1})); in TEST() local
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.h144 const Tensor& a_tensor = a.get_values()[i]; in OptionalBinaryAdd() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc295 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
367 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
Dmessage_wrappers_test.cc29 Tensor a_tensor(DT_INT32, TensorShape({2, 2})); in TensorA() local
/external/tensorflow/tensorflow/core/platform/
Ddevice_tracer_test.cc58 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc213 Tensor a_tensor(DT_FLOAT, TensorShape({m, k})); in TestQuantizeMatMul() local
376 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2, 3})); in TestQuantizeConcat() local
1235 Tensor a_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicateConsts() local
1280 Tensor a_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesNested() local
1330 Tensor a_tensor(DT_FLOAT, TensorShape({width})); in TestMergeDuplicatesInOut() local
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h885 const Tensor& a_tensor = a.tensors[i]; in TensorListBinaryAdd() local