EXPECT_EQ(6 * TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(a));
  EXPECT_EQ(TF_DataTypeSize(TF_UINT64), TF_TensorByteSize(b));

  int64_t new_dims[] = {3, 2};
  TF_Status* status = TF_NewStatus();
  TF_TensorBitcastFrom(a, TF_UINT64, b, new_dims, 2, status);
  ASSERT_EQ(TF_OK, TF_GetCode(status));
  TF_DeleteStatus(status);

  EXPECT_EQ(6, TF_TensorElementCount(a));
  EXPECT_EQ(6, TF_TensorElementCount(b));