ASSERT_EQ(errors::OK, s.code()) << s.message();
    x.reset(x_raw);
  }

  std::vector<AbstractTensorHandle*> outputs(1);
  absl::Status s = RunModel(RecordOperationWithNullGradientFunctionModel,
                            ctx.get(), {x.get()}, absl::MakeSpan(outputs),
                            /*use_function=*/!std::get<2>(GetParam()));

                            absl::Span<AbstractTensorHandle*> outputs,
                            bool use_function) {
  AbstractTensorHandle* model_outputs[1];

  // Run the model.
  TF_RETURN_IF_ERROR(
      RunModel(forward, ctx, inputs, model_outputs, use_function));
  AbstractTensorHandlePtr model_out(model_outputs[0]);

  TF_Tensor* model_out_tensor;
  TF_RETURN_IF_ERROR(GetValue(model_out.get(), &model_out_tensor));