Home
last modified time | relevance | path

Searched refs:called_computation_root (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instructions.cc1591 HloInstruction* HloCallableInstruction::called_computation_root() const { in called_computation_root() function in xla::HloCallableInstruction
1636 clone = called_computation_root(); in CloneAndAppendInstructionIntoCalledComputation()
1718 HloInstruction* root = called_computation_root(); in CloneAndAppendInstructionIntoCalledComputation()
2144 HloInstruction* called_computation_root) in HloCallInstruction() argument
2146 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()
H A Dhlo_instructions.h1186 HloInstruction* called_computation_root() const;
1340 HloInstruction* called_computation_root);
H A Dhlo_instruction.h1101 const Shape& shape, HloInstruction* called_computation_root);
H A Dhlo_instruction.cc1868 const Shape& shape, HloInstruction* called_computation_root) { in CreateCall() argument
1869 return std::make_unique<HloCallInstruction>(shape, called_computation_root); in CreateCall()