Searched refs:nested_computation_ (Results 1 – 2 of 2) sorted by relevance
47 nested_computation_(nested_computation) {} in IrEmitterNested()65 const auto& params = nested_computation_.parameter_instructions(); in CodegenNestedComputation()80 const HloInstruction* root = nested_computation_.root_instruction(); in CodegenNestedComputation()97 nested_computation_.name())), // The name of the function. in CodegenNestedComputation()120 for (const auto* hlo : nested_computation_.instructions()) { in CodegenNestedComputation()122 hlo != nested_computation_.root_instruction()) { in CodegenNestedComputation()128 TF_RETURN_IF_ERROR(nested_computation_.root_instruction()->Accept(this)); in CodegenNestedComputation()135 nested_computation_.root_instruction(); in CodegenNestedComputation()
85 const HloComputation& nested_computation_; variable