| /external/ComputeLibrary/src/core/GLES_COMPUTE/cs_shaders/ |
| D | helpers_cs.h | 285 #define LOAD(tensor_ptr, offset) tensor_ptr[offset] argument
286 #define STORE(tensor_ptr, offset, data) tensor_ptr[offset] = data argument
287 #define LOAD_CURRENT_ITEM(tensor_ptr, tensor_iter) tensor_ptr[CURRENT_ITEM_OFFSET(tensor_iter)] argument
288 #define STORE_CURRENT_ITEM(tensor_ptr, tensor_iter, data) tensor_ptr[CURRENT_ITEM_OFFSET(tensor_ite… argument
290 #define VLOAD2(return_type, tensor_ptr, offset) \ argument
294 #define VSTORE2(tensor_ptr, offset, data) \ argument
298 #define VLOAD2_CURRENT_ITEM(return_type, tensor_ptr, tensor_iter) VLOAD2(return_type, tensor_ptr, C… argument
299 #define VSTORE2_CURRENT_ITEM(tensor_ptr, tensor_iter, data) VSTORE2(tensor_ptr, CURRENT_ITEM_OFFSET… argument
301 #define VLOAD3(return_type, tensor_ptr, offset) \ argument
306 #define VSTORE3(tensor_ptr, offset, data) \ argument
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | internal_api_samples.cc | 39 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensors() local
58 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]); in CompareCPUGPUResults() local
115 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
123 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
201 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPI() local
208 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPI() local
239 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local
247 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local
326 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local
333 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local
[all …]
|
| D | delegate_testing.cc | 35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local
67 TfLiteTensor* tensor_ptr = cpu->tensor(cpu->outputs()[i]); in CompareCPUGPUResults() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 44 const int32* tensor_ptr = tensor.flat<int32>().data(); in IntTensorToInt64Vec() local
49 const int64* tensor_ptr = tensor.flat<int64>().data(); in IntTensorToInt64Vec() local
|
| D | quantized_add_op.cc | 351 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | task_utils.h | 41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 277 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
289 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
|
| D | test_util.cc | 312 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | parse_testdata.cc | 101 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | parse_testdata.cc | 98 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 271 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress()
|