Code for Memory-efficient array redistribution through portable collective communication

Add code button
avatar
Github Icontensorflow/tensorflow/blob/8c6d9ae2b497ac99ec0b5a4a9a537d4f66e5e678/tensorflow/compiler/xla/service/spmd/spmd_partitioner.cc

Human or AI?

Log in/sign up for free to see all code implementations