Searched refs:compile_ops (Results 1 – 3 of 3) sorted by relevance
39 def experimental_jit_scope(compile_ops=True, separate_compiled_gradients=False): argument80 if callable(compile_ops):82 return attr_value_pb2.AttrValue(b=compile_ops(node_def))84 xla_compile = attr_value_pb2.AttrValue(b=compile_ops)
260 compile_ops=use_xla_jit,
1877 compile_ops = True1880 def compile_ops(node_def): function1886 with jit.experimental_jit_scope(compile_ops=compile_ops):