Home
last modified time | relevance | path

Searched defs:a_tensor (Results 1 – 12 of 12) 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
251 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
313 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
Ddirect_session_test.cc97 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
678 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
1001 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmin_max_builder.cc28 const auto& a_tensor = context->tensors[a_tensor_id]; in PopulateSubGraph() 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/kernels/data/
Doptional_ops.h144 const Tensor& a_tensor = a.get_values()[i]; in OptionalBinaryAdd() local
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc47 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
331 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in FindMaxEigen() local
419 Tensor a_tensor(DT_FLOAT, TensorShape({1, kSize})); in TEST() local
537 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST() local
595 Tensor a_tensor(DT_FLOAT, TensorShape({1, 1})); in TEST() local
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_test.cc296 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
368 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/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/profiler/internal/gpu/
Ddevice_tracer_test.cc80 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
/external/tensorflow/tensorflow/core/kernels/
Dlist_kernels.h862 const Tensor& a_tensor = a.tensors()[i]; in TensorListBinaryAdd() local
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc601 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local