Searched refs:root_instruction_ (Results 1 – 2 of 2) sorted by relevance
81 root_instruction_(root_instruction), in HloComputation()99 root_found |= instruction.get() == root_instruction_; in HloComputation()383 root_instruction_->shape())) in set_root_instruction()385 << root_instruction_->shape(); in set_root_instruction()399 root_instruction_->shape())) { in set_root_instruction()406 root_instruction_ = new_root_instruction; in set_root_instruction()616 if (instruction == root_instruction_) { in ToCord()886 *program_shape.mutable_result() = root_instruction_->shape(); in ComputeProgramShape()
217 HloInstruction* root_instruction() const { return root_instruction_; } in root_instruction()710 HloInstruction* root_instruction_; variable