Searched refs:BroadcastToFeatureDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | unfuse_batch_norm.cc | 36 Value BroadcastToFeatureDim(Location loc, RankedTensorType result_type, in BroadcastToFeatureDim() function
147 BroadcastToFeatureDim(bn_op.getLoc(), input_type, bn_op.scale(), in matchAndRewrite()
150 BroadcastToFeatureDim(bn_op.getLoc(), input_type, bn_op.offset(), in matchAndRewrite()
153 BroadcastToFeatureDim(bn_op.getLoc(), input_type, bn_op.mean(), in matchAndRewrite()
155 auto broadcast_stddev = BroadcastToFeatureDim( in matchAndRewrite()
|