Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dunfuse_batch_norm.cc152 auto broadcast_mean = in matchAndRewrite() local
161 broadcast_mean); in matchAndRewrite()
/external/tensorflow/tensorflow/python/keras/
Dbackend.py2920 broadcast_mean = array_ops.reshape(mean, target_shape)
2931 normed = nn.batch_normalization(x, broadcast_mean, broadcast_var,