| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer_test.cc | 102 auto while_hlo = builder.AddInstruction( local
106 return while_hlo;
150 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); local
155 auto result = gpu::CanTransformWhileToFor(while_hlo);
170 auto while_hlo = BuildWhileInstruction(condition, body, 1, 0); local
175 auto result = gpu::CanTransformWhileToFor(while_hlo);
190 auto while_hlo = BuildWhileInstruction(condition, body, 0, 10); local
195 auto result = gpu::CanTransformWhileToFor(while_hlo);
209 auto while_hlo = BuildWhileInstruction(condition, body, 0, 0); local
214 auto result = gpu::CanTransformWhileToFor(while_hlo);
[all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| buffer_assignment.cc | 1285 const HloInstruction* while_hlo = instruction; local
[all...] |
| copy_insertion_test.cc | 641 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( local
644 return while_hlo;
649 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile( local
652 return while_hlo;
755 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile( local
758 return while_hlo;
787 auto while_hlo = BuildWhileInstruction(condition, body); local
815 auto while_hlo = BuildWhileInstruction(condition, body); local
1046 auto while_hlo = BuildWhileInstruction_InitPointsToConstant(); local
1063 auto while_hlo = BuildWhileInstruction_InitPointsToParameter(); local
1098 auto while_hlo = BuildWhileInstruction_InitPointsToAmbiguous(); local
1153 auto while_hlo = BuildWhileInstruction_InitPointsToNonDistinct(); local
1203 auto while_hlo = BuildWhileInstruction_InitPointsToInterfering(); local
[all...] |
