Searched refs:GetDimsInElems (Results 1 – 2 of 2) sorted by relevance
113 absl::Span<const int64> GetDimsInElems() const { return dims_in_elems_; } in GetDimsInElems() function
4036 mapping_scheme.GetDimsInElems()[kDimX] % tile_size_x == 0 && in EmitTile()4106 output_shape.element_type(), mapping_scheme.GetDimsInElems()); in EmitTileElementForCopy()4132 kernel_mapping_scheme.GetDimsInElems(), b_); in GetUnnormalizedIndex()4371 absl::Span<const int64> dims_in_elem = kernel_mapping_scheme.GetDimsInElems(); in GetUntransposedOutputLinearAddress()4697 absl::Span<const int64> dims_in_elems = mapping_scheme.GetDimsInElems(); in EmitTilingKernel()4752 return IrArray::Index(elem_multi_index, mapping_scheme.GetDimsInElems(), in EmitTilingKernel()