Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h259 HloInstruction* partition_ordinal,
Dspmd_partitioner_util.cc929 HloInstruction* partition_ordinal, in ExchangeHaloAndGetValidData() argument
1006 partition_ordinal, b); in ExchangeHaloAndGetValidData()