Method GetOutputTensor

tensorflow/lite/delegates/gpu/cl/inference_context.cc:393–402

}

Status InferenceContext::GetOutputTensor(ValueId id, CLCommandQueue* queue,
                                         TensorFloat32* result) {
  const auto& gpu_tensor = *GetTensor(id);
  const int4 dst_size = gpu_tensor.GetSizeWithDepth();
  const auto dst_shape = BHWC(1, dst_size.y, dst_size.x, dst_size.z);
  result->id = id;
  result->shape = dst_shape;
  result->data.resize(dst_shape.DimensionsProduct());
  return gpu_tensor.ReadData(queue, result);
}

Status RunGraphTransforms(GraphFloat32* graph) {
  auto merge_padding_transform = NewMergePaddingWithAdd();