TF_CHECK_OK(MessageToBuffer(node_in, buffer));
  TF_CHECK_OK(BufferToMessage(buffer, &node_out));
  TF_DeleteBuffer(buffer);

  protobuf::util::MessageDifferencer differencer;
  EXPECT_TRUE(differencer.Compare(node_in, node_out));
}

TEST(CAPI, TestTensorNonScalarBytesAllocateDelete) {
  const int batch_size = 4;
  const int num_dims = 2;
  int64_t* dims = new int64_t[num_dims];

  CHECK(tensorflow::unwrap(concatOp)->OpDef());
  TFE_OpAddInput(concatOp, inputs[0], status);
  CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
  EXPECT_FALSE(tensorflow::unwrap(concatOp)->OpDef())
      << "Inference context is still present";
  TFE_OpAddInput(concatOp, inputs[1], status);
  CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);

  tensorflow::AttrValueMap attr_values = ExtractAttrs(concatOp);

  ASSERT_NE(*func, nullptr);
}

REGISTER_OP("CustomOp")
    .Output("output: float32")
    .Attr("index: int")
    .SetShapeFn(tensorflow::shape_inference::UnknownShape);

void NodeWithPlaceholderAttrHelper(TF_Graph* graph, TF_Status* s,
                                   const char* name, const char* placeholder,
                                   TF_Operation** op) {

        "//tensorflow_federated/...",
        "//tensorflow_models/google/recml/...",
        "//third_party/cloud_tpu/convergence_tools/sdc_monitoring/...",
        "//third_party/cloud_tpu/inference_converter/...",
        "//third_party/py/cloud_ml_autoflow/...",
        "//third_party/py/envlogger/...",
        "//third_party/py/gldm/...",
        "//third_party/py/guesslang/...",

TF_CAPI_EXPORT extern void TF_RemoveAllControlInputs(TF_Graph* graph,
                                                     TF_Operation* op);

// Set if `graph` requires shape inference functions.
TF_CAPI_EXPORT extern void TF_SetRequireShapeInferenceFns(TF_Graph* graph,
                                                          bool require);