Searched refs:TFE_TensorDebugInfoOnDeviceNumDims (Results 1 – 4 of 4) sorted by relevance
34 ASSERT_EQ(0, TFE_TensorDebugInfoOnDeviceNumDims(debug_info)); in TEST()53 ASSERT_EQ(2, TFE_TensorDebugInfoOnDeviceNumDims(debug_info));
77 TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims( in TFE_TensorDebugInfoOnDeviceNumDims() function
219 TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims(
1135 int rank = TFE_TensorDebugInfoOnDeviceNumDims(debug_info); in TFE_Py_TensorShapeOnDevice()