Searched refs:GetOutputTensor (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | slice_test.cc | 80 const TfLiteTensor* GetOutputTensor() { in GetOutputTensor() function in tflite::__anon871d8d3d0111::SliceOpModel
290 EXPECT_NE(m.GetOutputTensor()->allocation_type, kTfLiteDynamic); in TEST_P()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | object_reader.h | 133 TfLiteTensor* GetOutputTensor(int index) const;
|
| D | object_reader.cc | 230 TfLiteTensor* ObjectReader::GetOutputTensor(int index) const { in GetOutputTensor() function in tflite::gpu::ObjectReader
|
| D | model_builder.cc | 672 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.h | 133 absl::Status GetOutputTensor(ValueId id, TensorFloat32* result);
|
| D | inference_context.cc | 353 absl::Status InferenceContext::GetOutputTensor(ValueId id, in GetOutputTensor() function in tflite::gpu::metal::InferenceContext
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/ |
| D | performance_profiling.cc | 180 RETURN_IF_ERROR(context.GetOutputTensor(graph_cl.outputs()[i]->id, in RunSerializedTest()
206 RETURN_IF_ERROR(serialized_context.GetOutputTensor( in RunSerializedTest()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | inference_context.h | 99 absl::Status GetOutputTensor(ValueId id, CLCommandQueue* queue,
|
| D | inference_context.cc | 911 absl::Status InferenceContext::GetOutputTensor(ValueId id, in GetOutputTensor() function in tflite::gpu::cl::InferenceContext
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/benchmarking/ |
| D | main.mm | 83 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]));
|
| /external/armnn/src/armnn/ |
| D | LoadedNetwork.cpp | 1405 const armnn::Tensor GetOutputTensor(const LayerBindingId layerId, const OutputTensors& outputTensor… in GetOutputTensor() function
|