Searched defs:tile_dimensions (Results 1 – 3 of 3) sorted by relevance
3382 IrEmitterUnnested::ValueVector3 tile_dimensions, in EmitXTileLoop()3429 const ThreadIdInfo& thread_id_info, ValueVector3 tile_dimensions, in EmitTile()4137 ValueVector3 tile_dimensions; in EmitTilingKernel() local4247 ValueVector3 tile_dimensions) { in EmitHlo021Tile()4754 ValueVector3 tile_dimensions) { in EmitIRForReduction()
1676 int64_t tile_dimensions = sharding.tile_assignment().num_dimensions(); in GroupShardingOnDims() local1704 int64_t tile_dimensions = sharding.tile_assignment().num_dimensions(); in GetManualSubgroupSharding() local
1761 auto tile_dimensions = shape.layout().tiles(0).dimensions(); in ArraySize() local