| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_schedule_test.cc | 251 const HloInstruction* xla_while = in TEST_F() local
320 HloInstruction* xla_while = in TEST_F() local
|
| D | hlo_module_dce.cc | 50 const auto* xla_while = instruction; in RunWhileDCE() local
|
| D | hlo_ordering.cc | 272 const HloInstruction* xla_while = use.instruction; in UsesBeforeValueDefinition() local
297 const HloInstruction* xla_while = value.defining_instruction(); in UsesBeforeValueDefinition() local
|
| D | hlo_alias_analysis_test.cc | 359 auto xla_while = builder.AddInstruction( in TEST_F() local
548 auto xla_while = builder.AddInstruction( in TEST_F() local
815 auto xla_while = builder.AddInstruction( in TEST_F() local
965 auto xla_while = builder.AddInstruction( in TEST_F() local
1219 auto xla_while = builder.AddInstruction( in TEST_F() local
|
| D | hlo_ordering_test.cc | 149 auto xla_while = builder.AddInstruction( in TEST_F() local
232 auto xla_while = builder.AddInstruction( in TEST_F() local
|
| D | loop_schedule_linearizer.cc | 71 HloInstruction* xla_while, HloAliasAnalysis& alias_analysis) { in AddControlEdgesForLoopWrites()
|
| D | copy_insertion_test.cc | 738 auto xla_while = BuildWhileInstructionWithCustomInit(loop_state_shape_, in BuildWhileInstruction_InitPointsToInterfering() local
1541 auto xla_while = builder.AddInstruction( in TEST_F() local
1818 auto xla_while = builder.AddInstruction( in TEST_F() local
1959 auto xla_while = builder.AddInstruction(HloInstruction::CreateWhile( in TEST_F() local
1982 for (const HloInstruction* xla_while : whiles) { in TEST_F() local
2016 auto xla_while = builder.AddInstruction( in TEST_F() local
2175 HloInstruction* xla_while = builder.AddInstruction( in BM_ParallelWhiles() local
2240 HloInstruction* xla_while = builder.AddInstruction( in BM_ManyElementTuple() local
|
| D | hlo_liveness_analysis.cc | 204 auto* xla_while = callsite.instruction(); in PropagateLivenessToParameterCallers() local
|
| D | copy_insertion.cc | 169 const HloInstruction* xla_while, in IndicesToCopyForWhile()
280 HloInstruction* xla_while) { in AddCopiesForWhile()
|
| D | dfs_hlo_visitor_with_default.h | 210 Status HandleWhile(HloInstructionPtr xla_while) override { in HandleWhile()
|
| D | hlo_dataflow_analysis_test.cc | 422 auto xla_while = builder.AddInstruction( in TEST_P() local
858 auto xla_while = builder.AddInstruction( in TEST_P() local
1118 auto xla_while = builder.AddInstruction( in TEST_P() local
1466 auto xla_while = builder.AddInstruction( in TEST_P() local
|
| D | hlo_verifier.cc | 920 Status ShapeVerifier::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
1815 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
|
| D | hlo_dataflow_analysis.cc | 724 bool HloDataflowAnalysis::UpdateWhileValueSet(HloInstruction* xla_while) { in UpdateWhileValueSet()
|
| D | hlo_cost_analysis.cc | 960 Status HloCostAnalysis::HandleWhile(const HloInstruction* xla_while) { in HandleWhile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 262 Status HandleWhile(HloInstruction* xla_while) override { in HandleWhile()
|
| D | ir_emitter.cc | 2357 Status IrEmitter::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | while_test.cc | 449 auto xla_while = While(condition, body, init); in XLA_TEST_F() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 2431 Status IrEmitterUnnested::HandleWhile(HloInstruction* xla_while) { in HandleWhile()
3778 const HloInstruction* xla_while, in CheckWhileBuffersShareAllocation()
|