Home
last modified time | relevance | path

Searched defs:while_hlo (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dwhile_transformer_test.cc102 auto while_hlo = builder.AddInstruction( in BuildWhileInstruction() local
130 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); in TEST_F() local
141 auto while_hlo = BuildWhileInstruction(condition, body, 1, 0); in TEST_F() local
152 auto while_hlo = BuildWhileInstruction(condition, body, 0, 10); in TEST_F() local
163 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); in TEST_F() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion_test.cc650 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
658 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
766 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit() local
798 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
826 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
1057 auto while_hlo = BuildWhileInstruction_InitPointsToConstant(); in TEST_F() local
1074 auto while_hlo = BuildWhileInstruction_InitPointsToParameter(); in TEST_F() local
1109 auto while_hlo = BuildWhileInstruction_InitPointsToAmbiguous(); in TEST_F() local
1164 auto while_hlo = BuildWhileInstruction_InitPointsToNonDistinct(); in TEST_F() local
1214 auto while_hlo = BuildWhileInstruction_InitPointsToInterfering(); in TEST_F() local
Dbfloat16_propagation_test.cc599 auto while_hlo = builder.AddInstruction( in TEST_F() local
665 auto while_hlo = builder.AddInstruction( in TEST_F() local
742 auto while_hlo = builder.AddInstruction( in TEST_F() local
Dbfloat16_propagation.cc164 HloInstruction* while_hlo) { in DetermineWhileComputationsPrecision()
Ddynamic_dimension_inference_test.cc521 HloInstruction* while_hlo = nullptr; in TEST_F() local
Dbuffer_assignment.cc1562 const HloInstruction* while_hlo = instruction; in BuildColocatedBufferSets() local
Dhlo_evaluator.cc1319 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile()