| /external/tensorflow/tensorflow/compiler/xla/service/ |
| while_loop_invariant_code_motion_test.cc | 84 builder.AddInstruction(HloInstruction::CreateWhile(
143 builder.AddInstruction(HloInstruction::CreateWhile(
197 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
236 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
270 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
311 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
356 builder.AddInstruction(HloInstruction::CreateWhile(
405 builder.AddInstruction(HloInstruction::CreateWhile(
432 builder.AddInstruction(HloInstruction::CreateWhile(
|
| while_loop_simplifier_test.cc | 91 builder.AddInstruction(HloInstruction::CreateWhile(
211 builder.AddInstruction(HloInstruction::CreateWhile(
247 builder.AddInstruction(HloInstruction::CreateWhile(
277 builder.AddInstruction(HloInstruction::CreateWhile(
300 builder.AddInstruction(HloInstruction::CreateWhile(
333 builder.AddInstruction(HloInstruction::CreateWhile(
391 auto* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
439 builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_scheduling_test.cc | 74 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
|
| flatten_call_graph_test.cc | 126 builder.AddInstruction(HloInstruction::CreateWhile(
136 builder.AddInstruction(HloInstruction::CreateWhile(
172 builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_alias_analysis_test.cc | 361 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
454 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
456 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0));
458 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1));
539 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile(
552 HloInstruction::CreateWhile(tuple_shape, condition2, outer_body, tuple));
625 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
[all...] |
| copy_insertion_test.cc | 641 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
649 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
755 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile(
[all...] |
| call_inliner_test.cc | 104 HloInstruction::CreateWhile(pred, call_false, call_false, init_value));
|
| while_util.cc | 116 HloInstruction::CreateWhile(new_while_shape, new_while_condition,
|
| buffer_assignment_test.cc | 701 auto while_op = builder.AddInstruction(HloInstruction::CreateWhile(
[all...] |
| hlo_ordering_test.cc | 197 HloInstruction::CreateWhile(scalar_shape, condition, body, constant));
260 HloInstruction::CreateWhile(scalar_shape, condition, body, constant));
|
| call_graph_test.cc | 316 builder.AddInstruction(HloInstruction::CreateWhile(
326 builder.AddInstruction(HloInstruction::CreateWhile(
439 a_while = builder.AddInstruction(HloInstruction::CreateWhile(
450 entry_while = builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_dataflow_analysis_test.cc | 508 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
609 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
611 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0));
613 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1));
687 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile(
700 HloInstruction::CreateWhile(tuple_shape, condition, outer_body, tuple));
[all...] |
| hlo_dce_test.cc | 249 auto live_while = builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_rematerialization_test.cc | 110 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
|
| liveness_util_test.cc | 409 auto whil = builder.AddInstruction(HloInstruction::CreateWhile(
|
| while_loop_simplifier.cc | 508 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile(
|
| hlo_instruction.h | 444 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
[all...] |
| heap_simulator_test.cc | 442 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
[all...] |
| hlo_instruction_test.cc | [all...] |
| hlo_instruction.cc | 471 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateWhile(
[all...] |
| user_computation.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| cpu_copy_insertion_test.cc | 89 HloInstruction::CreateWhile(scalar_shape_, condition, body, param_0));
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| while_transformer_test.cc | 103 HloInstruction::CreateWhile(GetLoopStateShape(ind_var_tuple_index),
|
| /external/tensorflow/tensorflow/compiler/xla/tools/parser/ |
| hlo_parser.cc | 590 instruction = builder->AddInstruction(HloInstruction::CreateWhile(
[all...] |
