| 3620 | } |
| 3621 | |
| 3622 | XlaOp BatchNormTraining(const XlaOp operand, const XlaOp scale, |
| 3623 | const XlaOp offset, const XlaOp side_input, |
| 3624 | float epsilon, int64 feature_index, |
| 3625 | size_t reserve_space_size, bool use_reserve_space, |
| 3626 | bool is_activation_relu) { |
| 3627 | return operand.builder()->BatchNormTraining( |
| 3628 | operand, scale, offset, side_input, epsilon, feature_index, |
| 3629 | reserve_space_size, use_reserve_space, is_activation_relu); |
| 3630 | } |
| 3631 | |
| 3632 | XlaOp BatchNormInference(const XlaOp operand, const XlaOp scale, |
| 3633 | const XlaOp offset, const XlaOp mean, |