tf.types.experimental.AtomicFunction

Base class for graph functions.

Inherits From: Callable

An AtomicFunction encapsulates a single graph function definition.

AtomicFunction can be called directly only if no captures are needed according to the FunctionType. If captures are present, please use call_with_captures instead.

AtomicFunction does not support gradients. Please use the parent ConcreteFunction if you need gradient support.

function_type Returns a FunctionType describing this callable.

Methods

call_with_captures

View source

Calls this AtomicFunction with captures as defined by its FunctionType.

Args
args Tuple containing positional arguments
kwargs Dict containing keyword arguments
captures Tuple of tensors supplying captured tensor values.

Returns
A structured output value based on the inputs.

__call__

View source

Executes this callable.

This behaves like a regular op - in eager mode, it immediately starts execution, returning results. In graph mode, it creates ops which return symbolic TensorFlow values (like tf.Tensor, tf.data.Dataset, etc.). For example, tf.function callables typically generate a tf.raw_ops.PartitionedCall op, but not always - the exact operations being generated are an internal implementation detail.

Args
*args positional argument for this call
**kwargs keyword arguments for this call

Returns
The execution results.