Home
last modified time | relevance | path

Searched defs:halo_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dcustom_call_handler.cc287 Shape halo_shape = input.hlo()->shape(); in HandleCustomCallSPMDInternal_RotateRight() local
Dspmd_partitioner_util.cc763 auto halo_shape = hlo->shape(); in ExchangeHalo() local
825 auto halo_shape = hlo->shape(); in ExchangeHalo() local