pkg syscall (darwin-arm64), type Log2phys_t struct
pkg syscall (darwin-arm64), type Log2phys_t struct, Contigbytes int64
pkg syscall (darwin-arm64), type Log2phys_t struct, Devoffset int64
pkg syscall (darwin-arm64), type Log2phys_t struct, Flags uint32
pkg syscall (darwin-arm64), type Msghdr struct
pkg syscall (darwin-arm64), type Msghdr struct, Control *uint8

                                            mean_var_type, mean_var_type};
      auto bn_train_op = rewriter.create<mhlo::BatchNormTrainingOp>(
          op.getLoc(), operand_types, bn_train_input, op.getScale(),
          op.getOffset(), op.getEpsilon(), feature_dim.getInt());
      // HLO op outputs a tuple of tensors. Extract those results.
      Value y_out = bn_train_op.getResult(0);
      Value batch_mean = bn_train_op.getResult(1);