Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcustom_call_handler.cc206 auto sort = b_.AddInstruction(HloInstruction::CreateSort( in HandleCustomCallTopK()
H A Dspmd_partitioner.cc2538 HloInstruction* final_sort = b_.AddInstruction(HloInstruction::CreateSort( in HandleSort()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_dimension_inference_test.cc1143 auto* sort = builder.AddInstruction(HloInstruction::CreateSort( in TEST_F()
1182 HloInstruction::CreateSort(ShapeUtil::MakeTupleShape({shape, shape}), 1, in TEST_F()
H A Dhlo_creation_utils.cc552 return builder->AddInstruction(HloInstruction::CreateSort( in MakeSortHlo()
H A Dhlo_instruction.h1033 static std::unique_ptr<HloInstruction> CreateSort(
H A Ddynamic_padder.cc705 HloInstruction* sort = reshape->AddInstruction(HloInstruction::CreateSort( in RewriteDynamicReshapeCombineInput()
H A Dhlo_instruction.cc353 instruction = CreateSort(shape, proto.dimensions(0), all_operands(), in CreateFromProto()
1775 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateSort( in CreateSort() function in xla::HloInstruction
H A Dhlo_parser.cc1792 HloInstruction::CreateSort(*shape, dimensions->at(0), operands, in CreateInstruction()