Searched refs:separate_compiled_gradients (Results 1 – 4 of 4) sorted by relevance
42 def experimental_jit_scope(compile_ops=True, separate_compiled_gradients=False): argument132 attr_value_pb2.AttrValue(b=bool(separate_compiled_gradients))
239 with jit.experimental_jit_scope(True, separate_compiled_gradients=True):243 with jit.experimental_jit_scope(True, separate_compiled_gradients=True):293 compiled=True, noinline=True, separate_compiled_gradients=True)
270 separate_compiled_gradients=False): argument298 separate_compiled_gradients=separate_compiled_gradients):
1235 separate_compiled_gradients = kwargs.pop("separate_compiled_gradients", None)1239 b=bool(separate_compiled_gradients))