| 151 | } |
| 152 | |
| 153 | TF_CAPI_EXPORT extern int TFE_TensorDebugInfoOnDeviceNumDims( |
| 154 | TFE_TensorDebugInfo* debug_info) { |
| 155 | return debug_info->dev_dims.size(); |
| 156 | } |
| 157 | |
| 158 | TF_CAPI_EXPORT extern int64_t TFE_TensorDebugInfoOnDeviceDim( |
| 159 | TFE_TensorDebugInfo* debug_info, int dim_index) { |