Searched refs:separate_compiled_gradients (Results 1 – 4 of 4) sorted by relevance
39 def experimental_jit_scope(compile_ops=True, separate_compiled_gradients=False): argument90 attr_value_pb2.AttrValue(b=bool(separate_compiled_gradients))
222 with jit.experimental_jit_scope(True, separate_compiled_gradients=True):226 with jit.experimental_jit_scope(True, separate_compiled_gradients=True):274 compiled=True, noinline=True, separate_compiled_gradients=True)
233 separate_compiled_gradients=False): argument261 separate_compiled_gradients=separate_compiled_gradients):
1160 separate_compiled_gradients = kwargs.pop("separate_compiled_gradients", None)1164 b=bool(separate_compiled_gradients))