Home
last modified time | relevance | path

Searched refs:TFE_TensorDebugInfoOnDeviceDim (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_debug_test.cc44 EXPECT_EQ(3, TFE_TensorDebugInfoOnDeviceDim(debug_info, 0));
45 EXPECT_EQ(2, TFE_TensorDebugInfoOnDeviceDim(debug_info, 1));
Dc_api_debug.cc158 TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim( in TFE_TensorDebugInfoOnDeviceDim() function
Dc_api.h251 TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim(
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc1077 tensorflow::int64 dim_size = TFE_TensorDebugInfoOnDeviceDim(debug_info, i); in TFE_Py_TensorShapeOnDevice()