TF_DeleteStatus(status);
}

TEST(CAPI, TestBitcastFrom_Reshape) {
  int64_t dims[] = {2, 3};
  TF_Tensor* a =
      TF_AllocateTensor(TF_UINT64, dims, 2, 6 * TF_DataTypeSize(TF_UINT64));
  TF_Tensor* b =
      TF_AllocateTensor(TF_UINT64, nullptr, 0, TF_DataTypeSize(TF_UINT64));
  EXPECT_NE(a, nullptr);
  EXPECT_NE(b, nullptr);

  EXPECT_EQ(6, TF_TensorElementCount(a));

  std::vector<int64_t> dims(4, 1);
  TFE_Op* op = TFE_NewOp(ctx, "AvgPool", status);
  ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);

  TF_Tensor* tensor =
      TF_AllocateTensor(TF_FLOAT, dims.data(), dims.size(), sizeof(float));
  float tensor_data[] = {1};
  memcpy(TF_TensorData(tensor), tensor_data, TF_TensorByteSize(tensor));