Searched refs:tile_assignment_last_dim_replicate (Results 1 – 2 of 2) sorted by relevance
83 const Array<int64_t>& tile_assignment_last_dim_replicate, in PartialTile() argument85 if (tile_assignment_last_dim_replicate.num_dimensions() == 1 || in PartialTile()86 tile_assignment_last_dim_replicate.dimensions().back() == in PartialTile()87 tile_assignment_last_dim_replicate.num_elements()) { in PartialTile()90 if (tile_assignment_last_dim_replicate.dimensions().back() == 1) { in PartialTile()91 auto new_tile_dims = tile_assignment_last_dim_replicate.dimensions(); in PartialTile()93 auto fully_tiled = tile_assignment_last_dim_replicate; in PartialTile()99 tile_assignment_last_dim_replicate.num_elements() / in PartialTile()100 tile_assignment_last_dim_replicate.dimensions().back()); in PartialTile()104 group_id *= tile_assignment_last_dim_replicate.dim(i); in PartialTile()[all …]
74 const Array<int64_t>& tile_assignment_last_dim_replicate,