Searched refs:primitive_uses_outfeed (Results 1 – 2 of 2) sorted by relevance
204 return any(primitive_uses_outfeed(eqn.primitive, eqn.params)216 def primitive_uses_outfeed(prim: core.Primitive, params: Dict) -> bool: function257 if primitive_uses_outfeed(prim, params):
1118 if not xla.primitive_uses_outfeed(eqn.primitive, eqn.params):