xla::BatchNormExpander Class Reference

#include <batchnorm_expander.h>

Inherits HloPassInterface.

Detailed Description

Todo:

See how the function work

Google docs:

A pass which rewrites batch norm operations into more operations. Breaking a big operation into smaller operations helps leverage our generic fusion logic.

---

The documentation for this class was generated from the following files:

• tensorflow/tensorflow/compiler/xla/service/batchnorm_expander.h
• tensorflow/tensorflow/compiler/xla/service/batchnorm_expander.cc