Searched refs:dev_dims (Results 1 – 2 of 2) sorted by relevance
131 std::vector<int64> dev_dims; in TensorDebugInfo() local132 dev_dims.reserve(rank); in TensorDebugInfo()135 dev_dims.push_back(padded_shape.dimensions(0)); in TensorDebugInfo()139 dev_dims.push_back(padded_shape.dimensions(dim_index)); in TensorDebugInfo()143 return new TFE_TensorDebugInfo(dev_dims); in TensorDebugInfo()149 std::vector<int64> dev_dims = TensorShapeAsVector(*handle_, status); in TensorDebugInfo() local153 return new TFE_TensorDebugInfo(dev_dims); in TensorDebugInfo()163 return debug_info->dev_dims.size(); in TFE_TensorDebugInfoOnDeviceNumDims()168 return debug_info->dev_dims[dim_index]; in TFE_TensorDebugInfoOnDeviceDim()
86 : dev_dims(dims) {} in TFE_TensorDebugInfo()89 std::vector<tensorflow::int64> dev_dims; member