| /external/tensorflow/tensorflow/compiler/xla/service/ |
| while_loop_simplifier.cc | 56 HloInstruction* while_init = while_op->mutable_operand(0); local
61 if (!while_init->shape().IsTuple()) {
89 const int64 tuple_size = ShapeUtil::TupleElementCount(while_init->shape());
98 // The HLO verifier ensures that while_input's shape matches while_init's
174 while_init->shape().tuple_shapes(old_idx));
248 // Add a new while_init instruction that repackages the old while_init
250 // clean this up in the common case where while_init is a tuple op. (It's
257 while_init->shape().tuple_shapes(old_idx), while_init, old_idx)))
313 auto* while_init = while_op->mutable_operand(0); local
507 auto while_init = while_op->operand(0); local
645 auto* while_init = while_op->mutable_operand(0); local
772 auto* while_init = while_op->mutable_operand(0); local
[all...] |
| while_loop_analysis.cc | 156 auto* while_init = while_op->operand(0); local
157 if (while_init->opcode() != HloOpcode::kTuple) {
158 VLOG(2) << "While init expected to be a tuple: " << while_init->ToString();
347 auto* while_init = while_op->mutable_operand(0); local
348 auto* indvar_init = while_init->mutable_operand(*indvar_tuple_idx);
|
| hlo_memory_scheduler_test.cc | 275 HloInstruction* while_init = local
280 r1f32, cond_computation, body_computation, while_init));
|
| copy_insertion.cc | 277 HloInstruction* while_init = xla_while->mutable_operand(0); local
280 xla_while->parent()->DeepCopyInstruction(while_init, &indices_to_copy));
281 TF_RETURN_IF_ERROR(while_init->ReplaceUseWith(xla_while, while_init_copy));
[all...] |
| ar_crs_combiner.cc | 123 auto init_tuples = GetAllTuples(while_instr->while_init());
|
| heap_simulator_test.cc | 167 HloInstruction* while_init = local
173 r1f32, cond_computation, body_computation, while_init));
197 std::vector<HloInstruction*> entry_comp_vec = {while_init, while_loop, bcast,
[all...] |
| copy_insertion_test.cc | 1753 auto while_init = builder.AddInstruction(HloInstruction::CreateTuple( local
[all...] |
| hlo_instruction.h | [all...] |
| hlo_instruction.cc | [all...] |
