Searched refs:non_io_hlo (Results 1 – 1 of 1) sorted by relevance
66 for (const HloInstruction* non_io_hlo : non_io_hlos) { in EmitBasePointersForHlos() local67 if (already_bound_for_this_function.contains(non_io_hlo)) { in EmitBasePointersForHlos()70 already_bound_for_this_function.insert(non_io_hlo); in EmitBasePointersForHlos()72 if (non_io_hlo->opcode() == HloOpcode::kGetTupleElement) { in EmitBasePointersForHlos()77 ->GetUniqueTopLevelSlice(non_io_hlo->LatestNonGteAncestor()) in EmitBasePointersForHlos()86 BindHloToIrValue(*non_io_hlo, in EmitBasePointersForHlos()87 EmitGetTupleElement(non_io_hlo, base_ptr)); in EmitBasePointersForHlos()92 if (!buffer_assignment_->HasTopLevelAllocation(non_io_hlo)) { in EmitBasePointersForHlos()97 non_io_hlo->shape(), in EmitBasePointersForHlos()104 buffer_assignment_->GetUniqueSlice(non_io_hlo, index); in EmitBasePointersForHlos()[all …]