Searched refs:dev_dims (Results 1 – 2 of 2) sorted by relevance
123 std::vector<int64> dev_dims; in TFE_TensorHandleTensorDebugInfo() local124 dev_dims.reserve(rank); in TFE_TensorHandleTensorDebugInfo()127 dev_dims.push_back(padded_shape.dimensions(0)); in TFE_TensorHandleTensorDebugInfo()131 dev_dims.push_back(padded_shape.dimensions(dim_index)); in TFE_TensorHandleTensorDebugInfo()135 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()141 std::vector<int64> dev_dims = TensorShapeAsVector(handle, status); in TFE_TensorHandleTensorDebugInfo() local145 return new TFE_TensorDebugInfo(dev_dims); in TFE_TensorHandleTensorDebugInfo()155 return debug_info->dev_dims.size(); in TFE_TensorDebugInfoOnDeviceNumDims()160 return debug_info->dev_dims[dim_index]; in TFE_TensorDebugInfoOnDeviceDim()
96 : dev_dims(dims) {} in TFE_TensorDebugInfo()99 std::vector<tensorflow::int64> dev_dims; member