Home
last modified time | relevance | path

Searched refs:TFE_TensorDebugInfo (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc52 TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo( in TFE_TensorHandleTensorDebugInfo()
135 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()
145 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()
149 TFE_TensorDebugInfo* debug_info) { in TFE_DeleteTensorDebugInfo()
154 TFE_TensorDebugInfo* debug_info) { in TFE_TensorDebugInfoOnDeviceNumDims()
159 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
Dc_api.h219 typedef struct TFE_TensorDebugInfo TFE_TensorDebugInfo; typedef
227 TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo(
232 TFE_TensorDebugInfo* debug_info);
239 TFE_TensorDebugInfo* debug_info);
252 TFE_TensorDebugInfo* debug_info, int dim_index);
Dc_api_debug_test.cc26 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status); in TEST()
39 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
Dc_api_internal.h94 struct TFE_TensorDebugInfo { struct
95 TFE_TensorDebugInfo(const std::vector<tensorflow::int64>& dims) in TFE_TensorDebugInfo() argument
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc1063 TFE_TensorDebugInfo* debug_info = in TFE_Py_TensorShapeOnDevice()