Home
last modified time | relevance | path

Searched defs:tensor_ptr (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
Dinternal_api_samples.cc39 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
325 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local
332 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local
[all …]
Ddelegate_testing.cc35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.h115 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
131 if (TfLiteTensor* tensor_ptr = input_tensor(tensor_index)) { in typed_input_tensor() local
147 if (TfLiteTensor* tensor_ptr = output_tensor(tensor_index)) { in typed_output_tensor() local
/external/tensorflow/tensorflow/core/kernels/
Dslice_op.cc44 const int32* tensor_ptr = tensor.flat<int32>().data(); in IntTensorToInt64Vec() local
49 const int64* tensor_ptr = tensor.flat<int64>().data(); in IntTensorToInt64Vec() local
Dquantized_add_op.cc351 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtask_utils.h41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor()
/external/tensorflow/tensorflow/lite/
Dinterpreter.h270 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
282 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse_test.cc71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
Dtest_util.cc286 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.cc101 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.cc98 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc218 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_ops_test.cc655 TfLiteTensor* tensor_ptr = interpreter_->tensor(output_); in GetOutput() local