tf.distribute.experimental.TPUStrategy

Install Learn

Introduction

New to TensorFlow?
TensorFlow

The core open source ML library
For JavaScript

TensorFlow.js for ML using JavaScript
For Mobile & Edge

TensorFlow Lite for mobile and edge devices
For Production

TensorFlow Extended for end-to-end ML components

API

TensorFlow (v2.9.0)
r1.15
Versions…

TensorFlow.js
TensorFlow Lite
TFX

Resources

Models & datasets

Pre-trained models and datasets built by Google and the community
Tools

Ecosystem of tools to help you use TensorFlow
Libraries & extensions

Libraries and extensions built on TensorFlow
TensorFlow Certificate program

Differentiate yourself by demonstrating your ML proficiency
Learn ML

Educational resources to learn the fundamentals of ML with TensorFlow
Responsible AI

Resources and tools to integrate Responsible AI practices into your ML workflow

Community

Join

Explore ways to get involved
Blog

Stay up to date with all things TensorFlow
Forum ↗

Discussion platform for the TensorFlow community
Groups

User groups, interest groups and mailing lists
Contribute

Guide for contributing to code and documentation

Why TensorFlow

About
Case studies

GitHub

TensorFlow Core v2.3.0

Overview Python C++ Java

Install
Learn
- More
API
- More
- Overview
- Python
- C++
- Java
Resources
- More
Community
- More
Why TensorFlow
- More
GitHub

Overview
All Symbols
Python v2.3.0
tf
tf.audio
tf.autodiff
- Overview
- ForwardAccumulator
tf.autograph
- Overview
- set_verbosity
- to_code
- to_graph
- trace
- experimental
tf.bitwise
tf.compat
- Overview
- as_bytes
- as_str
- as_str_any
- as_text
- dimension_at_index
- dimension_value
- forward_compatibility_horizon
- forward_compatible
- path_to_str
- v1
  - Overview
  - add_check_numerics_ops
  - add_to_collection
  - add_to_collections
  - all_variables
  - argmax
  - argmin
  - arg_max
  - arg_min
  - assert_equal
  - assert_greater
  - assert_greater_equal
  - assert_integer
  - assert_less
  - assert_less_equal
  - assert_near
  - assert_negative
  - assert_none_equal
  - assert_non_negative
  - assert_non_positive
  - assert_positive
  - assert_rank
  - assert_rank_at_least
  - assert_rank_in
  - assert_scalar
  - assert_type
  - assert_variables_initialized
  - assign
  - assign_add
  - assign_sub
  - AttrValue
  - AttrValue.ListValue
  - batch_gather
  - batch_scatter_update
  - batch_to_space
  - batch_to_space_nd
  - bincount
  - boolean_mask
  - case
  - clip_by_average_norm
  - colocate_with
  - cond
  - ConditionalAccumulator
  - ConditionalAccumulatorBase
  - ConfigProto
  - ConfigProto.DeviceCountEntry
  - ConfigProto.Experimental
  - confusion_matrix
  - constant
  - container
  - control_flow_v2_enabled
  - convert_to_tensor
  - convert_to_tensor_or_indexed_slices
  - convert_to_tensor_or_sparse_tensor
  - count_nonzero
  - count_up_to
  - create_partitioned_variables
  - decode_csv
  - decode_raw
  - delete_session_tensor
  - depth_to_space
  - device
  - DeviceSpec
  - Dimension
  - disable_control_flow_v2
  - disable_eager_execution
  - disable_resource_variables
  - disable_tensor_equality
  - disable_v2_behavior
  - disable_v2_tensorshape
  - enable_control_flow_v2
  - enable_eager_execution
  - enable_resource_variables
  - enable_tensor_equality
  - enable_v2_behavior
  - enable_v2_tensorshape
  - Event
  - executing_eagerly
  - executing_eagerly_outside_functions
  - expand_dims
  - extract_image_patches
  - FixedLengthRecordReader
  - fixed_size_partitioner
  - floor_div
  - foldl
  - foldr
  - gather
  - gather_nd
  - get_collection
  - get_collection_ref
  - get_default_graph
  - get_default_session
  - get_local_variable
  - get_seed
  - get_session_handle
  - get_session_tensor
  - get_variable
  - get_variable_scope
  - global_variables
  - global_variables_initializer
  - GPUOptions
  - GPUOptions.Experimental
  - GPUOptions.Experimental.VirtualDevices
  - gradients
  - GraphDef
  - GraphKeys
  - GraphOptions
  - hessians
  - HistogramProto
  - IdentityReader
  - initialize_all_tables
  - initialize_all_variables
  - initialize_local_variables
  - initialize_variables
  - InteractiveSession
  - is_variable_initialized
  - LMDBReader
  - load_file_system_library
  - local_variables
  - local_variables_initializer
  - LogMessage
  - make_template
  - map_fn
  - MetaGraphDef
  - MetaGraphDef.CollectionDefEntry
  - MetaGraphDef.MetaInfoDef
  - MetaGraphDef.MetaInfoDef.FunctionAliasesEntry
  - MetaGraphDef.SignatureDefEntry
  - min_max_variable_partitioner
  - model_variables
  - moving_average_variables
  - multinomial
  - NameAttrList
  - NameAttrList.AttrEntry
  - NodeDef
  - NodeDef.AttrEntry
  - NodeDef.ExperimentalDebugInfo
  - norm
  - no_regularizer
  - ones_like
  - OptimizerOptions
  - op_scope
  - pad
  - parse_example
  - parse_single_example
  - placeholder
  - placeholder_with_default
  - Print
  - py_func
  - quantize_v2
  - random_normal_initializer
  - random_poisson
  - random_uniform_initializer
  - ReaderBase
  - reduce_all
  - reduce_any
  - reduce_join
  - reduce_logsumexp
  - reduce_max
  - reduce_mean
  - reduce_min
  - reduce_prod
  - reduce_sum
  - report_uninitialized_variables
  - reset_default_graph
  - resource_variables_enabled
  - reverse_sequence
  - RunMetadata
  - RunMetadata.FunctionGraphs
  - RunOptions
  - RunOptions.Experimental
  - RunOptions.Experimental.RunHandlerPoolOptions
  - scalar_mul
  - scan
  - scatter_add
  - scatter_div
  - scatter_max
  - scatter_min
  - scatter_mul
  - scatter_nd_add
  - scatter_nd_sub
  - scatter_nd_update
  - scatter_sub
  - scatter_update
  - serialize_many_sparse
  - serialize_sparse
  - Session
  - SessionLog
  - setdiff1d
  - set_random_seed
  - shape
  - size
  - space_to_batch
  - space_to_depth
  - SparseConditionalAccumulator
  - SparseTensorValue
  - sparse_add
  - sparse_concat
  - sparse_matmul
  - sparse_merge
  - sparse_placeholder
  - sparse_reduce_max
  - sparse_reduce_max_sparse
  - sparse_reduce_sum
  - sparse_reduce_sum_sparse
  - sparse_segment_mean
  - sparse_segment_sqrt_n
  - sparse_segment_sum
  - sparse_split
  - sparse_to_dense
  - squeeze
  - string_split
  - string_to_hash_bucket
  - string_to_number
  - substr
  - Summary
  - Summary.Audio
  - Summary.Image
  - Summary.Value
  - SummaryMetadata
  - SummaryMetadata.PluginData
  - tables_initializer
  - TensorInfo
  - TensorInfo.CompositeTensor
  - TensorInfo.CooSparse
  - TextLineReader
  - TFRecordReader
  - to_bfloat16
  - to_complex128
  - to_complex64
  - to_double
  - to_float
  - to_int32
  - to_int64
  - trainable_variables
  - transpose
  - truncated_normal_initializer
  - tuple
  - uniform_unit_scaling_initializer
  - Variable
  - VariableAggregation
  - VariableScope
  - variables_initializer
  - variable_axis_size_partitioner
  - variable_creator_scope
  - variable_op_scope
  - variable_scope
  - verify_tensor_all_finite
  - where
  - while_loop
  - WholeFileReader
  - wrap_function
  - zeros_like
  - app
    Overview
    run
  - audio
    Overview
  - autograph
    Overview
    to_code
    to_graph
    
    experimental
    Overview
  - bitwise
    Overview
  - compat
    Overview
  - config
    Overview
    
    experimental
    Overview
    
    optimizer
    Overview
    
    threading
    Overview
  - data
    Overview
    Dataset
    FixedLengthRecordDataset
    get_output_classes
    get_output_shapes
    get_output_types
    Iterator
    make_initializable_iterator
    make_one_shot_iterator
    TextLineDataset
    TFRecordDataset
    
    experimental
    Overview
    choose_from_datasets
    Counter
    CsvDataset
    make_batched_features_dataset
    make_csv_dataset
    map_and_batch_with_legacy_function
    RaggedTensorStructure
    RandomDataset
    sample_from_datasets
    SparseTensorStructure
    SqlDataset
    StatsAggregator
    TensorArrayStructure
    TensorStructure
    
    service
    Overview
  - debugging
    Overview
    assert_shapes
    
    experimental
    Overview
  - distribute
    Overview
    get_loss_reduction
    MirroredStrategy
    OneDeviceStrategy
    Strategy
    StrategyExtended
    
    cluster_resolver
    Overview
    
    experimental
    Overview
    CentralStorageStrategy
    MultiWorkerMirroredStrategy
    ParameterServerStrategy
    TPUStrategy
  - distributions
    Overview
    Bernoulli
    Beta
    Categorical
    Dirichlet
    DirichletMultinomial
    Distribution
    Exponential
    Gamma
    kl_divergence
    Laplace
    Multinomial
    Normal
    RegisterKL
    ReparameterizationType
    StudentT
    Uniform
  - dtypes
    Overview
  - errors
    Overview
    error_code_from_exception_type
    exception_type_from_error_code
    raise_exception_on_not_ok_status
  - estimator
    Overview
    BaselineClassifier
    BaselineEstimator
    BaselineRegressor
    classifier_parse_example_spec
    DNNClassifier
    DNNEstimator
    DNNLinearCombinedClassifier
    DNNLinearCombinedEstimator
    DNNLinearCombinedRegressor
    DNNRegressor
    Estimator
    LinearClassifier
    LinearEstimator
    LinearRegressor
    regressor_parse_example_spec
    
    experimental
    Overview
    dnn_logit_fn_builder
    KMeans
    linear_logit_fn_builder
    
    export
    Overview
    
    inputs
    Overview
    numpy_input_fn
    pandas_input_fn
    
    tpu
    Overview
    InputPipelineConfig
    RunConfig
    TPUConfig
    TPUEstimator
    TPUEstimatorSpec
    
    experimental
    Overview
    EmbeddingConfigSpec
  - experimental
    Overview
    output_all_intermediates
  - feature_column
    Overview
    categorical_column_with_vocabulary_file
    input_layer
    linear_model
    make_parse_example_spec
    shared_embedding_columns
  - flags
    Overview
    adopt_module_key_flags
    ArgumentParser
    ArgumentSerializer
    BaseListParser
    BooleanFlag
    BooleanParser
    CantOpenFlagFileError
    CsvListSerializer
    declare_key_flag
    DEFINE
    DEFINE_alias
    DEFINE_bool
    DEFINE_enum
    DEFINE_enum_class
    DEFINE_flag
    DEFINE_float
    DEFINE_integer
    DEFINE_list
    DEFINE_multi
    DEFINE_multi_enum
    DEFINE_multi_enum_class
    DEFINE_multi_float
    DEFINE_multi_integer
    DEFINE_multi_string
    DEFINE_spaceseplist
    DEFINE_string
    disclaim_key_flags
    doc_to_help
    DuplicateFlagError
    EnumClassFlag
    EnumClassParser
    EnumFlag
    EnumParser
    Error
    Flag
    FlagHolder
    FlagNameConflictsWithMethodError
    FLAGS
    FlagValues
    flag_dict_to_args
    FloatParser
    get_help_width
    IllegalFlagValueError
    IntegerParser
    ListParser
    ListSerializer
    mark_bool_flags_as_mutual_exclusive
    mark_flags_as_mutual_exclusive
    mark_flags_as_required
    mark_flag_as_required
    MultiEnumClassFlag
    MultiFlag
    multi_flags_validator
    register_multi_flags_validator
    register_validator
    text_wrap
    UnparsedFlagAccessError
    UnrecognizedFlagError
    ValidationError
    validator
    WhitespaceSeparatedListParser
    
    tf_decorator
    Overview
    make_decorator
    rewrap
    TFDecorator
    unwrap
    
    tf_stack
    Overview
    CurrentModuleFilter
    extract_stack
    FrameSummary
    StackSummary
    StackTraceFilter
    StackTraceMapper
    StackTraceTransform
  - gfile
    Overview
    Copy
    DeleteRecursively
    Exists
    FastGFile
    Glob
    IsDirectory
    ListDirectory
    MakeDirs
    MkDir
    Remove
    Rename
    Stat
    Walk
  - graph_util
    Overview
    convert_variables_to_constants
    extract_sub_graph
    must_run_on_cpu
    remove_training_nodes
    tensor_shape_from_node_def_name
  - image
    Overview
    crop_and_resize
    draw_bounding_boxes
    extract_glimpse
    resize
    ResizeMethod
    resize_area
    resize_bicubic
    resize_bilinear
    resize_image_with_pad
    resize_nearest_neighbor
    sample_distorted_bounding_box
  - initializers
    Overview
    he_normal
    he_uniform
    lecun_normal
    lecun_uniform
  - io
    Overview
    TFRecordCompressionType
    tf_record_iterator
    
    gfile
    Overview
  - keras
    Overview
    
    activations
    Overview
    
    applications
    Overview
    
    densenet
    Overview
    
    efficientnet
    Overview
    
    imagenet_utils
    Overview
    
    inception_resnet_v2
    Overview
    
    inception_v3
    Overview
    
    mobilenet
    Overview
    
    mobilenet_v2
    Overview
    
    nasnet
    Overview
    
    resnet
    Overview
    
    resnet50
    Overview
    
    resnet_v2
    Overview
    
    vgg16
    Overview
    
    vgg19
    Overview
    
    xception
    Overview
    
    backend
    Overview
    get_session
    name_scope
    set_session
    
    callbacks
    Overview
    TensorBoard
    
    constraints
    Overview
    
    datasets
    Overview
    
    boston_housing
    Overview
    
    cifar10
    Overview
    
    cifar100
    Overview
    
    fashion_mnist
    Overview
    
    imdb
    Overview
    
    mnist
    Overview
    
    reuters
    Overview
    
    estimator
    Overview
    model_to_estimator
    
    experimental
    Overview
    export_saved_model
    load_from_saved_model
    
    initializers
    Overview
    Constant
    glorot_normal
    glorot_uniform
    he_normal
    he_uniform
    Identity
    lecun_normal
    lecun_uniform
    Ones
    Orthogonal
    RandomNormal
    RandomUniform
    TruncatedNormal
    VarianceScaling
    Zeros
    
    layers
    Overview
    BatchNormalization
    CuDNNGRU
    CuDNNLSTM
    DenseFeatures
    disable_v2_dtype_behavior
    enable_v2_dtype_behavior
    GRU
    GRUCell
    LSTM
    LSTMCell
    
    experimental
    Overview
    
    preprocessing
    Overview
    CategoryEncoding
    IntegerLookup
    Normalization
    StringLookup
    TextVectorization
    
    losses
    Overview
    
    metrics
    Overview
    
    mixed_precision
    Overview
    
    experimental
    Overview
    
    models
    Overview
    
    optimizers
    Overview
    
    schedules
    Overview
    
    preprocessing
    Overview
    
    image
    Overview
    
    sequence
    Overview
    
    text
    Overview
    
    regularizers
    Overview
    
    utils
    Overview
    
    wrappers
    Overview
    
    scikit_learn
    Overview
  - layers
    Overview
    AveragePooling1D
    AveragePooling2D
    AveragePooling3D
    average_pooling1d
    average_pooling2d
    average_pooling3d
    BatchNormalization
    batch_normalization
    Conv1D
    conv1d
    Conv2D
    conv2d
    Conv2DTranspose
    conv2d_transpose
    Conv3D
    conv3d
    Conv3DTranspose
    conv3d_transpose
    Dense
    dense
    Dropout
    dropout
    Flatten
    flatten
    Layer
    MaxPooling1D
    MaxPooling2D
    MaxPooling3D
    max_pooling1d
    max_pooling2d
    max_pooling3d
    SeparableConv1D
    SeparableConv2D
    separable_conv1d
    separable_conv2d
    
    experimental
    Overview
    keras_style_scope
    set_keras_style
  - linalg
    Overview
    l2_normalize
    
    experimental
    Overview
  - lite
    Overview
    OpHint
    OpHint.OpHintArgumentTracker
    TFLiteConverter
    TocoConverter
    toco_convert
    
    constants
    Overview
    
    experimental
    Overview
    convert_op_hints_to_stubs
    get_potentially_supported_ops
    
    nn
    Overview
    dynamic_rnn
    TFLiteLSTMCell
    TfLiteRNNCell
  - logging
    Overview
    debug
    error
    fatal
    flush
    get_verbosity
    info
    log
    log_every_n
    log_first_n
    log_if
    set_verbosity
    TaskLevelStatusMessage
    vlog
    warn
    warning
  - lookup
    Overview
    StaticHashTable
    StaticVocabularyTable
    
    experimental
    Overview
  - losses
    Overview
    absolute_difference
    add_loss
    compute_weighted_loss
    cosine_distance
    get_losses
    get_regularization_loss
    get_regularization_losses
    get_total_loss
    hinge_loss
    huber_loss
    log_loss
    mean_pairwise_squared_error
    mean_squared_error
    Reduction
    sigmoid_cross_entropy
    softmax_cross_entropy
    sparse_softmax_cross_entropy
  - manip
    Overview
  - math
    Overview
    in_top_k
    log_softmax
    softmax
    
    special
    Overview
  - metrics
    Overview
    accuracy
    auc
    average_precision_at_k
    false_negatives
    false_negatives_at_thresholds
    false_positives
    false_positives_at_thresholds
    mean
    mean_absolute_error
    mean_cosine_distance
    mean_iou
    mean_per_class_accuracy
    mean_relative_error
    mean_squared_error
    mean_tensor
    percentage_below
    precision
    precision_at_k
    precision_at_thresholds
    precision_at_top_k
    recall
    recall_at_k
    recall_at_thresholds
    recall_at_top_k
    root_mean_squared_error
    sensitivity_at_specificity
    sparse_average_precision_at_k
    sparse_precision_at_k
    specificity_at_sensitivity
    true_negatives
    true_negatives_at_thresholds
    true_positives
    true_positives_at_thresholds
  - mixed_precision
    Overview
    
    experimental
    Overview
  - mlir
    Overview
    
    experimental
    Overview
  - nest
    Overview
  - nn
    Overview
    avg_pool
    batch_norm_with_global_normalization
    bidirectional_dynamic_rnn
    conv1d
    conv2d
    conv2d_backprop_filter
    conv2d_backprop_input
    conv2d_transpose
    conv3d
    conv3d_backprop_filter
    conv3d_transpose
    convolution
    crelu
    ctc_beam_search_decoder
    ctc_loss
    ctc_loss_v2
    depthwise_conv2d
    depthwise_conv2d_native
    dilation2d
    dropout
    dynamic_rnn
    embedding_lookup
    embedding_lookup_sparse
    erosion2d
    fractional_avg_pool
    fractional_max_pool
    fused_batch_norm
    max_pool
    max_pool_with_argmax
    moments
    nce_loss
    pool
    quantized_avg_pool
    quantized_conv2d
    quantized_max_pool
    quantized_relu_x
    raw_rnn
    relu_layer
    safe_embedding_lookup_sparse
    sampled_softmax_loss
    separable_conv2d
    sigmoid_cross_entropy_with_logits
    softmax_cross_entropy_with_logits
    softmax_cross_entropy_with_logits_v2
    sparse_softmax_cross_entropy_with_logits
    static_bidirectional_rnn
    static_rnn
    static_state_saving_rnn
    sufficient_statistics
    weighted_cross_entropy_with_logits
    weighted_moments
    xw_plus_b
    
    rnn_cell
    Overview
    BasicLSTMCell
    BasicRNNCell
    DeviceWrapper
    DropoutWrapper
    GRUCell
    LSTMCell
    LSTMStateTuple
    MultiRNNCell
    ResidualWrapper
    RNNCell
  - profiler
    Overview
    AdviceProto
    AdviceProto.Checker
    AdviceProto.CheckersEntry
    advise
    GraphNodeProto
    GraphNodeProto.InputShapesEntry
    MultiGraphNodeProto
    OpLogProto
    OpLogProto.IdToStringEntry
    profile
    ProfileOptionBuilder
    Profiler
    write_op_log
  - python_io
    Overview
  - quantization
    Overview
  - queue
    Overview
  - ragged
    Overview
    constant_value
    placeholder
    RaggedTensorValue
  - random
    Overview
    stateless_multinomial
    
    experimental
    Overview
  - raw_ops
    Overview
  - resource_loader
    Overview
    get_data_files_path
    get_path_to_datafile
    get_root_dir_with_all_resources
    load_resource
    readahead_file_path
  - saved_model
    Overview
    Builder
    build_signature_def
    build_tensor_info
    classification_signature_def
    contains_saved_model
    get_tensor_from_tensor_info
    is_valid_signature
    load
    main_op_with_restore
    predict_signature_def
    regression_signature_def
    simple_save
    
    builder
    Overview
    
    constants
    Overview
    
    experimental
    Overview
    
    loader
    Overview
    
    main_op
    Overview
    main_op
    
    signature_constants
    Overview
    
    signature_def_utils
    Overview
    MethodNameUpdater
    
    tag_constants
    Overview
    
    utils
    Overview
  - sets
    Overview
  - signal
    Overview
  - sparse
    Overview
  - spectral
    Overview
  - strings
    Overview
    length
    split
    substr
  - summary
    Overview
    all_v2_summary_ops
    audio
    FileWriter
    FileWriterCache
    get_summary_description
    histogram
    image
    initialize
    merge
    merge_all
    scalar
    SummaryDescription
    TaggedRunMetadata
    tensor_summary
    text
  - sysconfig
    Overview
  - test
    Overview
    assert_equal_graph_def
    compute_gradient
    compute_gradient_error
    get_temp_dir
    StubOutForTesting
    test_src_dir_path
  - tpu
    Overview
    batch_parallel
    bfloat16_scope
    core
    CrossShardOptimizer
    cross_replica_sum
    initialize_system
    outside_compilation
    PaddingSpec
    replicate
    rewrite
    shard
    shutdown_system
    
    experimental
    Overview
    AdagradParameters
    AdamParameters
    embedding_column
    FtrlParameters
    shared_embedding_columns
    StochasticGradientDescentParameters
    
    embedding
    Overview
  - train
    Overview
    AdadeltaOptimizer
    AdagradDAOptimizer
    AdagradOptimizer
    AdamOptimizer
    add_queue_runner
    assert_global_step
    basic_train_loop
    batch
    batch_join
    Checkpoint
    checkpoint_exists
    ChiefSessionCreator
    cosine_decay
    cosine_decay_restarts
    create_global_step
    do_quantize_training_on_graphdef
    exponential_decay
    export_meta_graph
    FtrlOptimizer
    generate_checkpoint_state_proto
    get_checkpoint_mtimes
    get_global_step
    get_or_create_global_step
    global_step
    GradientDescentOptimizer
    import_meta_graph
    init_from_checkpoint
    input_producer
    inverse_time_decay
    limit_epochs
    linear_cosine_decay
    LooperThread
    maybe_batch
    maybe_batch_join
    maybe_shuffle_batch
    maybe_shuffle_batch_join
    MomentumOptimizer
    MonitoredSession
    MonitoredSession.StepContext
    MonitoredTrainingSession
    natural_exp_decay
    NewCheckpointReader
    noisy_linear_cosine_decay
    Optimizer
    piecewise_constant
    polynomial_decay
    ProximalAdagradOptimizer
    ProximalGradientDescentOptimizer
    QueueRunner
    range_input_producer
    remove_checkpoint
    replica_device_setter
    RMSPropOptimizer
    Saver
    SaverDef
    Scaffold
    sdca_fprint
    sdca_optimizer
    sdca_shrink_l1
    SessionCreator
    SessionManager
    shuffle_batch
    shuffle_batch_join
    SingularMonitoredSession
    slice_input_producer
    start_queue_runners
    string_input_producer
    summary_iterator
    Supervisor
    SyncReplicasOptimizer
    update_checkpoint_state
    warm_start
    WorkerSessionCreator
    
    experimental
    Overview
    disable_mixed_precision_graph_rewrite
    enable_mixed_precision_graph_rewrite
    MixedPrecisionLossScaleOptimizer
    
    queue_runner
    Overview
  - user_ops
    Overview
    my_fact
  - version
    Overview
  - xla
    Overview
    
    experimental
    Overview
tf.config
tf.data
tf.debugging
tf.distribute
tf.dtypes
tf.errors
tf.estimator
tf.experimental
- Overview
- async_clear_error
- async_scope
- function_executor_type
- Optional
- dlpack
- tensorrt
tf.feature_column
tf.graph_util
- Overview
- import_graph_def
tf.image
tf.io
tf.keras
tf.linalg
tf.lite
- Overview
- Interpreter
- OpsSet
- Optimize
- RepresentativeDataset
- TargetSpec
- TFLiteConverter
- experimental
  - Overview
  - load_delegate
tf.lookup
tf.math
tf.mixed_precision
- Overview
- experimental
tf.mlir
- Overview
- experimental
  - Overview
  - convert_graph_def
tf.nest
tf.nn
tf.profiler
- Overview
- experimental
  - Overview
  - Profile
  - ProfilerOptions
  - start
  - stop
  - Trace
  - client
    Overview
    monitor
    trace
  - server
    Overview
    start
tf.quantization
tf.queue
tf.ragged
tf.random
tf.raw_ops
tf.saved_model
tf.sets
tf.signal
tf.sparse
tf.strings
tf.summary
tf.sysconfig
tf.test
tf.tpu
- Overview
- experimental
tf.train
tf.version
- Overview
tf.xla
- Overview
- experimental

Introduction
TensorFlow
For JavaScript
For Mobile & Edge
For Production

TensorFlow (v2.9.0)
r1.15
Versions…
TensorFlow.js
TensorFlow Lite
TFX

Models & datasets
Tools
Libraries & extensions
TensorFlow Certificate program
Learn ML
Responsible AI

Join
Blog
Forum ↗
Groups