Method BindArguments

tensorflow/lite/delegates/gpu/cl/kernels/gpu_operation.cc:125–133

}

Status ElementwiseOperation::BindArguments() {
  kernel_.ResetBindingCounter();
  RETURN_IF_ERROR(kernel_.SetMemoryAuto(src_[0]->GetMemoryPtr()));
  RETURN_IF_ERROR(BindArguments(&kernel_));
  RETURN_IF_ERROR(BindArgs(&kernel_, linked_operations_));
  RETURN_IF_ERROR(kernel_.SetMemoryAuto(dst_[0]->GetMemoryPtr()));
  RETURN_IF_ERROR(kernel_.SetBytesAuto(dst_[0]->GetSizeWithDepth()));
  return OkStatus();
}

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