Searched refs:GetOutputTensor (Results 1 – 5 of 5) sorted by relevance
96 TfLiteTensor* GetOutputTensor(int index) const;
217 TfLiteTensor* ObjectReader::GetOutputTensor(int index) const { in GetOutputTensor() function in tflite::gpu::ObjectReader
498 const TfLiteTensor* output = reader->GetOutputTensor(0); in Parse()1195 const TfLiteTensor* output = reader->GetOutputTensor(0); in Parse()
92 absl::Status GetOutputTensor(ValueId id, CLCommandQueue* queue,
719 absl::Status InferenceContext::GetOutputTensor(ValueId id, in GetOutputTensor() function in tflite::gpu::cl::InferenceContext