Method BindArguments

tensorflow/lite/delegates/gpu/cl/kernels/reshape.cc:104–116

}

Status Reshape::BindArguments() {
  kernel_.ResetBindingCounter();
  RETURN_IF_ERROR(kernel_.SetMemoryAuto(src_[0]->GetMemoryPtr()));
  RETURN_IF_ERROR(BindArgs(&kernel_, linked_operations_));
  RETURN_IF_ERROR(kernel_.SetMemoryAuto(dst_[0]->GetMemoryPtr()));
  RETURN_IF_ERROR(kernel_.SetBytesAuto(src_[0]->GetSizeWithDepth()));
  RETURN_IF_ERROR(kernel_.SetBytesAuto(dst_[0]->GetSizeWithDepth()));
  const int2 plane_size = int2(src_[0]->Width() * src_[0]->Channels(),
                               dst_[0]->Width() * dst_[0]->Channels());
  RETURN_IF_ERROR(kernel_.SetBytesAuto(plane_size));

  return OkStatus();
}

int3 Reshape::GetGridSize() const {
  const int grid_x = dst_[0]->Width();