| /external/tensorflow/tensorflow/compiler/xla/service/ |
| flatten_call_graph_test.cc | 127 builder.AddInstruction(HloInstruction::CreateWhile(
137 builder.AddInstruction(HloInstruction::CreateWhile(
173 builder.AddInstruction(HloInstruction::CreateWhile(
|
| while_loop_invariant_code_motion_test.cc | 86 builder.AddInstruction(HloInstruction::CreateWhile(
145 builder.AddInstruction(HloInstruction::CreateWhile(
199 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
239 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
282 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
335 auto* while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
384 builder.AddInstruction(HloInstruction::CreateWhile(
433 builder.AddInstruction(HloInstruction::CreateWhile(
461 builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_alias_analysis_test.cc | 366 HloInstruction::CreateWhile(tuple_shape, condition, body, param));
557 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
651 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
653 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0));
655 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1));
737 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile(
[all...] |
| copy_insertion_test.cc | 650 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
658 auto while_hlo = builder.AddInstruction(HloInstruction::CreateWhile(
766 auto while_hlo = builder->AddInstruction(HloInstruction::CreateWhile
[all...] |
| while_util.cc | 122 HloInstruction::CreateWhile(new_while_shape, new_while_condition,
245 computation->AddInstruction(HloInstruction::CreateWhile(
|
| call_graph_test.cc | 326 builder.AddInstruction(HloInstruction::CreateWhile(
336 builder.AddInstruction(HloInstruction::CreateWhile(
460 a_while = builder.AddInstruction(HloInstruction::CreateWhile(
471 entry_while = builder.AddInstruction(HloInstruction::CreateWhile(
|
| call_inliner_test.cc | 105 HloInstruction::CreateWhile(pred, call_false, call_false, init_value));
|
| while_loop_simplifier.cc | 263 auto* new_while_op = computation->AddInstruction(HloInstruction::CreateWhile(
438 computation->AddInstruction(HloInstruction::CreateWhile(
730 while_op, nested(computation->AddInstruction(HloInstruction::CreateWhile(
[all...] |
| buffer_assignment_test.cc | [all...] |
| hlo_ordering_test.cc | 151 HloInstruction::CreateWhile(scalar_shape, condition, body, constant));
234 HloInstruction::CreateWhile(scalar_shape, condition, body, constant));
|
| bfloat16_propagation_test.cc | 600 HloInstruction::CreateWhile(shape, cond, body, add));
666 HloInstruction::CreateWhile(shape, cond, body, add));
[all...] |
| hlo_dataflow_analysis_test.cc | 516 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
618 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple));
620 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0));
622 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1));
697 auto nested_while = outer_builder.AddInstruction(HloInstruction::CreateWhile(
[all...] |
| hlo_dce_test.cc | 249 auto live_while = builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_memory_scheduler_test.cc | 279 builder.AddInstruction(HloInstruction::CreateWhile(
|
| heap_simulator_test.cc | 80 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
172 builder.AddInstruction(HloInstruction::CreateWhile(
721 HloInstruction* while_op = builder.AddInstruction(HloInstruction::CreateWhile(
[all...] |
| hlo_rematerialization_test.cc | 116 auto while_inst = builder.AddInstruction(HloInstruction::CreateWhile(
|
| hlo_instruction_test.cc | [all...] |
| dynamic_dimension_inference_test.cc | 505 HloInstruction::CreateWhile(tuple_shape, condition, body, a_param));
|
| hlo_instruction.h | 710 static std::unique_ptr<HloInstruction> CreateWhile(const Shape& shape,
[all...] |
| tuple_points_to_analysis_test.cc | [all...] |
| hlo_instruction.cc | [all...] |
| hlo_parser.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),
|
