Searched defs:HloSharding (Results 1 – 1 of 1) sorted by relevance
219 HloSharding() in HloSharding() function230 explicit HloSharding(int64 device_id) in HloSharding() function235 explicit HloSharding(const Array<int64>& tile_assignment) in HloSharding() function240 explicit HloSharding(const std::vector<HloSharding>& tuple_shardings) in HloSharding() function