Searched refs:_flush_fun (Results 1 – 1 of 1) sorted by relevance
1260 def _flush_fun(cache): function1285 flush_op = tpu.outside_compilation(_flush_fun, cache.value())1287 flush_op = _flush_fun(cache.value())