Home
last modified time | relevance | path

Searched refs:copy_done_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dmemory_space_assignment.cc2881 copy_done_ = computation->AddInstruction( in Process()
2887 defining_position_ = HloPosition{copy_done_, {}}; in Process()
2898 ReplaceTupleWith(copy_done_, in Process()
2901 } else if (operand_shape != copy_done_->shape()) { in Process()
2903 << ", new shape = " << copy_done_->shape().ToString() in Process()
2906 HloInstruction::CreateBitcast(operand_shape, copy_done_)); in Process()
2908 replacement_instruction = copy_done_; in Process()
Dmemory_space_assignment.h618 HloInstruction* copy_done() const { return copy_done_; } in copy_done()
655 HloInstruction* copy_done_; variable