Home
last modified time | relevance | path

Searched refs:dev_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_debug.cc64 std::vector<tensorflow::int64> dev_dims = in TFE_TensorHandleTensorDebugInfo() local
69 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()
79 return debug_info->dev_dims.size(); in TFE_TensorDebugInfoOnDeviceNumDims()
84 return debug_info->dev_dims[dim_index]; in TFE_TensorDebugInfoOnDeviceDim()
Dtfe_tensor_debug_info_internal.h24 : dev_dims(dims) {} in TFE_TensorDebugInfo()
27 std::vector<tensorflow::int64> dev_dims; member