Searched refs:experimental_jit_scope (Results 1 – 6 of 6) sorted by relevance
52 with jit.experimental_jit_scope(use_jit):63 with jit.experimental_jit_scope(True):103 with jit.experimental_jit_scope(True):106 with jit.experimental_jit_scope(True):109 with jit.experimental_jit_scope(True):112 with jit.experimental_jit_scope(True):117 with jit.experimental_jit_scope(True):172 with jit.experimental_jit_scope(True):194 with jit.experimental_jit_scope():216 with jit.experimental_jit_scope():[all …]
42 def experimental_jit_scope(compile_ops=True, separate_compiled_gradients=False): function
34 jit_scope = jit.experimental_jit_scope
295 jit_scope = jit.experimental_jit_scope
42 jit_scope = jit.experimental_jit_scope
166 with jit.experimental_jit_scope():