Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc155 auto broadcast_stddev = BroadcastToFeatureDim( in matchAndRewrite() local
165 rewriter.create<mhlo::DivOp>(bn_op.getLoc(), result, broadcast_stddev); in matchAndRewrite()