xla
Supoort unordered sharding spec correctly
#5305
Merged
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Overview
Commits
3
Changes
View On
GitHub
Commits
Supoort non-ordered sharding spec correctly
JackCaoG
committed
3 years ago
use permute instead of transpose
JackCaoG
committed
3 years ago
use dim > 2 to suit TPU v3(otherwise can't be divide evenly)
JackCaoG
committed
3 years ago
Loading
