Searched refs:colocate_with_op (Results 1 – 2 of 2) sorted by relevance
79 Impl(const Scope& other, Tags::Colocate, const Operation& colocate_with_op,85 const Operation& colocate_with_op) const;
214 const Operation& colocate_with_op, bool clear_colocations) in Impl() argument231 : other.impl()->GetColocationConstraints(colocate_with_op)), in Impl()271 const Operation& colocate_with_op) const { in GetColocationConstraints()273 const AttrSlice attrs = colocate_with_op.node()->attrs(); in GetColocationConstraints()283 current_constraints.insert(colocate_with_op.node()->name()); in GetColocationConstraints()