Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h366 absl::optional<HloInstruction*> PadFromPartialReplicateShape(
Dspmd_partitioner_util.cc502 absl::optional<HloInstruction*> PadFromPartialReplicateShape( in PadFromPartialReplicateShape() function
Dspmd_partitioner.cc1016 auto padded_hlo = PadFromPartialReplicateShape( in ReshardFromPartialReplicateWithDynamicSlice()