Searched refs:dims_in_elems (Results 1 – 2 of 2) sorted by relevance
90 KernelMappingScheme(absl::Span<const int64> dims_in_elems,94 : dims_in_elems_{dims_in_elems[0], dims_in_elems[1], dims_in_elems[2]},
4697 absl::Span<const int64> dims_in_elems = mapping_scheme.GetDimsInElems(); in EmitTilingKernel() local4699 CeilOfRatio(dims_in_elems[0], mapping_scheme.GetTileSizeZ()), in EmitTilingKernel()4700 CeilOfRatio(dims_in_elems[1], mapping_scheme.GetTileSizeY()), in EmitTilingKernel()4701 CeilOfRatio(dims_in_elems[2], mapping_scheme.GetTileSizeX())}; in EmitTilingKernel()4737 dims_in_elems[i] - (dims_in_blocks[i] - 1) * tile_size_for_dim; in EmitTilingKernel()4770 constant(dims_in_elems[kDimZ] - in EmitTilingKernel()