View source on GitHub
|
tf.keras.Operation(
name=None
)
Methods
call
call(
*args, **kwargs
)
compute_output_spec
compute_output_spec(
*args, **kwargs
)
from_config
@classmethodfrom_config( config )
Creates an operation from its config.
This method is the reverse of get_config, capable of instantiating the
same operation from the config dictionary.
if "dtype" in config and isinstance(config["dtype"], dict):
policy = dtype_policies.deserialize(config["dtype"])
| Args | |
|---|---|
config
|
A Python dictionary, typically the output of get_config.
|
| Returns | |
|---|---|
| An operation instance. |
get_config
get_config()
Returns the config of the object.
An object config is a Python dictionary (serializable) containing the information needed to re-instantiate it.
quantized_call
quantized_call(
*args, **kwargs
)
rematerialized_call
rematerialized_call(
fn, *args, **kwargs
)
Enable rematerialization dynamically for an operation's call method.
| Args | |
|---|---|
fn
|
The original call or quantized_call method of an operation.
|
| Returns | |
|---|---|
| Rematerialized method. |
symbolic_call
symbolic_call(
*args, **kwargs
)
__call__
__call__(
*args, **kwargs
)
Call self as a function.
View source on GitHub