Searched defs:ref_tensor (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| H A D | cast_test_util.cc | 32 tflite::gpu::Tensor<BHWC, DataType::INT32> ref_tensor; in CastTests() local
67 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in CastToBoolTests() local
102 tflite::gpu::Tensor<BHWC, DataType::FLOAT32> ref_tensor; in CastFromBoolTests() local
|
| H A D | elementwise_test_util.cc | 993 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in LessTest() local
1034 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in LessEqualTest() local
1073 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in GreaterTest() local
1112 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in GreaterEqualTest() local
1151 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in EqualTest() local
1190 tflite::gpu::Tensor<BHWC, DataType::BOOL> ref_tensor; in NotEqualTest() local
|
| H A D | transpose_test_util.cc | 38 tflite::gpu::Tensor<BHWC, T> ref_tensor; in TransposeIntTest() local
80 tflite::gpu::Tensor<BHWC, T> ref_tensor; in TransposeUintTest() local
|
| H A D | add_test_util.cc | 36 tflite::gpu::Tensor<BHWC, T> ref_tensor; in AddTwoEqualIntTensorsTest() local
80 tflite::gpu::Tensor<BHWC, T> ref_tensor; in AddTwoEqualUintTensorsTest() local
|
| H A D | reduce_test_util.cc | 37 tflite::gpu::Tensor<BHWC, T> ref_tensor; in ReduceSumChannelsIntTest() local
77 tflite::gpu::Tensor<BHWC, T> ref_tensor; in ReduceProductChannelsUIntTest() local
|
| H A D | pooling_test_util.cc | 187 tflite::gpu::Tensor<BHWC, DataType::INT32> ref_tensor; in MaxPoolingIndicesTest() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | entry.h | 126 } ref_tensor; member
|
| H A D | executor.cc | 995 Tensor* ref_tensor = entry->ref_tensor.tensor; in PrepareInputs() local
|