Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcollective_decomposer_utils.cc45 return computation->AddInstruction(HloInstruction::CreatePartitionId()); in CreateStartIndicesForCollectiveDecomposition()
56 computation->AddInstruction(HloInstruction::CreatePartitionId()))); in CreateStartIndicesForCollectiveDecomposition()
81 computation->AddInstruction(HloInstruction::CreatePartitionId()); in CreateStartIndicesForCollectiveDecomposition()
Dhlo_instruction.h813 static std::unique_ptr<HloInstruction> CreatePartitionId(
Dhlo_instruction.cc619 instruction = CreatePartitionId(shape); in CreateFromProto()
1277 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreatePartitionId( in CreatePartitionId() function in xla::HloInstruction
2012 clone = CreatePartitionId(shape); in CloneWithNewOperands()
Dhlo_parser.cc1470 builder->AddInstruction(HloInstruction::CreatePartitionId()); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3436 return b->AddInstruction(HloInstruction::CreatePartitionId()); in GetDefaultCollectiveOpsCreator()