Searched refs:JaxprEqnRecipe (Results 1 – 1 of 1) sorted by relevance
443 if isinstance(self.recipe, JaxprEqnRecipe):538 class JaxprEqnRecipe(NamedTuple): class551 ) -> JaxprEqnRecipe:568 return JaxprEqnRecipe(object(), tuple(invars), map(ref, outvars), primitive,573 recipe: JaxprEqnRecipe) -> core.JaxprEqn: argument618 if isinstance(recipe, JaxprEqnRecipe):