| /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
40 const auto tensor_elements_count = tflite::NumElements(tensor_ptr); in FillInputTensors()
41 if (tensor_ptr->type == kTfLiteFloat32) { in FillInputTensors()
58 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]); in CompareCPUGPUResults() local
59 const float* cpu_out = tensor_ptr->data.f; in CompareCPUGPUResults()
64 for (int k = 0; k < tensor_ptr->bytes / 4; ++k) { in CompareCPUGPUResults()
115 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
117 if (tensor_ptr->type != kTfLiteFloat32) { in RunModelSampleWithInternalAPI()
123 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local
125 if (tensor_ptr->type != kTfLiteFloat32) { in RunModelSampleWithInternalAPI()
[all …]
|
| D | delegate_testing.cc | 35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local
36 const auto tensor_elements_count = tflite::NumElements(tensor_ptr); in FillInputTensor()
37 if (tensor_ptr->type == kTfLiteFloat32) { in FillInputTensor()
43 if (tensor_ptr->type == kTfLiteInt32) { in FillInputTensor()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_interpreter.h | 115 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
116 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
117 return GetTensorData<T>(tensor_ptr); in typed_tensor()
131 if (TfLiteTensor* tensor_ptr = input_tensor(tensor_index)) { in typed_input_tensor() local
132 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_input_tensor()
133 return GetTensorData<T>(tensor_ptr); in typed_input_tensor()
147 if (TfLiteTensor* tensor_ptr = output_tensor(tensor_index)) { in typed_output_tensor() local
148 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_output_tensor()
149 return GetTensorData<T>(tensor_ptr); in typed_output_tensor()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 270 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
271 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
272 return reinterpret_cast<T*>(tensor_ptr->data.raw); in typed_tensor()
282 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
283 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
284 return reinterpret_cast<const T*>(tensor_ptr->data.raw); in typed_tensor()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | task_utils.h | 41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor() argument
42 if (tensor_ptr->type == typeToTfLiteType<T>()) { in TypedTensor()
43 return reinterpret_cast<T*>(tensor_ptr->data.raw); in TypedTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | slice_op.cc | 44 const int32* tensor_ptr = tensor.flat<int32>().data(); in IntTensorToInt64Vec() local
46 out_ptr[i] = tensor_ptr[i]; in IntTensorToInt64Vec()
49 const int64* tensor_ptr = tensor.flat<int64>().data(); in IntTensorToInt64Vec() local
51 out_ptr[i] = tensor_ptr[i]; in IntTensorToInt64Vec()
|
| D | quantized_add_op.cc | 351 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
353 Requantize8x8To32Neon(tensor_ptr, tensor_0_64x2, tensor_mult_32x2); in VectorTensorAddition()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | embedding_lookup_sparse_test.cc | 71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
75 tensor_ptr[(i * columns + j) * features + k] = function(i, j, k); in Set3DWeightMatrix()
|
| D | test_util.cc | 286 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
287 CHECK(tensor_ptr != nullptr); in ExtractVector()
288 const int num_strings = GetStringCount(tensor_ptr); in ExtractVector()
292 const auto str = GetString(tensor_ptr, i); in ExtractVector()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 218 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress() argument
219 auto tensor_type = tensor_ptr->dtype(); in GetTensorAddress()
221 TYPECASE(DT_FLOAT, tensor_ptr, dest_ptr); in GetTensorAddress()
222 TYPECASE(DT_HALF, tensor_ptr, dest_ptr); in GetTensorAddress()
223 TYPECASE(DT_INT8, tensor_ptr, dest_ptr); in GetTensorAddress()
224 TYPECASE(DT_INT32, tensor_ptr, dest_ptr); in GetTensorAddress()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | parse_testdata.cc | 98 &csv](FloatTensor* tensor_ptr) { in ParseExamples() argument
104 FloatTensor& tensor = *tensor_ptr; in ParseExamples()
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | parse_testdata.cc | 101 &csv](FloatTensor* tensor_ptr) { in ParseExamples() argument
107 FloatTensor& tensor = *tensor_ptr; in ParseExamples()
|
| /external/tensorflow/tensorflow/lite/kernels/hashtable/ |
| D | hashtable_ops_test.cc | 655 TfLiteTensor* tensor_ptr = interpreter_->tensor(output_); in GetOutput() local
656 return std::vector<int>(tensor_ptr->data.i32, tensor_ptr->data.i32 + 1); in GetOutput()
|