Searched refs:nested_computation_ (Results 1 – 2 of 2) sorted by relevance
42 nested_computation_(nested_computation) {} in IrEmitterNested()61 nested_computation_.parameter_instructions()) { in CodegenNestedComputation()71 const HloInstruction* root = nested_computation_.root_instruction(); in CodegenNestedComputation()88 nested_computation_.name())), // The name of the function. in CodegenNestedComputation()111 for (const auto* hlo : nested_computation_.instructions()) { in CodegenNestedComputation()113 hlo != nested_computation_.root_instruction()) { in CodegenNestedComputation()119 TF_RETURN_IF_ERROR(nested_computation_.root_instruction()->Accept(this)); in CodegenNestedComputation()126 nested_computation_.root_instruction(); in CodegenNestedComputation()
71 const HloComputation& nested_computation_; variable