Home
last modified time | relevance | path

Searched refs:IsFusedBatchNormEx (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_factory.cc60 if (IsFusedBatchNormEx(node)) { in GetTransposer()
Dgeneric_layout_optimizer_transposer.cc941 DCHECK(IsFusedBatchNormEx(*node->node())); in TransposeNode()
1976 IsFusedBatchNormEx(node) || IsFusedBatchNormGrad(node) || in IsLayoutSensitiveOp()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h88 bool IsFusedBatchNormEx(const NodeDef& node);
Dop_types.cc288 bool IsFusedBatchNormEx(const NodeDef& node) { in IsFusedBatchNormEx() function