Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc55 TF_CAPI_EXPORT extern TFE_TensorDebugInfo* TFE_TensorHandleTensorDebugInfo( in TFE_TensorHandleTensorDebugInfo()
60 TFE_TensorDebugInfo* tensorflow::TensorHandleInterface::TensorDebugInfo( in TensorDebugInfo()
143 return new TFE_TensorDebugInfo(dev_dims); in TensorDebugInfo()
153 return new TFE_TensorDebugInfo(dev_dims); in TensorDebugInfo()
157 TFE_TensorDebugInfo* debug_info) { in TFE_DeleteTensorDebugInfo()
162 TFE_TensorDebugInfo* debug_info) { in TFE_TensorDebugInfoOnDeviceNumDims()
167 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
Dc_api.h200 typedef struct TFE_TensorDebugInfo TFE_TensorDebugInfo; typedef
208 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);
Dc_api_debug_test.cc26 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status); in TEST()
39 TFE_TensorDebugInfo* debug_info = TFE_TensorHandleTensorDebugInfo(h, status);
Dtensor_handle_interface.h54 virtual TFE_TensorDebugInfo* TensorDebugInfo(tensorflow::Status* status) = 0;
76 TFE_TensorDebugInfo* TensorDebugInfo(Status* status) override;
Dc_api_internal.h84 struct TFE_TensorDebugInfo { struct
85 explicit TFE_TensorDebugInfo(const std::vector<tensorflow::int64>& dims) in TFE_TensorDebugInfo() argument
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.cc1071 TFE_TensorDebugInfo* debug_info = in TFE_Py_TensorShapeOnDevice()