Searched refs:root_instruction_ (Results 1 – 2 of 2) sorted by relevance
73 root_instruction_(root_instruction), in HloComputation()88 root_found |= instruction.get() == root_instruction_; in HloComputation()281 root_instruction_->shape())) in set_root_instruction()283 << root_instruction_->shape(); in set_root_instruction()294 root_instruction_ = new_root_instruction; in set_root_instruction()505 s << (instruction == root_instruction_ ? "ROOT " : "") in ToString()714 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
153 HloInstruction* root_instruction() const { return root_instruction_; } in root_instruction()456 HloInstruction* root_instruction_; variable