Home
last modified time | relevance | path

Searched refs:IsFusedBatchNormGrad (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_factory.cc64 if (IsFusedBatchNormGrad(node)) { in GetTransposer()
Dgeneric_layout_optimizer_transposer.cc971 DCHECK(IsFusedBatchNormGrad(*node->node())); in TransposeNode()
1976 IsFusedBatchNormEx(node) || IsFusedBatchNormGrad(node) || in IsLayoutSensitiveOp()
Dlayout_optimizer.cc2090 } else if (IsFusedBatchNormGrad(*node)) { in Expand()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h89 bool IsFusedBatchNormGrad(const NodeDef& node);
Dop_types.cc292 bool IsFusedBatchNormGrad(const NodeDef& node) { in IsFusedBatchNormGrad() function