/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...] |