Home
last modified time | relevance | path

Searched refs:CreateConditional (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc241 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Dconditional_simplifier_test.cc96 builder.AddInstruction(HloInstruction::CreateConditional( in MakeConditional()
Dhlo_instruction.h933 static std::unique_ptr<HloInstruction> CreateConditional(
938 static std::unique_ptr<HloInstruction> CreateConditional(
Dcall_graph_test.cc263 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Dhlo_dataflow_analysis_test.cc1740 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1834 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1949 inner_builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
1969 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
Dhlo_instruction.cc252 CreateConditional(shape, cond_operands[0], all_computations(), in CreateFromProto()
1198 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction
1215 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateConditional( in CreateConditional() function in xla::HloInstruction
1799 clone = CreateConditional(shape, new_operands[0], in CloneWithNewOperands()
Dhlo_instruction_test.cc1475 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
1735 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Dconditional_code_motion.cc489 conditional_parent->AddInstruction(HloInstruction::CreateConditional( in ConvertSpecialMove()
Dlayout_assignment_test.cc797 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Ddynamic_dimension_inference_test.cc881 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Ddynamic_dimension_inference.cc1373 hlo->parent()->AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
Dbuffer_assignment_test.cc1101 auto conditional = builder.AddInstruction(HloInstruction::CreateConditional( in TEST_F()
Dhlo_parser.cc2230 instruction = builder->AddInstruction(HloInstruction::CreateConditional( in ParseInstructionRhs()
Dmemory_space_assignment_test.cc2895 builder.AddInstruction(HloInstruction::CreateConditional( in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc2223 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleSingleDevice()
2615 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleInfeed()
2888 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleConditional()
3034 return b_.AddInstruction(HloInstruction::CreateConditional( in HandleOutfeed()
Ddot_handler.cc1376 conditional = body_b.AddInstruction(HloInstruction::CreateConditional( in PartitionBaseCase()