message: Additional checks to exclude potential errors.
diff: @@ -30,6 +30,9 @@ Status ExecuteGPUOperation(const std::vector<TensorFloat32>& src_cpu,
  std::vector<Tensor> src(src_cpu.size());
  for (int i = 0; i < src_cpu.size(); ++i) {
  auto src_shape = src_cpu[i].shape;
  + if (src_shape.b != 1 && !op_def.batch_support) {
  + return InvalidArgumentError("op_def.batch_support must be enabled");
  + }
  RETURN_IF_ERROR(CreateTensor(*creation_context.context,
  *creation_context.device, src_shape,
  op_def.src_tensors[0], &src[i]));
  @@ -40,6 +43,9 @@ Status ExecuteGPUOperation(const std::vector<TensorFloat32>& src_cpu,
  std::vector<Tensor> dst(dst_cpu.size());
  for (int i = 0; i < dst_cpu.size(); ++i) {
  auto dst_shape = dst_sizes[i];
  + if (dst_shape.b != 1 && !op_def.batch_support) {
  + return InvalidArgumentError("op_def.batch_support must be enabled");
  + }
  RETURN_IF_ERROR(CreateTensor(*creation_context.context,
  *creation_context.device, dst_shape,
  op_def.dst_tensors[0], &dst[i]));
  