EXPECT_EQ(1.0, *(static_cast<float*>(TF_TensorData(t_c))));
  TF_DeleteStatus(status);
  TF_DeleteTensor(t_c);
  TF_DeleteBuffer(t_buffer);
}

REGISTER_OP("TestOpWithNoGradient")
    .Input("x: T")
    .Output("y: T")
    .Attr("T: {float, double}")
    .Doc(R"doc(
Test op with no grad registered.

x: input
y: output
)doc")