Home
last modified time | relevance | path

Searched defs:while_hlo (Results 1 – 6 of 6) 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.cc656 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
664 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
772 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit() local
804 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
857 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
925 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
985 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
1031 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
1267 auto while_hlo = BuildWhileInstruction_InitPointsToConstant(); in TEST_F() local
1284 auto while_hlo = BuildWhileInstruction_InitPointsToParameter(); in TEST_F() local
[all …]
Dbfloat16_propagation_test.cc629 auto while_hlo = builder.AddInstruction( in TEST_F() local
695 auto while_hlo = builder.AddInstruction( in TEST_F() local
772 auto while_hlo = builder.AddInstruction( in TEST_F() local
Dbfloat16_propagation.cc166 HloInstruction* while_hlo) { in DetermineWhileComputationsPrecision()
Ddynamic_dimension_inference_test.cc786 HloInstruction* while_hlo = nullptr; in TEST_F() local
Dhlo_evaluator.cc2031 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile()