Searched refs:dims_in_tiles_ (Results 1 – 2 of 2) sorted by relevance
108 return dims_in_tiles_; in GetDimensionsInTiles()115 return absl::c_accumulate(dims_in_tiles_, 1LL, std::multiplies<int64>()); in GetNumberOfTilesInTotal()190 std::vector<int64> dims_in_tiles_; variable
134 dims_in_tiles_ = ElementWiseCeilOfRatio<int64>(dims_in_elems_, tile_sizes_);136 absl::c_transform(req_block_sizes, dims_in_tiles_,141 dims_in_blocks_ = ElementWiseCeilOfRatio<int64>(dims_in_tiles_, block_sizes_);144 VLOG(10) << "dims_in_tiles_ = [" << absl::StrJoin(dims_in_tiles_, ",") << "]";