Skip to main content
Install
Learn
Introduction
New to TensorFlow?
Tutorials
Learn how to use TensorFlow with end-to-end examples
Guide
Learn framework concepts and components
Learn ML
Educational resources to master your path with TensorFlow
API
TensorFlow (v2.16.1)
Versions…
TensorFlow.js
TensorFlow Lite
TFX
Ecosystem
LIBRARIES
TensorFlow.js
Develop web ML applications in JavaScript
TensorFlow Lite
Deploy ML on mobile, microcontrollers and other edge devices
TFX
Build production ML pipelines
All libraries
Create advanced models and extend TensorFlow
RESOURCES
Models & datasets
Pre-trained models and datasets built by Google and the community
Tools
Tools to support and accelerate TensorFlow workflows
Responsible AI
Resources for every stage of the ML workflow
Recommendation systems
Build recommendation systems with open source tools
Community
Groups
User groups, interest groups and mailing lists
Contribute
Guide for contributing to code and documentation
Blog
Stay up to date with all things TensorFlow
Forum
Discussion platform for the TensorFlow community
Why TensorFlow
About
Case studies
/
English
中文 – 简体
GitHub
Sign in
TensorFlow v2.16.1
Overview
Python
C++
Java
More
Install
Learn
More
API
More
Overview
Python
C++
Java
More
Ecosystem
More
Community
More
Why TensorFlow
More
GitHub
Overview
TensorFlow Models
tfm
Overview
tfm.core
Overview
actions
Overview
EMACheckpointing
PruningAction
RecoveryAction
RecoveryCondition
get_eval_actions
get_train_actions
base_task
Overview
DifferentialPrivacyConfig
RuntimeConfig
Task
base_trainer
Overview
ExperimentConfig
Trainer
TrainerConfig
get_runtime_options
config_definitions
Overview
DataConfig
TaskConfig
exp_factory
Overview
get_exp_config
register_config_factory
export_base
Overview
ExportModule
export
get_timestamped_export_dir
file_writers
Overview
write_small_dataset
input_reader
Overview
InputReader
match_files
registry
Overview
lookup
register
savedmodel_checkpoint_manager
Overview
SavedModelCheckpointManager
make_saved_modules_directory_name
task_factory
Overview
get_task
get_task_cls
register_task_cls
tf_example_builder
Overview
BytesValueType
TfExampleBuilder
tf_example_feature_key
Overview
TfExampleFeatureKeyBase
dataclass
train_lib
Overview
OrbitExperimentRunner
maybe_create_best_ckpt_exporter
run_experiment
train_utils
Overview
BestCheckpointExporter
ExperimentParser
ParseConfigOptions
cast_leaf_nested_dict
convert_variables_to_constants_v2_as_graph
create_optimizer
create_trainer
get_leaf_nested_dict
parse_configuration
read_global_step_from_checkpoint
remove_ckpts
save_gin_config
serialize_config
try_count_flops
try_count_params
write_json_summary
write_model_params
write_summary
tfm.hyperparams
Overview
Config
OneOfConfig
ParamsDict
bind