Searched refs:called_computation_root (Results 1 – 4 of 4) sorted by relevance
1591 HloInstruction* HloCallableInstruction::called_computation_root() const { in called_computation_root() function in xla::HloCallableInstruction1636 clone = called_computation_root(); in CloneAndAppendInstructionIntoCalledComputation()1718 HloInstruction* root = called_computation_root(); in CloneAndAppendInstructionIntoCalledComputation()2144 HloInstruction* called_computation_root) in HloCallInstruction() argument2146 CHECK(called_computation_root != nullptr); in HloCallInstruction()2148 set_parent(called_computation_root->parent()); in HloCallInstruction()2149 set_metadata(called_computation_root->metadata()); in HloCallInstruction()2150 CloneAndAppendInstructionIntoCalledComputation(called_computation_root); in HloCallInstruction()
1186 HloInstruction* called_computation_root() const;1340 HloInstruction* called_computation_root);
1101 const Shape& shape, HloInstruction* called_computation_root);
1868 const Shape& shape, HloInstruction* called_computation_root) { in CreateCall() argument1869 return std::make_unique<HloCallInstruction>(shape, called_computation_root); in CreateCall()