| /external/tensorflow/tensorflow/compiler/xla/service/ |
| bfloat16_support.cc | 30 case HloOpcode::kWhile:
48 case HloOpcode::kWhile:
66 case HloOpcode::kWhile:
|
| while_util_test.cc | 71 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile);
102 ASSERT_EQ(while_instr->opcode(), HloOpcode::kWhile);
|
| flatten_call_graph.cc | 37 case HloOpcode::kWhile: {
|
| hlo_ordering.cc | 60 if (a_ancestor == b_ancestor && a_ancestor->opcode() == HloOpcode::kWhile) {
113 if (b.is_phi() && b.defining_instruction()->opcode() == HloOpcode::kWhile &&
150 if (use.instruction->opcode() == HloOpcode::kWhile) {
164 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) {
|
| hlo_alias_analysis.cc | 74 // because of aliasing rules (eg, in-place kWhile instruction).
79 // Gather the set of buffers with aliasing rules (eg, kWhile) which this
181 if (use.instruction->opcode() == HloOpcode::kWhile) {
199 if (callsite.instruction()->opcode() == HloOpcode::kWhile) {
220 if (callsite.instruction()->opcode() == HloOpcode::kWhile &&
237 if (value.defining_instruction()->opcode() == HloOpcode::kWhile) {
|
| while_loop_simplifier_test.cc | 129 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
145 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
158 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
174 ASSERT_EQ(while_op->opcode(), HloOpcode::kWhile);
403 return instr != while_op && instr->opcode() == HloOpcode::kWhile;
|
| bfloat16_conversion_folding.cc | 156 hlo->opcode() == HloOpcode::kWhile || //
|
| hlo_opcode.h | 130 V(kWhile, "while")
|
| while_loop_simplifier.cc | 127 CHECK_EQ(while_op->opcode(), HloOpcode::kWhile);
196 CHECK_EQ(while_op->opcode(), HloOpcode::kWhile);
287 CHECK_EQ(while_op->opcode(), HloOpcode::kWhile);
619 if (instr->opcode() == HloOpcode::kWhile) {
|
| liveness_util.cc | 240 user->opcode() == HloOpcode::kWhile) {
344 user->opcode() == HloOpcode::kWhile) {
|
| hlo_dataflow_analysis.cc | 373 // a phi-node only for kWhile to account for the dataflow through back-edges
446 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) {
497 // contrasts with kWhile instruction (which does define a phi value) which has
540 CHECK_EQ(xla_while->opcode(), HloOpcode::kWhile);
571 case HloOpcode::kWhile:
664 } else if (callsite.instruction()->opcode() == HloOpcode::kWhile) {
733 case HloOpcode::kWhile:
[all...] |
| hlo_element_type_converter.cc | 142 if (opcode == HloOpcode::kWhile || opcode == HloOpcode::kCall ||
|
| hlo_value.cc | 138 case HloOpcode::kWhile:
|
| while_loop_invariant_code_motion.cc | 272 return instr->opcode() == HloOpcode::kWhile;
|
| call_graph.cc | 58 case HloOpcode::kWhile:
|
| instruction_fusion.cc | 123 case HloOpcode::kWhile:
|
| copy_insertion.cc | 165 // Add kCopy instructions around the given kWhile instruction to eliminate any
177 // (1) The init value is copied before the kWhile instruction. Before:
181 // kWhile
191 // kWhile
196 // with the kWhile.
229 VLOG(2) << "Adding copies for kWhile instruction " << xla_while->name();
230 TF_RET_CHECK(xla_while->opcode() == HloOpcode::kWhile);
235 VLOG(2) << "No copies necessary for kWhile instruction "
302 // live-range interference. Generally interference can only occur around kWhile
310 if (instruction->opcode() == HloOpcode::kWhile) {
[all...] |
| hlo_verifier.cc | 283 // The shape of kWhile should match the shape of the body computation it
398 case HloOpcode::kWhile:
478 case HloOpcode::kWhile:
770 } else if (instruction->opcode() == HloOpcode::kWhile) {
[all...] |
| bfloat16_normalization.cc | 330 hlo->opcode() == HloOpcode::kWhile || //
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| for_thunk.cc | 28 : Thunk(Kind::kWhile, hlo),
|
| thunk.h | 56 kWhile,
|
| while_thunk.cc | 30 : Thunk(Kind::kWhile, hlo),
|
| gpu_copy_insertion.cc | 100 if (instruction->opcode() != HloOpcode::kWhile &&
|
| while_transformer.cc | 357 HloOpcode::kWhile, "while",
495 if (while_hlo->opcode() != HloOpcode::kWhile) {
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| parallel_task_assignment.cc | 186 if (instruction->opcode() == HloOpcode::kWhile) {
|
