tf_function constructs a callable that executes a TensorFlow graph created
by tracing the TensorFlow operations in f. This allows the TensorFlow
runtime to apply optimizations and exploit parallelism in the computation
defined by f.
tf_function(f, input_signature = NULL, autograph = TRUE, ...)the function to be compiled
A possibly nested sequence of tf$TensorSpec objects
specifying the shapes and dtypes of the tensors that will be supplied to
this function. If NULL, a separate function is instantiated for each
inferred input signature. If input_signature is specified, every input to
f must be a tensor.
TRUE or FALSE. If TRUE (the default), you can use tensors in
R control flow expressions if, while, for and break and they will
be traced into the tensorflow graph. A guide to getting started and
additional details can be found:
here
additional arguments passed on to tf.function (vary based on
Tensorflow version). See
here for
details.
A guide to getting started with
tf.function can
be found here.