Home
last modified time | relevance | path

Searched refs:GetOutputTensor (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dslice_test.cc80 const TfLiteTensor* GetOutputTensor() { in GetOutputTensor() function in tflite::__anon0aaaf30e0111::SliceOpModel
290 EXPECT_NE(m.GetOutputTensor()->allocation_type, kTfLiteDynamic); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/
H A Dmain.mm83 RETURN_IF_ERROR(gpu_context->GetOutputTensor(outputs[i]->id, &gpu_tensor));
313 RETURN_IF_ERROR(inference_context.GetOutputTensor(graph->outputs()[i]->id, &dst_tensors[i]));
342 serialized_context.GetOutputTensor(graph->outputs()[i]->id, &dst_tensors_v2[i]));
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
H A Dperformance_profiling.cc180 RETURN_IF_ERROR(context.GetOutputTensor(graph_cl.outputs()[i]->id, in RunSerializedTest()
206 RETURN_IF_ERROR(serialized_context.GetOutputTensor( in RunSerializedTest()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dobject_reader.h133 TfLiteTensor* GetOutputTensor(int index) const;
H A Dobject_reader.cc230 TfLiteTensor* ObjectReader::GetOutputTensor(int index) const { in GetOutputTensor() function in tflite::gpu::ObjectReader
H A Dmodel_builder.cc672 const TfLiteTensor* dst_tensor = reader->GetOutputTensor(0); in ResolveGroupedConvolution()
864 const TfLiteTensor* output = reader->GetOutputTensor(0); in Parse()
1222 const TfLiteTensor* output_tensor = reader->GetOutputTensor(0); in Parse()
1530 const TfLiteTensor* output = reader->GetOutputTensor(0); in Parse()
1763 const TfLiteTensor* output = reader->GetOutputTensor(0); in Parse()
2714 const TfLiteTensor* output = reader->GetOutputTensor(i); in Parse()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.h133 absl::Status GetOutputTensor(ValueId id, TensorFloat32* result);
H A Dinference_context.cc353 absl::Status InferenceContext::GetOutputTensor(ValueId id, in GetOutputTensor() function in tflite::gpu::metal::InferenceContext
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.h99 absl::Status GetOutputTensor(ValueId id, CLCommandQueue* queue,
H A Dinference_context.cc911 absl::Status InferenceContext::GetOutputTensor(ValueId id, in GetOutputTensor() function in tflite::gpu::cl::InferenceContext
/aosp_15_r20/external/armnn/src/armnn/
H A DLoadedNetwork.cpp1405 const armnn::Tensor GetOutputTensor(const LayerBindingId layerId, const OutputTensors& outputTensor… in GetOutputTensor() function