Searched refs:Broadcast1DToFeatureDim (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 505 static Value Broadcast1DToFeatureDim(Location loc, Value broadcast_to, in Broadcast1DToFeatureDim() function
1134 auto bias_broadcast = Broadcast1DToFeatureDim(loc, op.value(), op.bias(), in matchAndRewrite()
1876 Broadcast1DToFeatureDim(loc, act, mean, feature_dim, rewriter)); in matchAndRewrite()
1886 Broadcast1DToFeatureDim(loc, act, scaled_grad, feature_dim, in matchAndRewrite()
|