Searched refs:_def_passthrough (Results 1 – 1 of 1) sorted by relevance
266 def _def_passthrough(prim, argnums=(0,)): function274 _def_passthrough(lax.select_p, (0, 1, 2))275 _def_passthrough(lax.broadcast_in_dim_p)276 _def_passthrough(xla.device_put_p)278 _def_passthrough(lax_internal.tie_in_p, (0, 1))