Searched refs:translations_with_avals (Results 1 – 2 of 2) sorted by relevance
326 elif prim in translations_with_avals:327 rule = translations_with_avals[prim]446 elif eqn.primitive in translations_with_avals:447 rule = translations_with_avals[eqn.primitive]911 translations_with_avals: Dict[core.Primitive, Callable] = {} variable
210 xla.translations_with_avals[threefry2x32_p] = xla.lower_fun(1041 xla.translations_with_avals[random_gamma_p] = xla.lower_fun(