| 3630 | } |
| 3631 | |
| 3632 | XlaOp BatchNormInference(const XlaOp operand, const XlaOp scale, |
| 3633 | const XlaOp offset, const XlaOp mean, |
| 3634 | const XlaOp variance, float epsilon, |
| 3635 | int64 feature_index) { |
| 3636 | return operand.builder()->BatchNormInference( |
| 3637 | operand, scale, offset, mean, variance, epsilon, feature_index); |
| 3638 | } |
| 3639 | |
| 3640 | XlaOp BatchNormGrad(const XlaOp operand, const XlaOp scale, |
| 3641 | const XlaOp batch_mean, const XlaOp batch_var, |