GitLab

PiperOrigin-RevId: 382547358

PiperOrigin-RevId: 382462213

PiperOrigin-RevId: 382422745

PiperOrigin-RevId: 382408293

PiperOrigin-RevId: 382396878

PiperOrigin-RevId: 382382393

PiperOrigin-RevId: 382240268

PiperOrigin-RevId: 382236941

PiperOrigin-RevId: 382051540

PiperOrigin-RevId: 382051212

PiperOrigin-RevId: 381908843

PiperOrigin-RevId: 381869167

PiperOrigin-RevId: 381540013

PiperOrigin-RevId: 381516130

PiperOrigin-RevId: 381396116

PiperOrigin-RevId: 381368435

PiperOrigin-RevId: 381367878

PiperOrigin-RevId: 381366438

Adds a feature to process a dictionary for `input_paths` in `DataConfig`, to allow combining multiple datasets using a user defined combine_fn.

PiperOrigin-RevId: 381363688

This allows the exported TF Lite MoveNet model to be able to run on device GPU
using OpenCL/OpenGL.

PiperOrigin-RevId: 381343791

Add model export config to RetinaNet, which optionally cast model outputs to floats, and normalize output box coordinates.

PiperOrigin-RevId: 381334851

over new parameters in the CenterNet model.

PiperOrigin-RevId: 381331623

PiperOrigin-RevId: 381307288

PiperOrigin-RevId: 381302776

PiperOrigin-RevId: 381301573

PiperOrigin-RevId: 381152422

PiperOrigin-RevId: 381089283

In nlp/train.py and vision/beta/train.py, certain flags are marked as required. Additionally, in certain functions, error messages are improved if a necessary flag is not specified, which is a fallback in case a file calling define_flags() does not mark the necessary flags are required. Previously if any of these flags were not specified, it would crash with a cryptic error message, making it hard to tell what went wrong.

In a subsequent change, I will mark flags as required in more files which call define_flags().

PiperOrigin-RevId: 381066985

Before, it returned None. But almost every use of get_distribution_strategy() assumes an actual strategy is returned and crashes when None is returned. Returning the default strategy fixes these issues and is equivalent to using no strategy, as the default strategy is always in effect when no other strategy is used.

PiperOrigin-RevId: 380951055

Adds a video classification export module for example. The module is subject to change in the near future.

PiperOrigin-RevId: 380907598

PiperOrigin-RevId: 380858205

Some parts of the forward pass would previously overflow in float16. Such parts are now done in float32.

PiperOrigin-RevId: 380857663

PiperOrigin-RevId: 380687139

PiperOrigin-RevId: 380650647

PiperOrigin-RevId: 380470496