Searched defs:called_computation_root (Results 1 – 2 of 2) sorted by relevance
1591 HloInstruction* HloCallableInstruction::called_computation_root() const { in called_computation_root() function in xla::HloCallableInstruction2144 HloInstruction* called_computation_root) in HloCallInstruction()
1868 const Shape& shape, HloInstruction* called_computation_root) { in CreateCall()