Function TEST

tensorflow/c/eager/c_api_debug_test.cc:23–34

#include "tensorflow/core/platform/test.h"

TEST(CApiDebug, ScalarCPU) {
  TFE_TensorHandle* h = TestScalarTensorHandle(1.0f);
  TF_Status* status = TF_NewStatus();
  TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
  CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);

  ASSERT_EQ(0, TFE_TensorDebugInfoOnDeviceNumDims(debug_info));

  TFE_DeleteTensorDebugInfo(debug_info);
  TFE_DeleteTensorHandle(h);
  TF_DeleteStatus(status);
}

TEST(CApiDebug, 2DCPU) {
  TFE_TensorHandle* h = TestMatrixTensorHandle3X2();