}

TEST_F(CApiFunctionTest, TFGraphToFunctionWithStackTraces) {
  DefineFunction(func_name_, &func_);
  auto stack_traces = func_->record->stack_traces();

  EXPECT_EQ(stack_traces.size(), 4);
  EXPECT_EQ(stack_traces["neg"]->ToString({}), kNegStackToString);
  EXPECT_EQ(stack_traces["feed"]->ToString({}), kFeedStackToString);
}

  TF_Graph* graph_;
  TF_Graph* expected_graph_;
};

TEST_F(CApiGradientsTest, Gradients_GradInputs) { TestGradientsSuccess(true); }

TEST_F(CApiGradientsTest, Gradients_NoGradInputs) {
  TestGradientsSuccess(false);
}

TEST_F(CApiGradientsTest, OpWithNoGradientRegistered_GradInputs) {
  TestGradientsError(true);
}

TEST_F(CApiGradientsTest, OpWithNoGradientRegistered_NoGradInputs) {