Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h283 struct PartitioningState { struct
293 PartitionedHlo(HloInstruction* hlo, Shape base_shape, PartitioningState state) in PartitionedHlo() argument
334 const PartitioningState& state() const { return state_; } in state()
378 PartitioningState state_;
482 PartitionedHlo::PartitioningState MakePartitioningState() { in MakePartitioningState()
483 PartitionedHlo::PartitioningState state; in MakePartitioningState()
Dspmd_partitioner_util.h346 PartitionedHlo::PartitioningState CreatePerGroupPartitioningState(
347 const PartitionedHlo::PartitioningState& state,
Dspmd_partitioner_util.cc1671 PartitionedHlo::PartitioningState CreatePerGroupPartitioningState( in CreatePerGroupPartitioningState()
1672 const PartitionedHlo::PartitioningState& state, in CreatePerGroupPartitioningState()