| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | direct_session_with_tracking_alloc_test.cc | 51 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
|
| H A D | direct_session_test.cc | 98 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session_test.cc | 47 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | grpc_session_debug_test.cc | 55 Tensor a_tensor(DT_FLOAT, TensorShape({1, 2})); in CreateGraphDef() local
232 Tensor a_tensor(DT_STRING, TensorShape({2, 2})); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | quantize_nodes_test.cc | 213 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | master_test.cc | 296 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
|
| H A D | message_wrappers_test.cc | 29 Tensor a_tensor(DT_INT32, TensorShape({2, 2})); in TensorA() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | min_max_builder.cc | 28 const auto& a_tensor = context->tensors[a_tensor_id]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | tensor_list_util.cc | 58 const Tensor& a_tensor = a.tensors()[i]; in TensorListBinaryAdd() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | optional_ops_util.cc | 69 const Tensor& a_tensor = a.get_values()[i]; in OptionalBinaryAdd() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | atest.cpp | 117 auto a_tensor = tensor({a}); in TEST_F() local
|
| /aosp_15_r20/external/executorch/kernels/test/ |
| H A D | op_where_test.cpp | 56 …const auto a_tensor = tf_a.make(sizes, /*data=*/{ 1, 2, 3, 4, 5, 6, 6, 5, 4, 3, 2, 1}); in test_where() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | device_tracer_test.cc | 87 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | run_handler_test.cc | 601 Tensor a_tensor(DT_FLOAT, TensorShape({2, 2})); in Initialize() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/add/generic/neon/ |
| H A D | impl.cpp | 222 const auto a_tensor = is_broadcast_input_1 ? src0 : src1; in add_sub_q8_neon_fixedpoint() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuMulKernel.cpp | 376 const auto a_tensor = is_broadcast_input_1 ? src0 : src1; in mul_q8_neon_fixedpoint() local
|