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/tools/
Dhlo_control_flow_flattening.cc71 bool IsNotContainedInLoop(const HloInstruction& while_hlo, in IsNotContainedInLoop()
91 int GetLoopBound(const HloInstruction& while_hlo, const int default_loop_count, in GetLoopBound()
116 int GetLoopBoundWithOuterLoopMax(const HloInstruction& while_hlo, in GetLoopBoundWithOuterLoopMax()
133 HloInstruction* while_hlo, const CallGraph& call_graph) const { in FlattenWhileLoop()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcopy_insertion_test.cc585 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
593 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstruction() local
677 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( in BuildWhileInstructionWithCustomInit() local
709 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
761 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
828 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
887 auto while_hlo = module_->entry_computation()->root_instruction(); in TEST_F() local
933 auto while_hlo = BuildWhileInstruction(condition, body); in TEST_F() local
1169 auto while_hlo = BuildWhileInstruction_InitPointsToConstant(); in TEST_F() local
1186 auto while_hlo = BuildWhileInstruction_InitPointsToParameter(); in TEST_F() local
[all …]
Dbfloat16_propagation_test.cc586 auto while_hlo = builder.AddInstruction( in TEST_F() local
652 auto while_hlo = builder.AddInstruction( in TEST_F() local
729 auto while_hlo = builder.AddInstruction( in TEST_F() local
Dbfloat16_propagation.cc162 HloInstruction* while_hlo) { in DetermineWhileComputationsPrecision()
Ddynamic_dimension_inference_test.cc793 HloInstruction* while_hlo = nullptr; in TEST_F() local
Dhlo_evaluator.cc3265 HloInstruction* while_hlo) { in TryParseAndEvaluateWhileInductionVar()
3300 Status HloEvaluator::HandleWhile(HloInstruction* while_hlo) { in HandleWhile()