Searched refs:TileAssignment (Results 1 – 2 of 2) sorted by relevance
31 using TileAssignment = Array<int64>; variable48 OpSharding Tile(const Shape& tile_shape, const TileAssignment& tile_assignment);
42 const TileAssignment& tile_assignment) { in Tile()