Searched refs:xla_primitive_callable (Results 1 – 2 of 2) sorted by relevance
310 xla.xla_primitive_callable.cache_clear()879 cache_misses = xla.xla_primitive_callable.cache_info().misses882 cache_misses, xla.xla_primitive_callable.cache_info().misses,
238 compiled_fun = xla_primitive_callable(prim, *unsafe_map(arg_spec, args), **params)251 def xla_primitive_callable(prim, *arg_specs: Tuple[core.AbstractValue, function