Searched refs:outside_compilation (Results 1 – 5 of 5) sorted by relevance
284 return tpu.outside_compilation(host_computation)382 return tpu.outside_compilation(host_computation)
1285 flush_op = tpu.outside_compilation(_flush_fun, cache.value())1502 trace_op = tpu.outside_compilation(
472 def outside_compilation(computation, *args, **kwargs): function
525 string outside_compilation; in FunctionalizeLoop() local527 &outside_compilation) in FunctionalizeLoop()529 builder.Attr(kXlaOutsideCompilationAttrName, outside_compilation); in FunctionalizeLoop()
781 string outside_compilation; in BuildIfNode() local783 &outside_compilation) in BuildIfNode()785 builder.Attr(kXlaOutsideCompilationAttrName, outside_compilation); in BuildIfNode()