Searched refs:TFE_TensorDebugInfo (Results 1 – 5 of 5) sorted by relevance
54 TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo( in TFE_TensorHandleTensorDebugInfo()69 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()73 TFE_TensorDebugInfo* debug_info) { in TFE_DeleteTensorDebugInfo()78 TFE_TensorDebugInfo* debug_info) { in TFE_TensorDebugInfoOnDeviceNumDims()83 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
200 typedef struct TFE_TensorDebugInfo TFE_TensorDebugInfo; typedef208 TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(213 TFE_TensorDebugInfo* debug_info);220 TFE_TensorDebugInfo* debug_info);233 TFE_TensorDebugInfo* debug_info, int dim_index);
22 struct TFE_TensorDebugInfo { struct23 explicit TFE_TensorDebugInfo(const std::vector<tensorflow::int64>& dims) in TFE_TensorDebugInfo() argument
31 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status); in TEST()50 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
1124 TFE_TensorDebugInfo* debug_info = in TFE_Py_TensorShapeOnDevice()