Searched defs:tensor_ptr (Results 1 – 14 of 14) sorted by relevance
39 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensors() local58 TfLiteTensor* tensor_ptr = cpu->tensor(outputs[i]); in CompareCPUGPUResults() local115 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local123 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPI() local201 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPI() local208 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPI() local239 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local247 TfLiteTensor* tensor_ptr = in RunModelSampleWithInternalAPISerializedKernels() local326 TfLiteTensor* tensor_ptr = cpu_inference->tensor(in_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local333 TfLiteTensor* tensor_ptr = cpu_inference->tensor(out_refs[i]); in RunModelSampleWithInternalAPISerializedKernels() local[all …]
35 TfLiteTensor* tensor_ptr = interpreter->tensor(interpreter->inputs()[k]); in FillInputTensor() local67 TfLiteTensor* tensor_ptr = cpu->tensor(cpu->outputs()[i]); in CompareCPUGPUResults() local
44 const int32* tensor_ptr = tensor.flat<int32>().data(); in IntTensorToInt64Vec() local49 const int64_t* tensor_ptr = tensor.flat<int64_t>().data(); in IntTensorToInt64Vec() local
352 const uint8* tensor_ptr = &(tensor_data->value) + i; in VectorTensorAddition() local
41 T* TypedTensor(const TfLiteTensor* tensor_ptr) { in TypedTensor()
278 if (TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local290 if (const TfLiteTensor* tensor_ptr = tensor(tensor_index)) { in typed_tensor() local
71 float* tensor_ptr = GetTensorData<float>(tensor); in Set3DWeightMatrix() local
320 TfLiteTensor* tensor_ptr = interpreter_->tensor(index); in ExtractVector() local
101 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
103 &csv](FloatTensor* tensor_ptr) { in ParseExamples()
772 MetalSpatialTensor* tensor_ptr) { in SetTensor()
283 void* GetTensorAddress(const Tensor* tensor_ptr) { in GetTensorAddress()
700 Tensor* tensor_ptr) { in SetTensor()
2275 Tensor* tensor_ptr = context->mutable_output(0); in TestConvertVariableV2() local