return absl::InvalidArgumentError(absl::StrCat(
        "Provided null gradient_function for '", op_type, "'.\n",
        "If the intent is to treat this op as non-differentiable consider ",
        "using RegisterNotDifferentiable or ",
        "NotDifferentiableGradientFunction."));
  }
  return gradient_function->Compute(ctx_, output_gradients, result);
}

absl::Status TapeVSpace::BuildOnesLike(const TapeTensor& t,