Home
last modified time | relevance | path

Searched defs:a_tensor (Results 1 – 14 of 14) 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
170 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TestHWAccelerator() local
253 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
316 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST() local
Ddirect_session_test.cc98 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
679 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in TEST_F() local
1002 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/kernels/
Dtensor_list_util.cc58 const Tensor& a_tensor = a.tensors()[i]; in TensorListBinaryAdd() 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_util.cc69 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/backends/gpu/
Ddevice_tracer_test.cc87 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
/external/tensorflow/tensorflow/core/framework/
Drun_handler_test.cc601 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/
Dimpl.cpp222 const auto a_tensor = is_broadcast_input_1 ? src0 : src1; in add_sub_q8_neon_fixedpoint() local
/external/ComputeLibrary/src/cpu/kernels/
DCpuMulKernel.cpp376 const auto a_tensor = is_broadcast_input_1 ? src0 : src1; in mul_q8_neon_fixedpoint() local