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()336 root_instruction_->shape())) in set_root_instruction()338 << root_instruction_->shape(); in set_root_instruction()352 root_instruction_->shape())) { in set_root_instruction()359 root_instruction_ = new_root_instruction; in set_root_instruction()631 new_options.indent_amount(), instruction == root_instruction_) in ToString()843 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
190 HloInstruction* root_instruction() const { return root_instruction_; } in root_instruction()540 HloInstruction* root_instruction_; variable