Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc146 auto broadcast_scale = in matchAndRewrite() local
163 rewriter.create<mhlo::MulOp>(bn_op.getLoc(), result, broadcast_scale); in matchAndRewrite()