Searched refs:dims_in_elems_ (Results 1 – 2 of 2) sorted by relevance
123 dims_in_elems_(dims_in_elems.begin(), dims_in_elems.end()), in KernelMappingScheme()128 DCHECK_EQ(dims_in_elems_.size(), 3);134 dims_in_tiles_ = ElementWiseCeilOfRatio<int64>(dims_in_elems_, tile_sizes_);143 VLOG(10) << "dims_in_elems_ = [" << absl::StrJoin(dims_in_elems_, ",") << "]";152 DCHECK_EQ(normalized_shape_index.size(), dims_in_elems_.size()); in GetUnnormalizedIndex()
105 return dims_in_elems_; in GetDimensionsInElements()184 std::vector<int64> dims_in_elems_; variable