Home
last modified time | relevance | path

Searched refs:ShuffleWithinEachPartitionUsingOneHot (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dfft_handler.cc110 HloInstruction* ShuffleWithinEachPartitionUsingOneHot(HloInstruction* hlo, in ShuffleWithinEachPartitionUsingOneHot() function
394 result = ShuffleWithinEachPartitionUsingOneHot(result, num_partitions_, in HandleFft()