tfp.experimental.nn.util.tfcompile
Stay organized with collections
Save and categorize content based on your preferences.
Centralizes TF compilation related options.
tfp.experimental.nn.util.tfcompile(
func=None, tf_function=True, xla_best_effort=True, xla_compile_all=False
)
Args |
func
|
Python callable to wrapped with the specified TF compilation
directives.
Default value: None .
|
tf_function
|
bool representing whether the resulting function should be
tf.function decoreated.
Default value: True .
|
xla_best_effort
|
bool representing whether XLA auto-clustering compilation
should be performed. (This argument is ignored if the function is executed
eagerly.)
Default value: True .
|
xla_compile_all
|
bool representing whether XLA compilation should be
performed. (This argument overrides both tf_function and
xla_best_effort .
Default value: False .
|
Returns |
wrapped_func
|
A Python callable with the specified compilation directives
embedded.
|
Example Usage
tfn = tfp.experimental.nn
# Use style #1.
@tfn.util.tfcompile(xla_compile_all=True)
def foo(...):
...
# Use style #2.
def foo(...):
...
foo = tfn.util.tfcompile(xla_compile_all=True)(foo)
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2023-11-21 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2023-11-21 UTC."],[],[]]