Searched refs:dev_dims (Results 1 – 2 of 2) sorted by relevance
64 std::vector<tensorflow::int64> dev_dims = in TFE_TensorHandleTensorDebugInfo() local69 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()
24 : dev_dims(dims) {} in TFE_TensorDebugInfo()27 std::vector<tensorflow::int64> dev_dims; member