Searched defs:while_hlo (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | while_transformer_test.cc | 102 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/ |
D | copy_insertion_test.cc | 656 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 …]
|
D | bfloat16_propagation_test.cc | 629 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
|
D | bfloat16_propagation.cc | 166 HloInstruction* while_hlo) { in DetermineWhileComputationsPrecision()
|
D | dynamic_dimension_inference_test.cc | 786 HloInstruction* while_hlo = nullptr; in TEST_F() local
|
D | hlo_evaluator.cc | 2031 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile()
|