Home | History | Annotate | Download | only in encoder
      1 /*
      2  * Copyright (c) 2018, Alliance for Open Media. All rights reserved
      3  *
      4  * This source code is subject to the terms of the BSD 2 Clause License and
      5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
      6  * was not distributed with this source code in the LICENSE file, you can
      7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
      8  * Media Patent License 1.0 was not distributed with this source code in the
      9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
     10  */
     11 
     12 #ifndef AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_
     13 #define AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_
     14 
     15 #ifdef __cplusplus
     16 extern "C" {
     17 #endif
     18 
     19 #include "av1/encoder/ml.h"
     20 
     21 #define FEATURE_SIZE 10
     22 #define LABEL_SIZE 16
     23 // nn model for ab partition pruning, 128x128.
     24 static const float av1_ab_partition_nn_weights_128_layer0[FEATURE_SIZE * 64] = {
     25   -0.715251f, -0.015767f, -0.667353f, -0.345255f, 0.177887f,  -0.469759f,
     26   0.426152f,  0.489798f,  0.469865f,  0.773821f,  0.088517f,  0.074585f,
     27   0.838754f,  0.048449f,  -0.007584f, 0.638968f,  0.233305f,  -0.319236f,
     28   -0.257124f, -0.170869f, 0.137180f,  0.114852f,  -0.721241f, -0.947962f,
     29   -0.411298f, 0.494306f,  -0.060435f, -0.648421f, -0.126624f, 0.072686f,
     30   -0.143904f, -0.115839f, -0.175527f, -0.117728f, 0.040686f,  -0.189925f,
     31   0.134361f,  -0.258070f, -0.177558f, 0.158049f,  0.168668f,  -0.062919f,
     32   0.341986f,  0.038100f,  -0.435577f, -0.321255f, 0.203213f,  0.213061f,
     33   0.533304f,  0.359296f,  -0.079558f, 0.004637f,  0.663904f,  0.043779f,
     34   0.383018f,  1.136559f,  -0.084155f, 0.333057f,  -0.199011f, 0.152059f,
     35   -0.078419f, -0.167752f, -0.093651f, 0.083171f,  -0.190143f, 0.086195f,
     36   -0.280632f, -0.160663f, -0.017298f, 0.122628f,  -0.138116f, 0.062927f,
     37   0.222462f,  0.626979f,  0.426928f,  0.117170f,  -0.240457f, 0.053750f,
     38   0.038017f,  0.007359f,  -0.017595f, 0.101407f,  0.332891f,  0.074933f,
     39   0.306498f,  0.219380f,  -0.151638f, -0.247976f, 0.343405f,  0.121256f,
     40   0.049173f,  0.171474f,  -0.139608f, -1.016599f, -0.345553f, -0.901138f,
     41   0.243401f,  0.059928f,  -0.089396f, -0.195565f, 0.364705f,  -0.020400f,
     42   -1.383672f, 0.413018f,  0.536950f,  -0.020904f, -1.335306f, -0.732290f,
     43   0.102885f,  0.315290f,  -0.208521f, -0.081811f, 0.182300f,  0.125712f,
     44   -0.593833f, -0.220639f, -0.314155f, 0.188327f,  0.118503f,  0.524427f,
     45   -1.083859f, -1.130640f, 0.390352f,  -0.045591f, 0.113160f,  -0.009149f,
     46   -0.096183f, 0.115829f,  0.377752f,  0.318396f,  -0.591983f, 0.004797f,
     47   -0.497377f, -0.342248f, 0.079546f,  -0.025249f, -0.295972f, 0.615501f,
     48   -0.464372f, 0.418315f,  -0.173556f, 0.105217f,  0.298073f,  0.082478f,
     49   0.033223f,  0.977341f,  -0.372982f, -0.052337f, 0.154124f,  0.396787f,
     50   0.536654f,  -0.139061f, -0.223702f, 0.229666f,  -0.846766f, 0.107723f,
     51   0.563839f,  -0.483141f, 0.304813f,  -0.765283f, 0.070964f,  0.151101f,
     52   0.275188f,  0.490303f,  1.175892f,  0.085377f,  -0.191200f, 0.544532f,
     53   -0.365075f, 0.167546f,  0.052183f,  -0.220529f, -0.212227f, -0.144988f,
     54   -0.273356f, -0.062023f, 0.103993f,  -0.238493f, -0.161204f, -0.054611f,
     55   -0.166672f, 0.128327f,  0.461751f,  -0.545822f, 0.739798f,  0.594386f,
     56   -0.163192f, -0.332501f, 0.363834f,  -0.065043f, 0.474812f,  -0.138811f,
     57   0.170924f,  -0.778142f, -0.316474f, -0.508065f, -0.039986f, -0.478001f,
     58   0.340591f,  0.041783f,  0.055419f,  0.015155f,  -0.981830f, -1.355237f,
     59   0.347516f,  1.155327f,  0.081319f,  0.274163f,  -0.327230f, -0.113478f,
     60   0.556552f,  -0.055986f, 0.217318f,  -0.445351f, 0.325759f,  0.526547f,
     61   -0.657434f, -0.572214f, -0.037087f, 0.081384f,  0.064518f,  0.014892f,
     62   0.215279f,  1.834504f,  -0.242107f, 0.079810f,  0.129558f,  0.079588f,
     63   -0.035189f, -0.221745f, -0.163414f, 0.043978f,  -1.028662f, -0.623609f,
     64   1.130336f,  0.664661f,  -0.063975f, -0.415863f, 0.018581f,  0.157758f,
     65   0.200570f,  0.063420f,  0.901039f,  -0.746286f, 0.196230f,  -0.290592f,
     66   0.042373f,  -0.502500f, 0.183638f,  0.103394f,  -0.298858f, 0.145436f,
     67   0.196916f,  0.108319f,  -0.448572f, -0.881385f, 0.302497f,  0.121679f,
     68   -0.021327f, 0.025150f,  0.481306f,  -0.359634f, 0.350257f,  -0.228647f,
     69   -0.669860f, 0.260025f,  -0.034182f, 0.619247f,  -0.158826f, -0.405864f,
     70   0.674112f,  -0.027885f, -0.325274f, -0.241492f, 0.036024f,  -0.437685f,
     71   -0.091458f, -0.109295f, -0.350676f, 0.044706f,  0.297059f,  0.016290f,
     72   1.121203f,  1.289062f,  -1.299476f, -1.129221f, 0.103752f,  0.131302f,
     73   -0.263265f, 0.222155f,  -0.229908f, 0.013922f,  -0.226001f, -0.248383f,
     74   -0.004415f, -0.020958f, 0.055634f,  0.086200f,  0.114556f,  -0.184061f,
     75   -0.096210f, -0.146466f, -0.249618f, -0.195998f, 0.088758f,  0.023781f,
     76   -0.264460f, 0.157026f,  -0.235228f, -0.102564f, 0.043463f,  -0.187823f,
     77   -0.257500f, -0.199049f, -0.242210f, 0.030448f,  0.221604f,  0.151804f,
     78   -0.100404f, -0.073931f, 0.144749f,  -0.001572f, -1.438079f, -0.233716f,
     79   0.733422f,  1.727080f,  -0.036397f, 0.027551f,  0.425321f,  0.085703f,
     80   0.031186f,  0.032333f,  -0.675130f, 1.437733f,  -0.202392f, -0.525003f,
     81   0.087048f,  0.328194f,  -0.079989f, -0.391088f, -0.238732f, -0.120660f,
     82   -0.139600f, 0.154665f,  0.026202f,  -0.233501f, -0.009046f, -0.149187f,
     83   -0.199646f, 0.115375f,  0.209762f,  -0.014875f, 0.124038f,  -0.119985f,
     84   1.079625f,  -0.461513f, 0.614114f,  0.021003f,  0.439449f,  -0.824834f,
     85   -0.299701f, 0.193817f,  -0.870551f, -1.262313f, -0.079517f, 0.341570f,
     86   0.305310f,  -0.089721f, -0.317314f, -0.075631f, 0.127172f,  -0.208635f,
     87   1.191922f,  0.163141f,  0.564285f,  0.286352f,  0.480865f,  0.173094f,
     88   -0.094034f, -0.071339f, -0.328992f, -0.006382f, 0.314705f,  0.090258f,
     89   -0.016099f, 0.193230f,  0.188061f,  0.398144f,  0.722781f,  0.769949f,
     90   0.025442f,  -0.162016f, 0.070192f,  -0.056946f, -0.100957f, -0.219934f,
     91   -0.203492f, -0.015454f, -0.013272f, -0.098008f, 0.051707f,  -0.017493f,
     92   0.527446f,  0.083605f,  0.588318f,  0.878215f,  0.028747f,  -0.146479f,
     93   -0.345170f, -0.136059f, -0.152005f, -0.203634f, 0.232702f,  -0.101340f,
     94   -0.027733f, -0.282611f, 0.265366f,  0.082362f,  -0.265420f, -0.131124f,
     95   0.166303f,  0.040194f,  -0.100710f, 0.579151f,  -0.530136f, 0.163422f,
     96   -0.998821f, -1.565311f, -1.774785f, -2.493372f, 0.116970f,  -0.090302f,
     97   1.723272f,  0.552370f,  -0.295954f, -0.439095f, -0.266730f, 0.027936f,
     98   0.539616f,  -0.234902f, -0.167601f, -0.149877f, -0.242983f, 0.122353f,
     99   -0.121620f, -0.205517f, -0.180144f, -0.264208f, 0.151500f,  -0.159378f,
    100   0.029145f,  -0.050892f, -0.223407f, -0.246239f, 0.043152f,  -0.018460f,
    101   0.169972f,  -0.187769f, -0.034670f, -0.238330f, 0.288070f,  -0.093243f,
    102   -0.437105f, -0.573376f, 0.660073f,  0.285727f,  0.408470f,  0.158475f,
    103   0.032699f,  0.056280f,  -0.237176f, -0.083003f, 0.105598f,  -0.169522f,
    104   -0.260420f, -0.121100f, -0.173983f, -0.195693f, -0.232028f, 0.224940f,
    105   0.029124f,  0.009580f,  -0.252034f, 0.103087f,  1.156561f,  0.603848f,
    106   -0.562805f, -1.652742f, -0.568288f, -1.829395f, 0.046169f,  0.076095f,
    107   1.490819f,  0.415893f,  -0.277788f, -0.115787f, 0.093750f,  0.270726f,
    108   -0.395983f, -0.353742f, 0.034605f,  0.005342f,  0.184537f,  0.086445f,
    109   0.156417f,  1.476367f,  0.122587f,  0.002145f,  0.431057f,  -0.381184f,
    110   -1.646457f, -0.014009f, -0.671224f, 0.193726f,  -0.019247f, -0.031267f,
    111   -0.046208f, 0.298733f,  0.064734f,  0.616984f,  0.039381f,  0.182722f,
    112   -0.116670f, 0.233093f,  -1.214374f, -0.817970f, -0.064394f, -0.584783f,
    113   0.077697f,  -0.266720f, 0.130875f,  -0.235295f, -0.265754f, -0.159999f,
    114   -0.250114f, -0.183017f, 0.194403f,  -0.105808f, -0.169215f, -0.240866f,
    115   -0.026662f, -0.045123f, -0.036175f, -0.167471f, -0.192908f, -0.232602f,
    116   -0.267036f, -0.112500f, -0.257944f, -0.111909f, -0.802226f, -0.008800f,
    117   0.881460f,  -0.678603f, 0.008666f,  -0.252053f, -0.341035f, -0.175290f,
    118   0.183012f,  0.385991f,  0.079888f,  -0.014039f, -0.148653f, 0.671778f,
    119   -0.130219f, 1.086467f,  0.129267f,  -0.040400f, -0.201221f, -0.077005f,
    120   0.015890f,  0.000781f,  0.137764f,  1.389546f,  0.172152f,  0.047279f,
    121   -0.042783f, 0.127740f,  0.141467f,  -0.335738f, -1.396392f, 0.031496f,
    122   0.357385f,  0.343602f,  -0.714553f, 0.311014f,  0.132845f,  0.061149f,
    123   0.006796f,  0.568106f,  -0.255949f, 0.104134f,  -0.993447f, 0.298135f,
    124   -0.406590f, -0.049228f, -0.578570f, -0.188561f, -0.107046f, 0.374095f,
    125   0.068481f,  0.036240f,  -0.495801f, 0.180574f,  -0.766129f, 0.886967f,
    126   -0.568868f, -0.936062f, -0.418886f, -0.058735f, -0.511964f, -0.438596f,
    127   0.019016f,  -0.015837f, 0.600197f,  0.429773f,  0.315026f,  0.319667f,
    128   0.214617f,  -0.017316f, 0.270257f,  -0.040524f, 0.695803f,  -0.015223f,
    129   -1.554965f, 0.356997f,  -1.472428f, 0.024637f,  -0.562958f, 0.870351f,
    130   0.193635f,  0.036063f,  0.328638f,  0.200274f,  -1.634707f, 0.110534f,
    131   0.420104f,  -0.072042f, -0.006404f, 0.171680f,
    132 };
    133 
    134 static const float av1_ab_partition_nn_bias_128_layer0[64] = {
    135   0.643147f,  -1.348826f, 0.431627f,  0.000000f,  0.102717f,  -0.772628f,
    136   -0.034351f, -0.761977f, -0.638397f, 0.541969f,  -0.391311f, 0.563076f,
    137   0.148553f,  0.267217f,  -0.788092f, 0.544573f,  -0.546280f, 0.000000f,
    138   -0.446945f, 0.127732f,  0.270624f,  -0.219435f, -1.220203f, 0.324584f,
    139   0.110885f,  0.276547f,  0.179726f,  -0.375160f, 0.026401f,  -0.032595f,
    140   0.000000f,  -0.047932f, -0.648602f, -0.512637f, -0.031661f, -0.236761f,
    141   0.476453f,  -0.028021f, -0.013673f, -0.015578f, -0.920077f, 0.000000f,
    142   0.915351f,  -0.209962f, 0.000000f,  -0.025731f, 0.218288f,  0.000000f,
    143   0.047726f,  -0.813077f, -1.263281f, 0.239087f,  0.278614f,  -0.030753f,
    144   0.000000f,  0.346744f,  -0.948543f, -1.174211f, 0.216377f,  0.498913f,
    145   0.853918f,  0.002504f,  -0.190403f, 0.452050f,
    146 };
    147 
    148 static const float av1_ab_partition_nn_weights_128_layer1[64 * LABEL_SIZE] = {
    149   0.179769f,  1.499417f,  -0.445135f, -0.142278f, -0.337661f, 0.682064f,
    150   -0.203213f, 0.302171f,  0.226877f,  -0.422169f, 1.687586f,  0.783773f,
    151   0.220995f,  0.253482f,  0.370435f,  -1.342775f, 0.337229f,  -0.271473f,
    152   0.291796f,  1.362227f,  -1.751397f, -0.086178f, 0.725496f,  -0.118597f,
    153   0.227963f,  -0.501577f, 0.223849f,  -0.122421f, -0.123437f, -0.051045f,
    154   -0.020115f, 0.212711f,  0.246025f,  0.088120f,  -0.168995f, 1.740190f,
    155   -0.195098f, 0.680339f,  -0.589572f, -0.075244f, 0.878766f,  0.064092f,
    156   -3.548527f, 0.001660f,  0.107926f,  -0.169501f, -0.455212f, 0.123045f,
    157   -1.836998f, 0.330365f,  1.301475f,  0.454761f,  -0.576552f, -0.190761f,
    158   0.208459f,  0.618483f,  1.383364f,  0.970718f,  0.390174f,  0.406252f,
    159   -0.564519f, -0.312062f, 1.345712f,  -0.151873f, 0.109290f,  0.408847f,
    160   0.391243f,  0.152024f,  0.181764f,  -0.036263f, -0.160466f, 0.153595f,
    161   0.049163f,  -0.753012f, -1.804062f, 0.347475f,  -2.746580f, 0.575618f,
    162   0.261799f,  0.210505f,  -0.302054f, -0.109872f, 0.199506f,  -1.182971f,
    163   0.723668f,  0.177758f,  -0.338202f, 0.254396f,  -0.220023f, 0.043504f,
    164   0.669866f,  -0.040816f, -0.402730f, 0.017990f,  0.215523f,  -0.216816f,
    165   0.454826f,  -0.726067f, -0.018750f, -0.928679f, 0.154315f,  -0.465641f,
    166   0.144566f,  -0.030064f, -0.054667f, -0.154055f, 0.625384f,  1.323795f,
    167   -0.159496f, 0.097072f,  -0.463197f, -0.057938f, 0.750290f,  -0.233061f,
    168   0.412631f,  -0.535223f, -0.151423f, -0.154583f, 0.024721f,  -0.494448f,
    169   0.230594f,  -0.980138f, -0.653968f, 0.126079f,  0.051814f,  -0.053219f,
    170   -0.421708f, -0.228853f, 0.237885f,  0.888157f,  0.059655f,  0.241295f,
    171   0.210443f,  0.228238f,  0.119127f,  -0.051989f, -0.355408f, 0.182215f,
    172   0.244277f,  -0.104577f, -0.558035f, -0.023270f, 0.054571f,  0.700646f,
    173   -0.223006f, 0.115523f,  0.023391f,  0.437264f,  0.709477f,  -0.531212f,
    174   -0.094731f, 0.328161f,  -0.105418f, -0.133511f, 0.497168f,  -0.030948f,
    175   -0.407132f, -0.043943f, 0.155505f,  0.251945f,  0.205010f,  0.167160f,
    176   0.083654f,  -0.636810f, 0.401315f,  -0.398414f, 0.290046f,  0.206846f,
    177   0.042218f,  0.168150f,  0.843181f,  -0.671242f, -0.202392f, -0.073301f,
    178   0.142895f,  0.237466f,  0.212145f,  -0.091828f, 0.187038f,  -0.720841f,
    179   -0.616069f, -0.238021f, 0.065365f,  0.434119f,  0.179023f,  -0.040107f,
    180   -0.430734f, -0.297368f, 0.575954f,  0.382619f,  -0.709787f, -0.320810f,
    181   0.242342f,  -0.047614f, 0.705216f,  0.098077f,  0.357179f,  0.046017f,
    182   0.115074f,  -0.412305f, -0.272304f, 0.048096f,  -0.803811f, 0.275000f,
    183   0.642198f,  0.180286f,  -0.087178f, -0.112707f, -0.394443f, 0.201989f,
    184   0.241759f,  -1.038870f, 0.728124f,  0.800559f,  -1.296268f, 0.198612f,
    185   -0.053478f, 0.414344f,  -0.510529f, 0.124179f,  -2.219115f, -0.074583f,
    186   -0.143055f, 0.001697f,  0.810811f,  -0.657140f, 0.186818f,  -0.936414f,
    187   0.539578f,  -0.308244f, -0.126624f, -0.204767f, 0.091145f,  -0.049340f,
    188   0.252014f,  0.394582f,  0.018764f,  -0.060377f, -0.019133f, 0.064083f,
    189   0.069211f,  -0.526693f, 0.209850f,  -0.481466f, -0.468302f, -0.100407f,
    190   0.241018f,  -1.037781f, 0.038539f,  -2.113840f, -0.974895f, 0.163187f,
    191   0.425132f,  -0.772546f, -1.261254f, -0.217488f, -0.971748f, -0.805640f,
    192   -0.745175f, -0.177077f, 0.217658f,  0.381431f,  -0.052338f, 0.087176f,
    193   -0.165972f, 0.085937f,  0.472564f,  -0.796627f, -2.453307f, 0.569664f,
    194   -0.233010f, -0.192134f, 0.064339f,  -0.111411f, -0.262469f, -0.410022f,
    195   0.519993f,  -0.684620f, 0.393460f,  -0.277753f, -0.153624f, 0.528984f,
    196   -0.415558f, -0.445863f, 0.588512f,  -0.142439f, -0.132127f, 0.199776f,
    197   -0.579284f, 0.119488f,  -0.033590f, -0.503846f, -0.674979f, 0.335125f,
    198   0.020519f,  0.233973f,  -0.297998f, -0.051511f, 0.518626f,  -0.412782f,
    199   -0.074045f, 0.130523f,  0.465751f,  -0.117795f, 2.535813f,  0.352108f,
    200   -0.499228f, 0.379784f,  0.056699f,  0.173142f,  -0.076519f, -0.026666f,
    201   0.017834f,  0.492333f,  0.093364f,  0.037867f,  -0.165420f, -0.356429f,
    202   -0.562334f, 0.057656f,  -0.307544f, 0.085857f,  -0.559851f, 0.107230f,
    203   -0.398633f, 0.152618f,  -0.216835f, -0.024539f, 0.026044f,  -0.249519f,
    204   -0.563594f, -0.746025f, 0.025265f,  -0.298888f, -0.185243f, 0.058794f,
    205   0.233696f,  -0.115223f, 0.144617f,  -0.864390f, 0.619944f,  -0.023980f,
    206   0.019481f,  0.225252f,  0.416552f,  -0.115993f, 0.935387f,  0.744386f,
    207   0.053353f,  -0.052582f, -0.065650f, 0.228488f,  -0.032042f, -0.371252f,
    208   -0.003638f, -0.736984f, -0.203776f, 0.030922f,  -0.065577f, -0.031643f,
    209   -0.049253f, -0.054640f, 0.787134f,  0.545414f,  -0.140297f, -0.124274f,
    210   -0.110011f, -0.029552f, 0.657005f,  0.214973f,  -0.374300f, 0.251642f,
    211   0.276591f,  0.030566f,  -0.145470f, 0.350579f,  -0.356436f, -0.052694f,
    212   -0.063966f, -0.751008f, -1.042392f, 0.328892f,  -0.425058f, -0.421571f,
    213   -0.571889f, -1.141472f, -0.125216f, 0.212713f,  -0.485170f, -0.088791f,
    214   0.124589f,  0.023237f,  0.077635f,  0.020901f,  -0.271402f, -0.321424f,
    215   -0.513946f, -0.867872f, -0.284593f, 0.106276f,  0.220192f,  -0.143532f,
    216   -0.014648f, 0.073402f,  0.327256f,  -0.139803f, 0.168763f,  0.048199f,
    217   -0.122526f, 0.111713f,  -0.134257f, 0.810364f,  -0.085222f, -0.259221f,
    218   -0.239349f, 0.044448f,  0.205031f,  0.413113f,  -0.107720f, -0.018816f,
    219   -0.247741f, -0.004963f, 0.041170f,  -0.158019f, 0.134839f,  0.129502f,
    220   0.800488f,  -1.041584f, -0.129336f, 0.170834f,  0.566586f,  -0.230443f,
    221   0.437937f,  -0.149922f, -0.046665f, -0.094646f, 0.200070f,  0.072943f,
    222   -0.076943f, -0.084971f, -0.515843f, -0.146720f, 0.472869f,  -0.444731f,
    223   -0.100877f, 0.545196f,  -1.786626f, -0.482946f, 0.500509f,  -0.843257f,
    224   0.200374f,  0.045103f,  -0.575718f, -0.164335f, -0.232522f, -0.021825f,
    225   -0.139490f, 0.356058f,  -0.352075f, 0.061751f,  -0.200616f, -1.180921f,
    226   -0.181355f, -0.137459f, 0.247574f,  0.181541f,  0.184314f,  -0.961482f,
    227   0.493615f,  0.910261f,  -2.279238f, 0.648631f,  -0.055526f, -0.037137f,
    228   0.038643f,  0.136609f,  -0.819373f, -0.040840f, -0.265989f, 0.006877f,
    229   0.454651f,  -0.595323f, -0.099500f, -0.263717f, 0.150456f,  0.245077f,
    230   -0.268666f, 0.162232f,  -0.516451f, -0.024501f, 0.188046f,  -0.002262f,
    231   0.261319f,  0.004173f,  0.746982f,  0.174761f,  0.470447f,  -0.159558f,
    232   -0.385240f, 0.023084f,  -0.133520f, -0.220607f, -0.018731f, -0.373558f,
    233   -0.707763f, -1.850150f, -0.807404f, -0.168063f, -0.071435f, -0.160740f,
    234   -0.478789f, -1.070674f, -0.489740f, -0.255796f, 0.100486f,  -0.153361f,
    235   0.334394f,  -0.569472f, -0.198118f, 0.255922f,  0.104717f,  -0.065179f,
    236   0.111879f,  -0.447237f, 1.373623f,  -0.190191f, -0.063311f, 0.337529f,
    237   -0.138800f, 0.057009f,  -0.137006f, 0.641378f,  0.883147f,  -0.679655f,
    238   0.267717f,  -0.351602f, -0.135225f, 0.229398f,  -0.513225f, -1.120345f,
    239   0.528786f,  -0.051081f, 0.086653f,  0.140141f,  -0.563969f, 0.333402f,
    240   -0.174745f, 0.321093f,  -0.438641f, -0.005131f, 0.247415f,  0.110120f,
    241   -0.076308f, -0.083244f, 0.838944f,  -0.113043f, -0.013258f, -0.175028f,
    242   -0.179941f, 0.272676f,  -0.047946f, -0.088076f, -0.450031f, 0.053929f,
    243   -0.083549f, -0.089952f, -0.186253f, 0.257483f,  0.011019f,  0.586435f,
    244   0.060580f,  -0.052078f, 0.090277f,  -0.780869f, 0.969811f,  -0.025349f,
    245   -0.281917f, 0.014857f,  0.231863f,  -0.228601f, -0.003861f, 0.226550f,
    246   0.141825f,  -0.102171f, -0.010387f, 0.220378f,  -2.561975f, -0.497071f,
    247   -0.315117f, 0.371981f,  0.138247f,  0.625031f,  -0.308133f, -0.217876f,
    248   0.005615f,  -0.860179f, 0.747491f,  0.006356f,  -0.057024f, -0.483189f,
    249   0.055592f,  -0.316834f, 0.069858f,  0.218788f,  -0.200044f, 0.227588f,
    250   0.215496f,  -0.055324f, -0.393147f, -0.394062f, -0.253264f, -0.075619f,
    251   -0.152512f, -0.332995f, 0.129053f,  0.178668f,  -0.302694f, 0.030678f,
    252   0.925896f,  0.964375f,  0.169021f,  -0.218657f, -0.627204f, 0.206437f,
    253   -0.521336f, 0.176206f,  0.142733f,  0.139248f,  0.411682f,  0.181544f,
    254   0.224850f,  -0.935547f, -0.558208f, 0.348096f,  0.342129f,  -0.389340f,
    255   -0.236308f, -0.132099f, 0.073642f,  0.089391f,  -0.306901f, -0.397842f,
    256   0.444282f,  0.074623f,  -0.051075f, -0.106617f, -0.184037f, -0.239046f,
    257   -0.138761f, 0.120794f,  -0.647577f, -0.336471f, 0.527899f,  -0.164234f,
    258   -0.028354f, 1.083678f,  -0.251534f, -0.145903f, -0.182783f, 0.070976f,
    259   -0.199590f, -0.400306f, -0.029763f, -0.548042f, -0.266270f, -0.118084f,
    260   -1.152632f, 0.383685f,  -0.105895f, -0.096829f, 0.118382f,  0.047447f,
    261   -0.019051f, 0.310180f,  -0.162793f, -0.029574f, 0.058054f,  -0.636017f,
    262   0.490639f,  0.158347f,  -0.385701f, -0.147057f, 1.285825f,  -1.276083f,
    263   -0.021795f, -0.101600f, 0.163254f,  0.267160f,  -2.317864f, -0.098598f,
    264   -0.296337f, -0.309017f, 0.164127f,  -0.270012f, -0.071187f, -0.262270f,
    265   0.075415f,  -0.368328f, 0.186728f,  -0.158031f, 0.481663f,  0.515950f,
    266   -0.162551f, 0.497981f,  0.262196f,  0.168479f,  0.726066f,  -0.243856f,
    267   -0.058998f, 0.140168f,  0.053242f,  -0.624623f, -0.249480f, 0.055197f,
    268   -1.376804f, 0.417571f,  0.203784f,  0.174370f,  -0.155531f, -0.029400f,
    269   -0.491473f, 0.079811f,  -0.080123f, 1.345900f,  0.637077f,  0.434862f,
    270   -1.787438f, 0.005756f,  -0.362706f, 0.179458f,  -0.288263f, 0.516788f,
    271   -0.921248f, 0.043794f,  -0.137729f, -0.196171f, -0.046295f, -0.793781f,
    272   -0.156532f, -0.132566f, 0.517989f,  -0.154321f, -0.054174f, -0.077900f,
    273   -0.373316f, -0.117718f, 0.188986f,  -0.476188f, -0.245312f, 0.181439f,
    274   -0.161024f, -0.229059f, -3.079907f, -0.225452f, -0.594355f, -0.558027f,
    275   -0.135429f, 0.125766f,  -0.081314f, -0.350894f, -0.163165f, -1.936507f,
    276   -0.205966f, 0.031472f,  0.744446f,  -0.006680f, -0.837551f, 0.605862f,
    277   -0.854929f, -1.543750f, -0.307704f, -0.240517f, 0.178240f,  -0.183586f,
    278   -0.010307f, 0.099373f,  -0.228278f, 0.175236f,  -0.000133f, 0.104491f,
    279   -1.540545f, -0.570971f, -0.252885f, 0.483036f,  0.052531f,  0.260214f,
    280   -0.515016f, -0.602081f, -0.485690f, -0.730710f, 0.163719f,  -1.775975f,
    281   -0.298634f, 0.323626f,  -0.373579f, -0.872977f, 0.619574f,  0.026862f,
    282   -0.122531f, -0.084698f, -2.436297f, 0.483996f,  -0.203640f, -0.302157f,
    283   -0.150666f, -0.238320f, 0.089250f,  0.236485f,  -0.668654f, -0.122863f,
    284   0.491152f,  -0.226444f, -0.181248f, 0.120158f,  0.294027f,  0.250056f,
    285   0.307601f,  0.357875f,  -1.746455f, -0.175670f, 0.385447f,  -0.108808f,
    286   -0.090235f, -0.642504f, -0.486004f, -0.055160f, -0.068692f, 0.009736f,
    287   0.607555f,  -0.489426f, 0.150624f,  0.598114f,  -0.128816f, -0.445793f,
    288   -0.066524f, -0.254380f, 0.227106f,  -0.406495f, -0.121632f, -0.275960f,
    289   -0.136494f, 0.339457f,  -1.318132f, -0.417572f, -2.614077f, 0.324603f,
    290   -0.001211f, 0.375192f,  -0.473448f, -0.162510f, 0.099329f,  -0.277965f,
    291   0.101221f,  -0.060263f, 0.121867f,  -1.042140f, 0.440851f,  0.078898f,
    292   -0.209007f, -0.243699f, 0.715197f,  -0.093997f, 0.086022f,  -0.178203f,
    293   -2.275496f, -0.098413f, 0.199352f,  -0.526791f, -0.162086f, -0.197806f,
    294   -0.231657f, -0.269202f, -0.794294f, -0.223461f, 0.503584f,  0.416236f,
    295   0.064082f,  0.197655f,  0.340871f,  -0.186645f, -0.291498f, 0.433938f,
    296   -1.110063f, 0.003751f,  0.392738f,  0.069360f,  0.102088f,  -0.302128f,
    297   -1.518457f, 0.106939f,  0.404527f,  -0.306868f, -0.286928f, 0.729276f,
    298   -0.531710f, 0.745048f,  -0.168837f, -1.953886f, -0.258828f, -0.190252f,
    299   0.241877f,  -0.916744f, -0.030326f, -0.070541f, -0.271037f, 0.211303f,
    300   -0.489957f, 0.100850f,  0.323999f,  -0.802837f, -0.462408f, -0.079350f,
    301   -0.029374f, 0.131213f,  -0.825032f, 0.040202f,  0.351821f,  0.002869f,
    302   -0.132516f, -0.471264f, -0.297002f, 0.263913f,  0.033478f,  0.146161f,
    303   0.533229f,  -0.228608f, -0.200639f, -0.170955f, -0.915037f, 0.724491f,
    304   0.005151f,  0.018584f,  -0.029771f, -0.396038f, -0.159236f, 0.038691f,
    305   -1.197056f, 0.146302f,  0.226840f,  -0.852126f, 0.031214f,  0.108880f,
    306   0.562000f,  -0.134633f, -0.713343f, -0.342252f, -1.764521f, -0.114653f,
    307   0.515073f,  -0.080515f, -0.121155f, -0.865139f, -0.833694f, -0.368553f,
    308   0.347673f,  0.623379f,  0.722067f,  -0.492458f, -0.513263f, 0.585167f,
    309   0.721518f,  -0.693499f, 0.343725f,  -0.273861f, -0.040230f, -0.785664f,
    310   -0.157500f, -0.308445f, 0.054062f,  0.600131f,  -0.860887f, 0.434470f,
    311   -0.191382f, -0.306150f, -0.243965f, 0.705444f,  0.007789f,  -0.146154f,
    312   -0.054499f, -0.073500f, -1.067364f, 0.404936f,  -2.864590f, 0.182323f,
    313   0.326126f,  0.102405f,  -0.135800f, 1.128095f,  -0.012267f, -0.023996f,
    314   -0.264834f, -0.108967f, -1.176746f, -0.926666f, 0.082999f,  -0.498361f,
    315   0.083560f,  -0.210074f, 0.019225f,  -0.201614f, -0.904760f, 0.181421f,
    316   0.586384f,  -0.177706f, 0.065471f,  0.168552f,  0.054705f,  0.045241f,
    317   0.048057f,  -0.410957f, -2.188854f, -0.169812f, 0.015521f,  0.176856f,
    318   -0.179331f, -0.352640f, -0.491735f, -1.743206f, 0.044227f,  0.010454f,
    319   0.823643f,  -0.119781f, -0.098359f, 0.093119f,
    320 };
    321 
    322 static const float av1_ab_partition_nn_bias_128_layer1[LABEL_SIZE] = {
    323   -0.433195f, -0.120488f, -0.116721f, 0.112134f,  0.118170f, -0.259769f,
    324   -0.077530f, 0.394044f,  0.279167f,  -0.317988f, 0.189538f, 0.314776f,
    325   0.325655f,  -0.107123f, 0.591049f,  0.358744f,
    326 };
    327 
    328 static const NN_CONFIG av1_ab_partition_nnconfig_128 = {
    329   FEATURE_SIZE,  // num_inputs
    330   LABEL_SIZE,    // num_outputs
    331   1,             // num_hidden_layers
    332   {
    333       64,  // num_hidden_nodes
    334   },
    335   {
    336       av1_ab_partition_nn_weights_128_layer0,
    337       av1_ab_partition_nn_weights_128_layer1,
    338   },
    339   {
    340       av1_ab_partition_nn_bias_128_layer0,
    341       av1_ab_partition_nn_bias_128_layer1,
    342   },
    343 };
    344 
    345 // nn model for ab partition pruning, 64x64.
    346 static const float av1_ab_partition_nn_weights_64_layer0[FEATURE_SIZE * 64] = {
    347   -0.495347f, -0.049498f, -0.026804f, 0.030474f,  -0.289308f, -0.264193f,
    348   -0.141121f, -0.072562f, -0.391665f, -0.051491f, -0.234761f, 0.027155f,
    349   -0.038217f, 0.014872f,  -0.289728f, -0.233577f, -0.415875f, -0.343615f,
    350   -0.442543f, -0.482492f, 0.073510f,  0.007503f,  2.162329f,  -0.362849f,
    351   2.145915f,  -0.883135f, 0.185636f,  -0.062859f, -0.465574f, -0.486205f,
    352   -0.056710f, -0.330642f, -0.321860f, 0.042321f,  -0.348965f, 0.003542f,
    353   -0.291365f, -0.078164f, -0.345093f, -0.220272f, -0.471270f, -0.763853f,
    354   0.246622f,  0.199651f,  -0.663420f, -0.154152f, -1.220383f, 0.047138f,
    355   0.816811f,  0.083247f,  -0.218839f, 0.038143f,  -0.063436f, 0.015517f,
    356   -0.307320f, -0.166956f, -0.169499f, -0.399005f, -0.234638f, -0.162266f,
    357   0.050425f,  -0.221723f, -0.256942f, -0.287285f, 0.144011f,  -0.033245f,
    358   0.083649f,  0.119428f,  -0.056706f, -0.117805f, 0.021866f,  -0.257300f,
    359   -0.201378f, -0.217484f, -0.413780f, -0.145793f, 0.082792f,  -0.347247f,
    360   0.042539f,  -0.302697f, 1.652316f,  0.000701f,  -0.482843f, -0.160332f,
    361   -0.450099f, 0.212399f,  -4.715360f, -5.336774f, -5.375758f, -6.048339f,
    362   0.085956f,  -0.037767f, 1.052409f,  -0.931924f, -2.221907f, 0.268946f,
    363   0.015512f,  1.237094f,  -1.092185f, 0.418247f,  -0.082143f, -0.076914f,
    364   -0.060749f, -0.325440f, -0.296960f, -0.066815f, -0.158477f, -0.373945f,
    365   -0.122322f, -0.113495f, -0.097978f, -0.192816f, -0.270418f, 0.035840f,
    366   -0.015458f, -0.121071f, -0.279582f, -0.067683f, 0.097855f,  0.019839f,
    367   0.451127f,  0.004376f,  1.410392f,  3.255835f,  -0.344815f, 0.145202f,
    368   0.204132f,  0.171948f,  -0.527736f, -0.110353f, 0.901448f,  0.003238f,
    369   -3.822090f, 0.235462f,  1.024823f,  -0.821244f, 0.876056f,  2.553762f,
    370   -3.478597f, -2.076582f, -0.265515f, -0.055923f, -0.156980f, -0.164097f,
    371   -0.246040f, 0.039430f,  -0.071769f, -0.118847f, -0.304053f, -0.281541f,
    372   -0.226021f, -0.263091f, -0.127359f, -0.249410f, -0.051023f, 0.083911f,
    373   0.084721f,  0.168089f,  -0.272169f, -0.204998f, -0.008303f, -0.173998f,
    374   0.079376f,  -0.197426f, -0.199052f, -0.118794f, -0.063753f, -0.094769f,
    375   0.066176f,  -0.175832f, -0.238752f, -0.287960f, -0.134307f, -0.185953f,
    376   -0.385845f, 0.119769f,  -0.006567f, -0.382126f, -0.214221f, 0.038449f,
    377   -0.253484f, -0.282766f, -0.020249f, -0.193929f, 0.016281f,  -0.114423f,
    378   -0.145940f, -0.281621f, -0.007588f, -0.131470f, -0.189012f, -0.185699f,
    379   -0.279011f, -0.008132f, 0.208463f,  0.020569f,  -0.206803f, -0.213408f,
    380   -0.206131f, -0.290245f, 0.069701f,  -0.000371f, -0.307572f, -0.451785f,
    381   -0.300838f, -0.453186f, -0.301691f, 0.046327f,  -0.312668f, 0.058272f,
    382   -0.303131f, -0.376252f, 0.108384f,  -0.086623f, -0.100630f, -0.027330f,
    383   -0.003969f, 0.089502f,  -0.200722f, -0.107889f, 0.061843f,  -0.008478f,
    384   -0.265057f, -0.271132f, -0.073562f, 0.129337f,  -0.283698f, -0.353414f,
    385   0.076420f,  -0.244280f, -0.119537f, -0.105366f, -0.184692f, -0.038817f,
    386   -0.478507f, -0.118808f, -0.472979f, -0.305884f, -0.462813f, -0.189581f,
    387   -0.011932f, -0.585700f, 0.253212f,  -1.061900f, -0.205116f, -0.336407f,
    388   -0.762199f, 0.577737f,  0.230832f,  0.434440f,  -0.096713f, 0.038552f,
    389   -0.147800f, -0.213553f, 0.041740f,  -0.281907f, -0.026154f, -0.082356f,
    390   -0.331871f, -0.408247f, -0.129022f, -0.037550f, -0.310233f, -0.320883f,
    391   -0.391963f, -0.467392f, 0.027453f,  -0.394761f, -0.045544f, 0.076052f,
    392   0.483985f,  0.067093f,  0.141361f,  0.576772f,  0.859718f,  2.566515f,
    393   -0.025476f, 0.769738f,  -0.680235f, -1.683309f, -2.394131f, -0.000714f,
    394   -0.615021f, -0.195856f, -0.434035f, -0.295010f, -0.668659f, -0.245959f,
    395   0.551148f,  1.777227f,  -0.461630f, 0.043093f,  0.012293f,  -0.255841f,
    396   -0.097070f, -0.371156f, -0.146323f, -0.015508f, -0.103873f, -0.087476f,
    397   -0.297266f, -0.128699f, -0.149555f, 0.016534f,  -0.375498f, -0.346759f,
    398   -0.455156f, -0.147509f, -0.427076f, -0.354431f, -0.158025f, -0.164604f,
    399   -0.237038f, -0.010314f, -0.092884f, -0.397084f, -0.217980f, -0.127184f,
    400   -0.048421f, -0.144133f, 0.889073f,  0.012606f,  3.007608f,  -0.602584f,
    401   -1.849480f, -0.373159f, -1.890695f, -3.609938f, 0.811923f,  -1.867208f,
    402   -0.244326f, -0.018012f, -0.211192f, -0.220196f, 0.169363f,  0.119141f,
    403   -0.230715f, 0.083247f,  0.020367f,  -0.128629f, -0.217455f, -0.159640f,
    404   1.815952f,  -0.369238f, -1.186447f, -0.658753f, -0.511026f, -0.096934f,
    405   0.662971f,  0.486475f,  0.159746f,  -0.018932f, 3.692397f,  1.384353f,
    406   -0.401984f, -0.248380f, -0.140861f, 0.215248f,  -0.023711f, 0.059679f,
    407   -0.072260f, 0.004271f,  0.039545f,  -0.347971f, -0.081851f, -0.474896f,
    408   -0.181572f, 0.066736f,  -0.157822f, -0.163760f, -0.171113f, -0.089935f,
    409   -0.338281f, -0.421444f, -0.306687f, -0.085283f, -0.377953f, -0.138750f,
    410   -0.102701f, -0.312336f, 0.149831f,  0.007229f,  -0.155700f, -0.173611f,
    411   4.074261f,  1.342306f,  -1.272712f, 1.570899f,  -0.545093f, -0.317605f,
    412   -0.189440f, -0.133910f, -0.273190f, -0.108020f, -0.166107f, 0.021413f,
    413   -0.239130f, -0.067211f, 0.041957f,  -0.039234f, -1.003587f, -0.094412f,
    414   0.532512f,  -0.870538f, -1.118023f, -1.160983f, -0.736307f, -0.418752f,
    415   0.419466f,  0.492122f,  -0.004368f, -0.022096f, -1.115132f, 0.150886f,
    416   2.396852f,  2.660000f,  -0.376537f, 0.468628f,  0.149413f,  -0.074898f,
    417   -0.067154f, 0.021245f,  0.127857f,  0.294189f,  0.508056f,  0.390232f,
    418   -3.899177f, -3.414681f, -3.929195f, -4.160545f, -0.274323f, -0.052583f,
    419   -0.003545f, -0.433084f, -0.404891f, -0.145051f, -0.312367f, 0.004579f,
    420   -0.398724f, -0.372068f, -0.234279f, 0.017799f,  -0.424760f, -0.646717f,
    421   -0.047568f, 2.924664f,  -0.644165f, 0.359349f,  -0.294800f, 0.591746f,
    422   -0.404710f, -0.092358f, -0.250729f, 0.030829f,  -0.147149f, -0.476023f,
    423   -0.071803f, -0.482516f, -0.293117f, -0.215923f, -0.373122f, -0.085315f,
    424   -0.377052f, -0.449899f, -0.056452f, 0.138081f,  -0.085350f, -0.308391f,
    425   0.106661f,  0.176234f,  0.258869f,  -0.230172f, -0.233029f, -0.241208f,
    426   -0.067509f, -0.223172f, -0.118353f, -0.302478f, -0.579632f, -0.561326f,
    427   -0.158114f, -0.223167f, -0.026689f, 0.051863f,  0.212834f,  -0.304714f,
    428   -0.169071f, -0.193695f, -0.075682f, -0.170860f, -0.241008f, -0.044648f,
    429   0.280815f,  -0.002585f, -0.283552f, -0.037701f, -0.681169f, -0.274535f,
    430   -0.380595f, 0.109504f,  -0.111141f, -0.437685f, -0.094459f, 0.144206f,
    431   -0.106139f, -0.211832f, -0.054742f, -0.172813f, -0.295905f, -0.071907f,
    432   -0.418429f, -0.183240f, 0.031319f,  -0.095785f, -0.315447f, 0.069404f,
    433   -0.422910f, -0.029867f, -0.357321f, -0.199976f, -0.337707f, -0.070188f,
    434   -0.178198f, 0.177208f,  0.134688f,  -0.081933f, -0.229452f, -0.208872f,
    435   0.026287f,  -0.364040f, -0.063696f, -0.227443f, -0.234401f, -0.205699f,
    436   -0.267238f, -0.494125f, -0.056255f, 0.053715f,  -0.487754f, 0.014818f,
    437   0.087383f,  -0.077556f, -0.168085f, -0.436851f, -0.276286f, -0.137845f,
    438   -0.107606f, -0.103653f, -0.233766f, -0.419083f, 0.169185f,  0.010186f,
    439   -0.001587f, 0.086735f,  -2.465718f, 1.482185f,  1.621193f,  -2.081680f,
    440   1.386553f,  -3.204335f, -0.267111f, -0.004508f, 0.164712f,  0.274147f,
    441   1.724306f,  -2.273659f, 0.749574f,  -0.891905f, 0.105965f,  -0.030428f,
    442   -0.416018f, -0.300762f, 0.122911f,  -0.316908f, -0.292504f, 0.138666f,
    443   -0.161327f, -0.042143f, -0.249128f, 0.149210f,  -0.088987f, -0.654101f,
    444   -1.501843f, 0.216777f,  0.955914f,  0.524158f,  -1.642561f, -1.643626f,
    445   0.864797f,  -0.425451f, -2.115764f, -0.012502f, 0.065172f,  1.297270f,
    446   0.018845f,  1.167276f,  -0.470970f, -0.244995f, 0.374782f,  -1.811056f,
    447   -0.055430f, -0.024102f, -0.376519f, -0.339640f, -0.119177f, -0.277995f,
    448   -0.290095f, -0.081362f, -0.144139f, -0.118037f, -0.180357f, -0.217559f,
    449   -0.370683f, 0.172816f,  -0.265069f, 0.194321f,  -0.273478f, 0.037442f,
    450   -0.235552f, -0.078625f, -0.447541f, 0.016836f,  -0.271123f, -0.171481f,
    451   -0.321477f, -0.184826f, -0.442981f, -0.227273f, -0.370666f, -0.237232f,
    452   -0.257493f, -0.225714f, -0.153716f, -0.283487f, -0.155399f, 0.067697f,
    453   0.230343f,  -0.034318f, -0.022687f, -0.047090f,
    454 };
    455 
    456 static const float av1_ab_partition_nn_bias_64_layer0[64] = {
    457   -0.212182f, -0.233725f, -0.758846f, -0.158162f, 0.614743f,  -0.150944f,
    458   -0.075727f, -0.208414f, 1.054996f,  0.713758f,  -0.300051f, -0.151482f,
    459   -2.443570f, 0.430590f,  -0.129001f, -0.160733f, -0.230547f, -0.143228f,
    460   -0.140577f, -0.086812f, -0.212298f, -0.159557f, -0.055647f, -0.211423f,
    461   0.578161f,  -0.220318f, -0.210107f, -3.111584f, 0.604419f,  -0.232622f,
    462   -0.209924f, -0.130794f, -0.084097f, -0.036005f, 0.294594f,  -2.535531f,
    463   -0.209783f, -0.211189f, -2.766337f, 0.000000f,  0.450177f,  -1.754884f,
    464   3.262664f,  -0.209691f, -0.614886f, -0.211257f, -0.109096f, -0.190492f,
    465   -0.109007f, -0.026910f, -0.136035f, -0.212321f, -0.139320f, -0.212233f,
    466   -0.305430f, 0.739171f,  0.991277f,  -0.088150f, 0.086313f,  -0.023379f,
    467   -0.125366f, -0.063576f, -0.212169f, -0.047463f,
    468 };
    469 
    470 static const float av1_ab_partition_nn_weights_64_layer1[64 * LABEL_SIZE] = {
    471   -0.036800f, 0.528721f,  0.490767f,   0.144409f,  1.103640f,  0.361910f,
    472   -0.180069f, 0.068033f,  -14.868382f, 0.359013f,  0.322567f,  -0.199212f,
    473   0.906164f,  -0.488254f, 0.149653f,   -0.216394f, -0.099347f, 0.004936f,
    474   -0.111391f, 0.074848f,  -0.041709f,  0.147627f,  -0.018905f, 0.096116f,
    475   0.184817f,  -0.016241f, 0.115739f,   2.376754f,  0.637097f,  0.052954f,
    476   0.136428f,  0.225267f,  -0.181873f,  -0.142876f, 0.684048f,  0.658791f,
    477   0.105795f,  0.241705f,  1.381114f,   -0.209379f, 1.145949f,  0.795293f,
    478   -9.361877f, 0.198302f,  0.539600f,   0.092317f,  -0.081695f, 0.200777f,
    479   0.102334f,  0.081583f,  0.060948f,   -0.025110f, 0.160951f,  -0.020170f,
    480   0.234006f,  -0.029369f, 0.375036f,   0.270209f,  -0.556529f, 1.402949f,
    481   0.101777f,  -0.027331f, 0.004502f,   -0.153166f, -0.116651f, 0.151573f,
    482   -0.022187f, 0.144044f,  -0.108719f,  -0.129942f, -0.270321f, 0.227363f,
    483   1.892330f,  -0.661052f, -0.219398f,  -0.229417f, -0.856438f, -1.196988f,
    484   -0.081774f, 0.078847f,  -0.207057f,  -0.048947f, 0.152073f,  -0.243056f,
    485   -0.233329f, -0.288689f, -0.158333f,  -0.141177f, -0.715436f, 0.016947f,
    486   -0.093752f, 0.204984f,  -1.209782f,  0.155683f,  0.092239f,  0.146495f,
    487   0.813146f,  -0.027757f, 0.330982f,   2.173948f,  -0.028867f, -0.141815f,
    488   0.292708f,  -0.204794f, 0.014496f,   1.032799f,  1.312155f,  0.107020f,
    489   0.824752f,  -0.013945f, 0.184829f,   -0.041633f, 0.215300f,  -0.476088f,
    490   -0.053213f, 0.126862f,  -0.020777f,  0.082893f,  -0.223727f, -0.923063f,
    491   0.466529f,  0.082140f,  -0.845758f,  -1.140791f, -0.262033f, 0.138491f,
    492   0.151717f,  -0.182479f, -0.131128f,  0.055411f,  0.106771f,  0.125552f,
    493   0.297184f,  -0.257403f, -0.059884f,  -0.274903f, 2.694357f,  -0.108244f,
    494   0.025377f,  0.043092f,  -0.558317f,  3.517159f,  -0.270833f, -0.240676f,
    495   0.205100f,  -0.057068f, -0.140445f,  -0.193449f, -0.030061f, -0.286762f,
    496   -0.467523f, -0.012647f, 0.190564f,   0.022394f,  -0.101479f, 0.339684f,
    497   -0.902743f, -0.169578f, -0.178029f,  -0.041836f, -3.952108f, -0.028298f,
    498   -0.221137f, -0.733895f, -0.223895f,  0.039012f,  0.687867f,  0.021423f,
    499   0.113063f,  0.676087f,  -0.961000f,  -0.064847f, 0.712856f,  -0.192765f,
    500   -0.001132f, 0.016689f,  -0.236020f,  -0.766186f, -0.175729f, 0.012879f,
    501   -0.251064f, -0.105523f, -0.039212f,  -0.347584f, 0.304352f,  -0.034174f,
    502   -0.364258f, -0.685252f, -0.266115f,  -0.247345f, -0.155905f, 0.152283f,
    503   -0.156315f, 0.174082f,  -0.757654f,  0.102303f,  -2.192316f, -0.245815f,
    504   0.119882f,  -0.086542f, 1.987246f,   -1.353163f, -0.374813f, -0.233504f,
    505   -1.980895f, 0.692093f,  -0.168351f,  0.172700f,  -0.009052f, -0.015734f,
    506   0.106679f,  -0.060472f, -0.256813f,  -0.074874f, -0.207488f, -0.329515f,
    507   -0.418268f, -0.017940f, -0.036081f,  0.064719f,  -1.488016f, 0.020591f,
    508   -0.176325f, -0.141074f, 0.944494f,   0.150237f,  -0.249805f, -0.277280f,
    509   0.012686f,  0.132483f,  0.116123f,   0.013737f,  -0.116091f, 0.750340f,
    510   3.251343f,  -0.188864f, 1.096992f,   0.058467f,  -0.041433f, -0.037937f,
    511   -0.133294f, -0.137908f, -0.171132f,  0.106362f,  0.069383f,  -0.052662f,
    512   -0.177883f, -0.408049f, 0.680221f,   -0.117035f, -0.904240f, -1.395228f,
    513   0.154527f,  0.134427f,  0.022767f,   -0.158886f, -0.230316f, 0.161096f,
    514   0.362213f,  -0.235060f, -0.941620f,  0.055912f,  -0.049458f, -0.166632f,
    515   0.481418f,  0.930146f,  0.041108f,   0.033674f,  1.372066f,  -1.847709f,
    516   0.003324f,  0.259534f,  0.177014f,   -0.202761f, -0.262017f, -0.190852f,
    517   -0.102839f, 0.028338f,  0.187193f,   -0.041684f, 0.123973f,  -0.198576f,
    518   -0.110369f, -1.431400f, 0.208369f,   -0.302370f, -0.248549f, 0.062985f,
    519   0.673409f,  0.036662f,  -0.711340f,  -0.120584f, -0.189789f, 0.098812f,
    520   2.947819f,  0.216567f,  -0.414472f,  -0.181742f, 1.873779f,  -0.222726f,
    521   -0.782870f, 0.007889f,  0.015062f,   -0.554328f, 0.182928f,  -0.191430f,
    522   0.123636f,  -0.215460f, -0.225245f,  0.251516f,  -0.013025f, -1.359595f,
    523   -0.750602f, 0.342667f,  -0.141899f,  -0.687493f, -0.072639f, 0.048018f,
    524   -0.242107f, -0.031917f, -0.287472f,  -0.046088f, 0.832197f,  -0.016576f,
    525   -1.553349f, -0.216341f, 0.023077f,   -0.410867f, 4.243743f,  -0.514878f,
    526   -0.066007f, -0.160696f, -0.262678f,  -0.648790f, -0.430586f, 0.199940f,
    527   -0.202496f, -0.222241f, -0.016406f,  -0.121473f, 0.000828f,  -0.081584f,
    528   -0.152641f, -0.190166f, 0.644400f,   0.040196f,  -0.302104f, -1.143654f,
    529   -0.160327f, -0.320780f, -0.187006f,  0.037311f,  0.440618f,  -0.070733f,
    530   -0.117785f, 1.527539f,  -0.419310f,  0.001300f,  1.389956f,  -0.036366f,
    531   -0.269203f, 0.612265f,  2.721897f,   -0.086836f, -0.446999f, 0.012525f,
    532   -0.078317f, -0.287052f, -0.111188f,  -0.085181f, -0.164667f, -0.010466f,
    533   -0.569722f, -0.018888f, -0.101663f,  -1.147130f, -0.465204f, 0.114524f,
    534   -2.192402f, -0.221325f, 0.375748f,   0.206284f,  -0.261548f, -0.246257f,
    535   -0.143004f, -0.069981f, -0.057306f,  -0.116481f, -0.435903f, -0.314970f,
    536   0.013210f,  -0.010175f, 4.630571f,   -0.473226f, -0.197199f, -0.028204f,
    537   0.122907f,  2.475548f,  0.025011f,   -0.092603f, -0.127561f, -0.151330f,
    538   -0.077295f, 0.245016f,  -0.045005f,  0.183396f,  -0.330556f, -0.384887f,
    539   0.356374f,  -0.016618f, -0.463353f,  -1.291546f, -0.071986f, -0.311599f,
    540   0.072385f,  -0.430786f, -2.094788f,  0.202733f,  -0.910109f, -1.336543f,
    541   -0.086800f, -0.096413f, 1.544383f,   0.031860f,  -0.796211f, 0.762786f,
    542   3.250022f,  -0.441798f, -0.698537f,  0.062839f,  0.033525f,  -0.362996f,
    543   0.027022f,  -1.131264f, -0.228926f,  0.053885f,  -0.338628f, 0.155037f,
    544   -0.046844f, -0.888172f, -0.241767f,  0.084965f,  -0.617743f, -0.049896f,
    545   -0.036894f, -0.304783f, -0.002639f,  0.137957f,  0.052121f,  -0.131161f,
    546   -0.117200f, -0.253380f, -0.205561f,  -0.302450f, -0.047397f, -0.330518f,
    547   3.613420f,  -1.525951f, -0.026738f,  0.209150f,  -2.103534f, 2.019689f,
    548   -0.366199f, -0.095260f, 0.027417f,   -0.242512f, 0.162579f,  0.052113f,
    549   -0.293851f, -0.068138f, -0.005799f,  -0.344696f, -0.114824f, -0.431107f,
    550   -0.120058f, -1.139926f, -1.048379f,  0.036446f,  -0.323020f, -0.432945f,
    551   0.454151f,  -0.140058f, 0.050649f,   -0.094900f, -0.017278f, -0.238719f,
    552   1.193153f,  0.120447f,  -0.496061f,  0.917431f,  2.936126f,  -0.115521f,
    553   -0.347397f, -0.435325f, -0.004383f,  -0.211864f, 0.162383f,  -1.040726f,
    554   0.089537f,  -0.128579f, -0.133505f,  0.107129f,  -0.435657f, -0.180388f,
    555   0.043650f,  0.018709f,  -0.773242f,  -0.687192f, -0.120633f, -0.063626f,
    556   0.029912f,  0.113972f,  -0.403502f,  -0.127640f, -0.269625f, 0.129794f,
    557   -0.188539f, 0.041641f,  0.029769f,   -0.198374f, 1.401407f,  0.353887f,
    558   -0.219925f, 0.260515f,  1.157034f,   -2.992044f, -0.097618f, -0.064417f,
    559   -0.203626f, -0.008217f, -0.112339f,  -0.227407f, -0.155118f, 0.247705f,
    560   -0.012304f, -0.248447f, -0.913463f,  -0.064788f, -0.214619f, -0.251761f,
    561   -0.386861f, -0.040574f, -0.163219f,  -0.100700f, 1.488274f,  -0.071684f,
    562   -0.033626f, -0.006497f, -0.246945f,  -0.145221f, -3.747390f, 0.149609f,
    563   -0.263326f, -0.297385f, -1.039896f,  -0.083174f, -0.025473f, -0.235586f,
    564   -0.001087f, 0.254286f,  0.265106f,   0.007325f,  0.199239f,  0.134103f,
    565   -0.578211f, -0.259801f, -0.062373f,  2.368348f,  0.560556f,  -0.252260f,
    566   0.889997f,  -0.447872f, -0.059218f,  -0.095315f, -0.061667f, 0.183580f,
    567   -0.157479f, 0.055387f,  -0.831734f,  0.007606f,  -1.104906f, 0.301180f,
    568   -0.117115f, 0.212959f,  4.727223f,   -0.243833f, -0.397495f, -0.025021f,
    569   -0.367587f, -2.082058f, -0.217699f,  0.148111f,  0.252430f,  0.111088f,
    570   -0.260692f, 0.095124f,  -0.407774f,  -0.322169f, 0.002927f,  0.126169f,
    571   -1.272325f, -0.279772f, -0.373680f,  -0.485177f, -0.605458f, 0.021225f,
    572   -0.092031f, -0.226585f, 1.895162f,   0.037866f,  -0.275475f, 1.614360f,
    573   -0.014972f, -0.277679f, -3.449082f,  -0.092060f, -0.747873f, 0.020716f,
    574   2.776178f,  -0.049963f, 0.183999f,   -0.295259f, -0.028868f, 0.221895f,
    575   0.001265f,  0.336823f,  0.219372f,   0.112824f,  0.408132f,  -0.017940f,
    576   -0.311666f, 1.489606f,  -0.058093f,  -0.305659f, -0.491933f, -0.143847f,
    577   0.166115f,  0.042867f,  -0.123447f,  -0.087099f, -0.305395f, -0.365079f,
    578   -0.755801f, -0.160649f, 0.736260f,   -0.008611f, 0.095836f,  -0.017345f,
    579   5.697515f,  -0.498971f, -0.125280f,  0.199907f,  0.300053f,  0.605026f,
    580   -0.228225f, -0.259523f, 0.016384f,   0.146973f,  0.210258f,  0.226766f,
    581   -0.075178f, -0.050924f, 0.188496f,   -0.415266f, -0.484880f, -0.236384f,
    582   0.071931f,  -0.331863f, -0.601243f,  -0.232479f, -0.285272f, 0.123789f,
    583   -1.341333f, 0.037082f,  -0.315202f,  -1.587215f, -0.271576f, 0.003216f,
    584   -4.437186f, -0.256205f, -0.576589f,  -0.114147f, 2.153916f,  -0.369618f,
    585   0.271415f,  0.145036f,  -0.158731f,  -0.240938f, -0.187369f, 0.036325f,
    586   0.254771f,  0.211488f,  -0.240297f,  0.098417f,  -0.415011f, 2.334793f,
    587   -0.127252f, 0.020069f,  -0.168755f,  -0.448922f, -0.219207f, 0.016232f,
    588   -0.221935f, -0.269500f, -0.100636f,  0.102545f,  -0.809376f, -0.054979f,
    589   0.360713f,  -0.326541f, 0.112933f,   0.138073f,  4.229404f,  -0.763801f,
    590   -0.305429f, 0.199955f,  -1.787713f,  0.272866f,  0.109895f,  0.138466f,
    591   -0.250259f, -0.167162f, -0.212588f,  -0.217589f, -0.067125f, -0.077490f,
    592   -0.208970f, -0.006863f, -0.671146f,  -0.298320f, -0.165509f, 0.044597f,
    593   -1.408624f, -0.213957f, -0.220947f,  0.129718f,  1.316777f,  -0.098928f,
    594   -0.008121f, -0.558293f, -0.297290f,  -0.218873f, -4.346638f, -0.228174f,
    595   -0.204710f, -0.388864f, 2.697919f,   0.025260f,  0.857020f,  0.009921f,
    596   0.036915f,  -0.320275f, -0.087937f,  0.022636f,  0.236667f,  0.135496f,
    597   -0.059616f, -0.192955f, 0.009470f,   2.139589f,  -0.200449f, 0.129818f,
    598   1.017444f,  -0.608299f, 0.257914f,   -0.134306f, -0.033327f, 0.002855f,
    599   -0.338598f, 0.015559f,  0.117362f,   -0.166760f, 0.086903f,  -0.167666f,
    600   0.193523f,  0.033852f,  -1.147686f,  0.489468f,  -0.006969f, 0.125630f,
    601   1.557907f,  -1.604449f, -0.071114f,  0.096178f,  0.007065f,  0.200013f,
    602   0.213393f,  0.168466f,  -0.100568f,  -0.117861f, -0.161542f, -0.072561f,
    603   -1.069871f, -0.470138f, -0.352578f,  -1.503513f, -0.001394f, -0.380109f,
    604   0.065089f,  -0.281668f, 0.988953f,   -0.002778f, -0.659026f, -0.470692f,
    605   -0.407292f, 0.011710f,  -1.362085f,  0.184738f,  -0.135786f, -1.374241f,
    606   4.487930f,  -0.067274f, -0.956404f,  -0.233995f, 0.224527f,  -0.454556f,
    607   0.037900f,  -0.281658f, 0.208224f,   -0.254753f, 0.045740f,  0.051444f,
    608   -0.388281f, 0.257112f,  -0.485030f,  -0.082659f, 0.148103f,  -1.007456f,
    609   -0.022295f, 0.036984f,  -0.369401f,  -0.076943f, -0.007636f, -0.293022f,
    610   0.470466f,  0.199012f,  -2.158182f,  0.036577f,  -0.014725f, -0.229516f,
    611   2.236929f,  0.030945f,  -0.400045f,  0.109348f,  0.214691f,  -0.891516f,
    612   -0.251379f, -0.217358f, 0.013733f,   0.205573f,  -0.151725f, -0.191782f,
    613   -0.339630f, -0.163905f, -0.119191f,  -0.032516f, 0.503015f,  0.025772f,
    614   0.029094f,  -1.146153f, 0.216723f,   -0.330023f, 0.064695f,  -0.262521f,
    615   0.425612f,  -0.093080f, -0.489648f,  1.051293f,  -0.092332f, 0.095557f,
    616   -0.874132f, 0.218483f,  -0.127648f,  -1.605802f, 2.763617f,  -0.186734f,
    617   -1.243166f, -0.193514f, -0.173748f,  0.337822f,  0.183873f,  -0.251594f,
    618   -0.211582f, 0.144081f,  0.029620f,   -0.024853f, -0.385140f, 0.467341f,
    619   -0.928316f, -0.195442f, 0.917783f,   0.357084f,  0.174445f,  -0.073659f,
    620   -0.012811f, -0.115420f, -0.181147f,  -0.364449f, -0.567395f, -0.012969f,
    621   -1.680714f, 0.065323f,  0.198063f,   -0.244201f, 1.428545f,  -0.432539f,
    622   -0.208931f, -0.091205f, 0.957125f,   0.813519f,  -0.262677f, 0.246852f,
    623   0.015536f,  0.055026f,  0.067054f,   0.262103f,  -0.358115f, -0.095206f,
    624   -0.267522f, -0.402710f, -0.680397f,  -0.123627f, -0.385590f, -1.504680f,
    625   -0.169513f, -0.215338f, 0.043633f,   -0.079052f, -0.464410f, 0.122894f,
    626   -0.278231f, -2.456445f, -0.159917f,  -0.015597f, -0.735449f, -0.078854f,
    627   -0.400290f, -1.153870f, 3.657228f,   -0.287093f, -1.174355f, -0.102001f,
    628   -0.288281f, 0.185209f,  -0.145228f,  -0.200449f, -0.099914f, -0.138354f,
    629   0.254428f,  -0.161751f, -0.118206f,  0.296043f,  -0.482613f, 0.080932f,
    630   1.097605f,  -0.010190f, 0.232439f,   0.447617f,  -0.133508f, 0.115763f,
    631   -0.388589f, 0.174695f,  -0.236014f,  0.006284f,  -1.374129f, 0.092015f,
    632   -0.241419f, -0.231667f, 2.763950f,   -0.922932f, -0.061605f, 0.208740f,
    633   -1.597190f, 1.353325f,  -0.198528f,  0.250498f,  -0.013950f, -0.203861f,
    634   -0.254563f, 0.081931f,  -0.413369f,  0.011844f,  0.080961f,  -0.231161f,
    635   -1.234909f, -0.440843f, -0.174980f,  -0.315283f, -0.337474f, -0.123243f,
    636   -0.310001f, -0.271028f, 0.364179f,   0.022845f,  -0.535517f, -0.772936f,
    637   -0.188435f, 0.039667f,  -0.807463f,  0.266550f,  -0.288857f, -1.630789f,
    638   1.280155f,  0.065712f,  -0.279960f,  -0.300056f, 0.258440f,  -0.073781f,
    639   0.213878f,  0.042196f,  0.021360f,   0.211698f,  -0.003751f, -0.192673f,
    640   -0.137008f, 0.247878f,  -0.470604f,  0.073164f,  1.523241f,  0.734755f,
    641   -0.114126f, -0.193834f, -0.025759f,  0.263183f,
    642 };
    643 
    644 static const float av1_ab_partition_nn_bias_64_layer1[LABEL_SIZE] = {
    645   -0.343508f, -0.706936f, -0.160676f, -0.877101f, -0.517567f, -0.253254f,
    646   -0.148074f, 0.923430f,  -0.364770f, 0.203550f,  0.401216f,  0.938246f,
    647   -0.872737f, 0.718723f,  0.703398f,  2.560015f,
    648 };
    649 
    650 static const NN_CONFIG av1_ab_partition_nnconfig_64 = {
    651   FEATURE_SIZE,  // num_inputs
    652   LABEL_SIZE,    // num_outputs
    653   1,             // num_hidden_layers
    654   {
    655       64,  // num_hidden_nodes
    656   },
    657   {
    658       av1_ab_partition_nn_weights_64_layer0,
    659       av1_ab_partition_nn_weights_64_layer1,
    660   },
    661   {
    662       av1_ab_partition_nn_bias_64_layer0,
    663       av1_ab_partition_nn_bias_64_layer1,
    664   },
    665 };
    666 
    667 // nn model for ab partition pruning, 32x32.
    668 static const float av1_ab_partition_nn_weights_32_layer0[FEATURE_SIZE * 64] = {
    669   -0.323723f, -0.214013f, -0.007772f, -0.458851f, -0.125542f, -0.123860f,
    670   -0.410973f, -0.209389f, -0.087580f, -0.272881f, -0.168500f, -1.130845f,
    671   0.344916f,  -0.475017f, -0.362262f, -0.195662f, -0.566124f, 0.782163f,
    672   0.411575f,  -0.013378f, -0.318650f, -0.124678f, -0.612909f, -0.315788f,
    673   -0.263990f, -0.508783f, -0.048938f, -0.416407f, -0.402648f, -0.156644f,
    674   0.225887f,  -0.000493f, 2.682241f,  0.871204f,  0.059014f,  0.803542f,
    675   -1.407028f, -1.154669f, 1.388148f,  -0.293348f, -0.003669f, -0.009607f,
    676   1.330030f,  -0.337841f, 2.118617f,  1.033059f,  -0.084788f, 0.212904f,
    677   0.082405f,  -0.070579f, -0.494005f, -0.173392f, 0.039546f,  -0.463865f,
    678   0.077163f,  -0.434066f, 0.030835f,  -0.427139f, -0.560520f, -0.031606f,
    679   -0.368541f, -0.027458f, 0.370574f,  0.461418f,  1.087682f,  -0.572137f,
    680   -1.509596f, -0.765697f, -0.499383f, -0.277998f, -0.106492f, -0.129564f,
    681   -0.169133f, -0.269834f, -0.114270f, -0.275431f, 0.016339f,  -0.156744f,
    682   -0.267922f, 0.171216f,  0.110556f,  0.002954f,  -0.200327f, -0.187663f,
    683   3.691601f,  1.234152f,  0.186315f,  -0.125370f, -0.211235f, -0.554432f,
    684   -0.131072f, -0.124982f, -0.130339f, -0.235350f, 0.018903f,  0.012896f,
    685   -0.159372f, -0.269571f, -0.025709f, -0.221251f, 0.061919f,  0.016307f,
    686   0.384673f,  -0.134525f, -1.599126f, -0.416459f, -0.743052f, 0.670249f,
    687   -0.169709f, 0.421681f,  -0.033360f, -0.072817f, 0.003647f,  -0.110632f,
    688   -0.158651f, -0.095136f, 0.223759f,  0.165767f,  -0.269129f, -0.196075f,
    689   -0.023183f, -0.293420f, 0.014875f,  0.018688f,  -0.153407f, -0.172009f,
    690   -0.259947f, -0.124015f, 0.173653f,  -0.089103f, -0.021001f, -0.334230f,
    691   0.027177f,  0.103371f,  -0.183860f, -0.204051f, -0.023721f, -0.192297f,
    692   -0.143771f, -0.247106f, 0.218116f,  -0.013240f, 2.831783f,  1.483928f,
    693   -0.877025f, -0.313462f, -0.411320f, -0.447825f, 0.605977f,  0.234684f,
    694   -0.119150f, -0.075182f, -0.330463f, 0.071503f,  -0.254924f, -0.360071f,
    695   -0.037022f, 0.063261f,  -0.148759f, -0.238254f, -0.462018f, -0.027166f,
    696   0.065318f,  -0.235743f, -0.257194f, -0.094784f, 0.022423f,  0.055925f,
    697   0.086672f,  -0.021010f, 0.009965f,  -0.001648f, -0.104917f, -0.387443f,
    698   -0.102673f, -0.281706f, 0.145923f,  -0.233391f, -0.378365f, -0.145584f,
    699   -0.077751f, -0.121166f, 1.134565f,  -0.097500f, -0.749202f, -0.544566f,
    700   -1.361374f, -0.102494f, 1.089275f,  0.375299f,  -0.105091f, 0.037641f,
    701   -0.054248f, -0.282691f, -0.377797f, -0.066427f, -0.253815f, -0.329677f,
    702   -0.339326f, -0.128217f, -0.282905f, 0.014937f,  1.067185f,  -0.171764f,
    703   0.484458f,  0.396706f,  -0.557055f, -0.891596f, -0.257839f, -0.720879f,
    704   -0.218449f, -0.004755f, 1.572857f,  0.006229f,  1.962895f,  -0.029746f,
    705   -4.137691f, -2.185991f, -2.763477f, -0.520437f, -0.208708f, 0.006444f,
    706   -1.263078f, -0.304560f, 1.072374f,  2.556429f,  0.312850f,  0.257488f,
    707   -0.634264f, 0.156769f,  -0.188943f, 0.040295f,  -0.389915f, 0.085250f,
    708   -0.248525f, 0.045667f,  -0.776115f, -0.274680f, -0.448145f, -0.566161f,
    709   -1.285316f, 0.079060f,  0.389124f,  -0.510401f, -0.015299f, -0.664661f,
    710   0.099901f,  -0.470694f, -0.051593f, -1.076381f, -0.442104f, -0.197867f,
    711   -0.330011f, -0.448523f, -0.301018f, -0.442093f, -0.491953f, -0.582091f,
    712   -0.064569f, -0.156516f, 0.543522f,  -0.005924f, 0.161432f,  0.974793f,
    713   0.273712f,  1.104850f,  -0.290312f, 0.313417f,  -0.125370f, 0.136234f,
    714   -0.191227f, -0.165054f, 0.011872f,  -0.298871f, 0.095740f,  0.142760f,
    715   -0.215771f, -0.031437f, 0.101041f,  -0.085620f, 0.435387f,  0.002786f,
    716   1.971375f,  0.018392f,  -1.771940f, -0.401433f, 0.808263f,  -3.350013f,
    717   2.296952f,  -1.024403f, -0.041645f, -0.034799f, -0.024078f, -0.347301f,
    718   -0.276088f, -0.455907f, 0.266021f,  0.087348f,  -0.146566f, 0.040492f,
    719   -0.539866f, -0.206851f, -0.387874f, -0.125508f, -0.496676f, -0.373845f,
    720   -0.472356f, -0.357082f, -0.081254f, -0.456466f, 0.554713f,  0.002185f,
    721   -4.225019f, 0.344025f,  0.728796f,  -0.262936f, 1.383924f,  1.577300f,
    722   -2.653320f, -2.516156f, -0.301604f, -0.204105f, -0.138252f, -0.587536f,
    723   -0.097889f, -0.352414f, -0.288276f, -0.184340f, -0.122741f, -0.243376f,
    724   0.031970f,  -0.373402f, -0.396079f, 0.045566f,  0.072595f,  -0.222681f,
    725   -0.243802f, -0.340129f, -0.258494f, -0.192041f, -0.386112f, -0.240940f,
    726   -0.047268f, -0.555802f, -0.032514f, -0.241341f, -0.167463f, -0.478308f,
    727   -0.205936f, -0.316275f, 0.103729f,  -0.197893f, -0.128029f, -0.218796f,
    728   -0.167362f, -0.111814f, -0.126062f, -0.394260f, -0.025357f, -0.402697f,
    729   -0.587395f, -0.400385f, -0.259664f, -0.415588f, -0.338503f, -0.399166f,
    730   -0.270504f, 0.234505f,  0.272144f,  0.266938f,  -0.392395f, -0.011717f,
    731   -0.384221f, -0.473446f, -0.038420f, -0.241101f, -0.234402f, -0.275567f,
    732   -0.410454f, -0.377599f, -0.179099f, -0.138432f, -0.248083f, -0.543026f,
    733   -0.428043f, -0.239895f, -0.333193f, -0.103346f, -0.039038f, -0.171109f,
    734   -0.119432f, -0.222351f, 0.000450f,  0.208724f,  -0.510526f, -0.144656f,
    735   -0.316721f, -0.344846f, -0.244794f, -0.129134f, -0.045634f, -0.400183f,
    736   0.043714f,  -0.235414f, 0.115594f,  -0.195616f, -0.106693f, -0.124242f,
    737   0.083990f,  0.049110f,  -0.196130f, -0.059860f, -0.464235f, -0.516443f,
    738   -0.101521f, -0.422379f, -0.413955f, -0.042991f, -0.345263f, -0.129264f,
    739   -0.106911f, -0.140156f, -0.457841f, -0.199848f, -0.218954f, -0.329850f,
    740   -0.364097f, -0.335262f, -0.312254f, -0.299331f, -0.052710f, -0.251019f,
    741   -0.023459f, -0.222538f, 0.028849f,  -0.088038f, -0.301550f, -0.273566f,
    742   0.067295f,  -0.174608f, -0.445784f, -0.158366f, -0.567275f, -0.557652f,
    743   -0.353503f, -0.302092f, -0.302049f, -0.551793f, -0.034535f, -0.225190f,
    744   -0.210733f, -0.219377f, -0.057197f, -0.430933f, -0.025185f, -0.388150f,
    745   -0.086147f, -0.430088f, 0.058466f,  -0.152129f, -0.058411f, -0.236392f,
    746   -0.547669f, -0.613849f, -0.893774f, -0.351715f, -0.399227f, -0.454909f,
    747   -0.324501f, 0.000490f,  -0.282167f, -0.073163f, -0.281452f, 0.047932f,
    748   -0.175500f, 0.165220f,  -0.276212f, 0.062153f,  -0.217054f, -0.255487f,
    749   -0.146416f, -0.097718f, -0.173809f, -0.559328f, -0.055695f, -0.391193f,
    750   -0.132020f, -0.561184f, -0.308666f, -0.474053f, -0.219149f, -0.246558f,
    751   -0.158325f, 0.151907f,  -0.266835f, -0.144697f, -0.193960f, -0.046587f,
    752   -0.220028f, -0.247355f, 0.135584f,  0.016511f,  0.367705f,  -1.855877f,
    753   0.435622f,  0.444710f,  -3.372301f, -3.030489f, 1.013267f,  0.380951f,
    754   -0.170011f, -0.111415f, -0.456146f, -0.107254f, -0.095220f, -0.053078f,
    755   -0.135864f, -0.591949f, -0.252810f, -0.324799f, -0.094796f, -0.260969f,
    756   -0.391981f, -0.063170f, -0.336130f, -0.470127f, -0.405168f, -0.433219f,
    757   -0.309563f, -0.295462f, -0.552270f, -0.012300f, -0.057793f, -0.034494f,
    758   -0.446843f, -0.640160f, -1.188681f, -0.791361f, 0.543271f,  1.189112f,
    759   1.458468f,  -0.005876f, -0.927475f, 0.062038f,  -1.170818f, 0.338227f,
    760   -3.007096f, -4.559296f, -4.045457f, -5.953635f, -0.228386f, -0.266890f,
    761   -0.092595f, -0.377440f, -0.044534f, -0.053565f, -0.349268f, -0.415030f,
    762   -0.310094f, 0.062721f,  0.251422f,  -0.014350f, -1.282910f, 1.619560f,
    763   1.180566f,  -0.032163f, -1.322951f, -0.603601f, 1.443710f,  0.654650f,
    764   -0.393227f, 0.003536f,  0.029725f,  -0.108925f, -0.053911f, 0.133977f,
    765   -0.036145f, -0.168438f, 0.046989f,  -0.331463f, -0.176983f, -0.311922f,
    766   -0.272389f, -0.379592f, -0.399993f, -0.297873f, -0.193425f, -0.177524f,
    767   -0.258309f, -0.567312f, -0.260217f, -0.241869f, 0.024010f,  -0.032867f,
    768   -0.039424f, -0.063670f, 0.193808f,  -0.303514f, -0.013376f, -0.057761f,
    769   0.187922f,  0.006938f,  0.031810f,  0.180594f,  -1.198427f, 2.820662f,
    770   0.154986f,  -0.375518f, 0.116925f,  -0.795782f, -0.085139f, -0.079365f,
    771   -0.197936f, -0.321468f, -0.205271f, -0.558203f, -0.296235f, -0.151193f,
    772   -0.158282f, -0.245402f, -0.208504f, -0.042335f, -0.087426f, -0.557129f,
    773   -0.381427f, -0.441551f, -0.541011f, -0.060567f, -0.469305f, -0.032326f,
    774   -2.453587f, -0.045568f, -0.296932f, 0.613061f,  -0.320284f, 0.191620f,
    775   -0.827145f, -0.225277f, 0.275800f,  1.696635f,
    776 };
    777 
    778 static const float av1_ab_partition_nn_bias_32_layer0[64] = {
    779   -0.176206f, 0.660189f,  -0.186156f, -2.481963f, -1.564218f, -0.280424f,
    780   0.732684f,  -0.135581f, -2.193132f, -0.172771f, 0.605001f,  -0.060392f,
    781   -0.067190f, -0.132969f, -1.410812f, -0.298701f, -0.105963f, -0.086173f,
    782   0.632779f,  0.005585f,  1.310169f,  1.392136f,  -0.563860f, -0.051053f,
    783   0.660998f,  -0.214726f, -1.894342f, -0.128288f, -0.330721f, -0.053988f,
    784   -0.177726f, 1.200859f,  -0.178902f, -0.172620f, -0.184476f, -0.175559f,
    785   0.538503f,  -0.322158f, -0.219080f, -0.058208f, -0.171347f, -0.216060f,
    786   -0.174950f, -0.295740f, -0.184820f, -0.213896f, 1.317728f,  -0.020116f,
    787   -0.208096f, 0.000000f,  1.246166f,  -0.225421f, -0.181555f, 0.861761f,
    788   1.172429f,  -0.172892f, -0.737092f, -0.189904f, -0.179385f, -0.114618f,
    789   -1.384604f, -0.201713f, -0.271948f, 0.372351f,
    790 };
    791 
    792 static const float av1_ab_partition_nn_weights_32_layer1[64 * 16] = {
    793   -0.037828f,  1.529029f,  0.004927f,  1.475763f,  0.627172f,  0.325872f,
    794   -0.990757f,  0.129476f,  0.889958f,  -0.082031f, 0.332133f,  0.074422f,
    795   -0.176212f,  -0.074355f, 0.774378f,  0.110987f,  -0.155469f, 0.253310f,
    796   0.882538f,   0.253605f,  0.332436f,  -5.389474f, 0.278470f,  0.168644f,
    797   0.914611f,   0.154165f,  0.809262f,  -0.174734f, 0.923673f,  0.064716f,
    798   -0.070228f,  -0.228735f, 0.002312f,  0.112222f,  -0.045502f, -0.046004f,
    799   0.514101f,   0.306480f,  0.021232f,  -0.015955f, -0.288260f, 0.189177f,
    800   -0.104158f,  0.103273f,  0.096910f,  -0.086328f, 1.327289f,  -0.154247f,
    801   0.056676f,   -0.243327f, -0.646676f, 0.177221f,  -0.086761f, 0.729729f,
    802   -14.710893f, -0.044881f, 0.339003f,  -0.134737f, 0.073621f,  -0.162913f,
    803   1.215237f,   0.140723f,  0.138630f,  1.241719f,  0.204092f,  -0.463080f,
    804   -0.176086f,  1.125868f,  1.034814f,  0.225455f,  -0.203421f, -0.078787f,
    805   -0.527498f,  0.012491f,  -0.563307f, -0.170792f, 0.002679f,  0.116153f,
    806   0.211348f,   -0.191900f, -0.212505f, 0.263445f,  -0.074679f, -0.081441f,
    807   -0.815405f,  2.448215f,  0.781299f,  0.149542f,  -1.045162f, 0.043014f,
    808   0.217381f,   -0.094500f, -0.090427f, 0.025784f,  -0.228906f, -2.741798f,
    809   0.230475f,   -0.256112f, -0.103297f, 0.159121f,  -0.229793f, -0.014883f,
    810   -0.104131f,  -0.123816f, 0.164148f,  -0.052279f, -0.071845f, -0.041197f,
    811   0.208527f,   -0.234197f, -0.542336f, 0.020053f,  0.088870f,  0.014346f,
    812   2.502164f,   -0.010244f, -0.267792f, 0.844394f,  2.711486f,  -0.015262f,
    813   -0.868053f,  -0.295704f, 0.222289f,  -0.000286f, -0.352098f, -0.079000f,
    814   0.021267f,   -0.721739f, -0.240558f, -0.384775f, 0.065974f,  -2.161058f,
    815   0.195889f,   0.268966f,  -0.009329f, 0.014949f,  0.314943f,  0.235885f,
    816   0.072591f,   -0.127120f, 0.150784f,  0.105697f,  -1.297403f, -0.207509f,
    817   -0.217688f,  -0.076752f, 0.170952f,  -0.294235f, 0.449973f,  -1.712690f,
    818   0.860989f,   0.054757f,  -0.812627f, -0.105316f, -0.736230f, -0.133192f,
    819   -3.741608f,  0.495660f,  -0.288936f, 4.654852f,  -0.021305f, -0.308916f,
    820   0.049205f,   -0.259996f, 0.114248f,  -0.252647f, -0.253180f, -0.449314f,
    821   0.022979f,   0.063281f,  -0.196154f, 0.078295f,  -0.322317f, -0.145142f,
    822   0.300573f,   0.048385f,  -0.254787f, 0.123939f,  -1.263088f, -0.228565f,
    823   -0.389061f,  0.391084f,  2.322438f,  0.075009f,  0.225743f,  -0.198808f,
    824   -0.280538f,  -0.173939f, -0.120543f, -0.070792f, -0.417187f, -0.781056f,
    825   -0.102756f,  -1.760965f, 0.019149f,  -0.867342f, 0.347141f,  0.031588f,
    826   0.302572f,   -0.203573f, -0.357320f, -0.096078f, -0.527528f, 0.046699f,
    827   -0.108561f,  -0.167077f, -2.851509f, -0.307116f, 0.202720f,  -0.160280f,
    828   -0.215525f,  0.064355f,  -0.427220f, 1.516230f,  0.634453f,  0.099400f,
    829   -1.013887f,  -0.029740f, -0.093426f, -0.044272f, -1.297636f, -0.237614f,
    830   -0.160953f,  0.399036f,  -0.030685f, -0.113619f, -0.184704f, 0.040519f,
    831   -0.588252f,  -0.210235f, -0.067623f, -0.031841f, -0.107261f, -0.192582f,
    832   -0.253959f,  -0.430821f, -0.103184f, -0.280185f, -0.357723f, 0.197761f,
    833   -0.175087f,  -0.055171f, 1.642014f,  -0.192559f, -0.288147f, 0.610311f,
    834   4.688195f,   -0.128728f, -0.914869f, -0.108286f, 0.013789f,  0.092125f,
    835   0.019770f,   -0.178386f, 0.074164f,  -1.152658f, -0.216738f, -0.277286f,
    836   0.012381f,   0.418259f,  -0.680727f, -0.221917f, -0.485946f, 0.101672f,
    837   2.009457f,   0.054302f,  1.019838f,  -0.116170f, 0.165134f,  -0.112567f,
    838   0.852632f,   -0.385796f, -0.108666f, 0.053181f,  -0.311797f, -0.372875f,
    839   -0.675717f,  2.409268f,  -0.514720f, -0.214245f, -0.646596f, 0.009756f,
    840   0.203993f,   0.093617f,  -0.301290f, 0.253551f,  -0.128909f, -1.448442f,
    841   -0.186823f,  -0.278001f, -0.294993f, -0.176928f, -0.473605f, 0.062049f,
    842   -0.212084f,  -0.137326f, 0.012505f,  0.087850f,  -0.200413f, -0.394119f,
    843   -0.132224f,  0.146917f,  0.155746f,  0.198725f,  -0.322541f, 0.196391f,
    844   -0.945500f,  0.036736f,  -0.155646f, -0.677341f, 1.130545f,  -0.339554f,
    845   0.411628f,   -0.355813f, -0.249843f, 0.213694f,  -2.035607f, 0.055694f,
    846   -0.111669f,  0.408696f,  -0.067043f, -0.048182f, 0.398110f,  -0.067542f,
    847   1.459801f,   0.236833f,  -0.178806f, 0.168758f,  0.492387f,  0.099691f,
    848   -0.776680f,  -0.172865f, 0.204225f,  0.193982f,  0.575685f,  -0.062248f,
    849   0.011486f,   0.058571f,  -0.493391f, 0.026893f,  -0.900467f, 3.793129f,
    850   -0.634613f,  -0.064660f, -0.048262f, 0.361905f,  0.033641f,  0.245171f,
    851   -0.064671f,  0.034954f,  0.204358f,  -0.904023f, -0.052714f, -0.250134f,
    852   0.136700f,   0.000734f,  -0.371720f, 0.226483f,  0.217958f,  0.060559f,
    853   0.180111f,   0.000970f,  0.079556f,  -0.096775f, 0.093855f,  -0.026224f,
    854   -0.243664f,  0.004290f,  0.123281f,  -0.239476f, 1.230374f,  -0.107826f,
    855   -0.101982f,  -0.153917f, 5.464427f,  0.304375f,  -0.809957f, 0.090564f,
    856   -0.278416f,  -0.245555f, -2.078421f, 0.243093f,  -0.127666f, 0.052451f,
    857   -0.126662f,  -0.783505f, 0.025149f,  -1.422675f, -0.207769f, -0.362547f,
    858   0.115310f,   0.133390f,  1.264754f,  -0.027055f, -0.485312f, -0.240717f,
    859   -0.239722f,  0.146818f,  -1.265043f, -0.235553f, 0.267104f,  -0.021357f,
    860   -0.435949f,  -0.309371f, 0.049920f,  1.302721f,  -0.233978f, -0.097551f,
    861   -0.240631f,  -0.287821f, -0.378380f, -0.273131f, -3.075169f, 0.226404f,
    862   -0.029361f,  2.703590f,  -0.430659f, 0.067927f,  -0.387520f, -0.370630f,
    863   -0.229236f,  0.085653f,  -0.370956f, -0.065556f, -0.187859f, 0.068309f,
    864   -0.109299f,  -0.259898f, -0.103644f, -0.271199f, -0.209350f, 0.140993f,
    865   -0.196713f,  -0.135508f, -1.423209f, -0.406385f, -0.019956f, -0.864694f,
    866   5.963707f,   -0.201157f, 0.726377f,  -0.011076f, 0.010553f,  -0.102918f,
    867   -2.230088f,  -0.258098f, -0.039547f, -0.029262f, -0.082324f, -0.860222f,
    868   -0.094735f,  -1.381839f, 0.587298f,  -0.173048f, 0.721360f,  0.241900f,
    869   0.764302f,   -0.023609f, -1.173755f, 0.103912f,  -0.185363f, 0.078435f,
    870   -2.245062f,  -0.127269f, 0.202234f,  0.158975f,  -0.260909f, 0.098608f,
    871   -0.348247f,  1.732502f,  -0.412298f, -0.269602f, -0.425771f, -0.146243f,
    872   -0.530730f,  0.125716f,  -1.004419f, 0.145109f,  -0.059289f, 1.096304f,
    873   0.012891f,   0.045033f,  -0.306875f, 0.003514f,  -0.176110f, 0.037544f,
    874   -0.441537f,  -0.518921f, -0.262149f, -0.060407f, -0.379419f, -0.141245f,
    875   -0.128894f,  -0.176537f, -1.161318f, -0.249100f, -0.118330f, 0.042816f,
    876   1.173404f,   0.088312f,  -0.393568f, -0.175134f, 6.529819f,  -0.326652f,
    877   -0.631917f,  -0.393476f, 0.057781f,  -0.217748f, -1.781139f, -0.012614f,
    878   -0.212621f,  -0.720322f, -0.218498f, -0.388556f, -0.254796f, -0.248399f,
    879   -0.608744f,  -0.265146f, 0.238517f,  0.066882f,  -2.916806f, 0.054642f,
    880   0.282590f,   0.075248f,  0.010188f,  -0.133486f, 0.985945f,  -0.045849f,
    881   -0.347564f,  0.057320f,  -0.417920f, 0.063664f,  0.387062f,  -2.692059f,
    882   -0.535549f,  0.263736f,  0.327889f,  -0.070273f, -0.775254f, 0.147250f,
    883   3.309425f,   -0.212191f, -0.067204f, -2.912663f, -0.061496f, 0.084233f,
    884   0.022907f,   0.138421f,  -0.112159f, -0.288447f, -0.010799f, 0.056049f,
    885   -0.036527f,  0.021525f,  0.106649f,  -0.291883f, 0.088424f,  -0.057773f,
    886   -0.086031f,  0.015277f,  -0.318505f, -0.269049f, -1.008913f, -0.224785f,
    887   -0.025820f,  -0.649037f, 0.706381f,  0.096410f,  0.643776f,  -0.046743f,
    888   -0.009654f,  -0.024246f, 1.469255f,  -0.183536f, -0.370046f, -0.048442f,
    889   -0.376527f,  -0.431264f, -0.245109f, -0.093951f, 0.203683f,  -0.099872f,
    890   0.087210f,   0.160692f,  -3.527694f, -0.068891f, -0.228994f, -0.231817f,
    891   -0.241949f,  0.193613f,  0.979597f,  -0.091259f, 0.414424f,  -0.047341f,
    892   -0.209582f,  -0.295134f, -0.016824f, 0.460327f,  -0.072671f, 0.246234f,
    893   0.235896f,   0.127238f,  -1.068683f, 0.035648f,  2.254888f,  0.180105f,
    894   -0.260098f,  -2.322120f, -0.184249f, -0.314801f, -0.099969f, -0.272117f,
    895   -0.237916f,  0.031103f,  -0.274063f, -0.049384f, -0.044917f, 0.102477f,
    896   -0.342148f,  -0.257558f, -0.346300f, 0.115333f,  -0.115456f, 0.208354f,
    897   -0.359301f,  -0.167395f, 1.146514f,  -0.177861f, -0.098658f, -0.444570f,
    898   6.759993f,   -0.369772f, -0.831118f, 0.001866f,  -0.073298f, -0.072095f,
    899   0.811902f,   -0.431997f, -0.286587f, -0.269500f, 0.111492f,  -0.525364f,
    900   -0.351785f,  -2.463474f, -1.852659f, 0.135325f,  0.138267f,  0.100643f,
    901   -2.373278f,  -0.285514f, -0.395388f, -0.185016f, -0.030249f, -0.005767f,
    902   -0.716424f,  -0.031674f, 0.011147f,  0.057405f,  -0.215873f, -0.094401f,
    903   0.573528f,   -1.223820f, 0.414852f,  -0.059053f, -0.076488f, -0.287168f,
    904   -0.842640f,  0.174084f,  -0.567186f, 0.336629f,  -0.062514f, 2.075448f,
    905   -0.061680f,  -0.131529f, -0.098994f, -0.204111f, -0.347865f, 0.108516f,
    906   -0.049616f,  -0.069212f, -0.273935f, -0.096545f, -0.210784f, -0.284698f,
    907   0.141501f,   -0.176924f, -0.361341f, -0.251197f, -0.286694f, 0.245569f,
    908   -1.521661f,  -0.122639f, -0.015760f, -0.718912f, 5.877828f,  0.146916f,
    909   0.151767f,   0.220785f,  -0.032298f, 0.230902f,  0.663943f,  -0.252613f,
    910   0.057718f,   -0.436038f, -0.323994f, -1.139787f, -0.042489f, -1.326298f,
    911   -1.031206f,  -0.104136f, 0.389897f,  0.127602f,  -2.667789f, -0.212366f,
    912   -0.506262f,  -0.009115f, -0.213202f, 0.076167f,  -1.629405f, 0.055129f,
    913   0.375393f,   -0.150272f, -0.241515f, -0.326497f, 0.100069f,  0.410703f,
    914   0.340622f,   0.042437f,  -0.349945f, 0.041176f,  -1.178950f, 0.030992f,
    915   0.933908f,   -0.035844f, -0.098660f, 1.030584f,  -0.092043f, -0.355739f,
    916   -0.305562f,  0.036161f,  -0.049558f, -0.033225f, -0.403856f, -0.088276f,
    917   0.215493f,   -0.149105f, -0.013363f, 0.025886f,  -0.101306f, -0.205781f,
    918   -1.072487f,  -0.076019f, 0.077555f,  0.131003f,  1.267763f,  -0.008954f,
    919   -0.327617f,  -0.246539f, 6.664081f,  -0.404403f, -1.442489f, 0.191301f,
    920   -0.336361f,  0.181156f,  0.833108f,  0.007879f,  -0.194464f, -1.029408f,
    921   -0.036268f,  -0.927110f, -0.379190f, -0.293443f, -1.848579f, -0.242548f,
    922   -0.065990f,  0.203160f,  -0.291788f, 0.000680f,  0.587011f,  -0.241289f,
    923   0.037034f,   0.000552f,  1.072308f,  -0.387230f, -0.230050f, 0.292322f,
    924   -0.720001f,  0.034109f,  -0.467260f, 2.211644f,  -1.839191f, -0.048797f,
    925   -0.083469f,  -0.334686f, -0.269056f, 0.051295f,  1.319904f,  -0.035603f,
    926   -0.018457f,  -0.824915f, -0.212285f, -0.230516f, -0.035093f, -0.400843f,
    927   -0.305469f,  -0.099011f, 0.014225f,  -0.452772f, 0.170331f,  -0.389312f,
    928   -0.115084f,  -0.014770f, -0.429387f, -0.155961f, -0.568200f, -0.037853f,
    929   -0.125137f,  0.067228f,  -1.329271f, -0.117874f, -0.132499f, -0.218376f,
    930   -0.588325f,  -0.320024f, 0.085695f,  -0.235047f, -0.217790f, 0.103015f,
    931   -0.698644f,  0.017766f,  -0.058299f, 0.199411f,  -0.122485f, -0.563949f,
    932   -0.349011f,  -0.557045f, -0.131165f, 0.002281f,  0.118559f,  -0.210302f,
    933   -1.153815f,  0.116738f,  -0.236007f, -0.003487f, -0.006885f, -0.244816f,
    934   0.953222f,   0.093748f,  0.266869f,  0.241869f,  -0.860832f, -0.387012f,
    935   -0.338986f,  2.097515f,  -1.942512f, -0.298021f, 0.543911f,  -0.043214f,
    936   0.082125f,   -0.120242f, 0.712231f,  0.213327f,  -0.301687f, -0.544011f,
    937   -0.392131f,  0.004302f,  0.004825f,  -0.317440f, -0.107518f, -0.293407f,
    938   -0.159111f,  -0.080367f, 0.132663f,  -0.017726f, -0.237521f, -0.190297f,
    939   -0.361633f,  0.200518f,  -0.538296f, -0.027975f, -0.381704f, -0.016963f,
    940   0.630105f,   -0.190997f, -0.287840f, -0.603488f, 3.605598f,  -0.276614f,
    941   -1.346383f,  0.186912f,  -0.047575f, -0.189232f, -1.519072f, 0.097816f,
    942   -0.223722f,  0.304924f,  -0.213022f, -1.052433f, -0.322283f, -1.706734f,
    943   -2.458027f,  0.237976f,  0.171050f,  -0.103139f, -0.278689f, 0.329824f,
    944   -0.262448f,  -0.122916f, -0.236398f, -0.013848f, -0.969160f, -0.374907f,
    945   0.091018f,   -0.386471f, -0.723940f, 0.064956f,  -0.057652f, 1.321024f,
    946   -1.397418f,  -0.143136f, 0.272468f,  -0.030749f, 0.037324f,  0.069316f,
    947   -0.904925f,  -0.333693f, -0.117709f, 2.279598f,  -0.428065f, -0.131157f,
    948   -0.014288f,  -0.402862f, -0.666090f, 0.017070f,  -0.028333f, 0.002481f,
    949   0.197156f,   -0.038120f, -0.271062f, -0.188275f, -0.021370f, -0.070849f,
    950   -0.905007f,  -0.095886f, -0.093055f, -0.121821f, -1.239812f, -0.411799f,
    951   -0.089948f,  -0.936827f, 1.437569f,  -0.388908f, 0.126170f,  0.186162f,
    952   -0.018819f,  -0.138364f, -1.066412f, -0.138222f, -0.022186f, 0.107331f,
    953   -0.230436f,  -1.352605f, -0.161323f, -1.081810f, -0.933825f, -0.136675f,
    954   0.378157f,   0.113377f,  -0.850610f, 0.080245f,  -0.087305f, -0.002852f,
    955   0.044408f,   -0.188172f, -1.891998f, 0.092189f,  0.125325f,  -0.105090f,
    956   -0.848510f,  -0.396308f, -0.384130f, 2.007509f,  -1.480787f, -0.126946f,
    957   0.314767f,   0.000195f,  -0.285628f, -0.110442f, -0.293948f, 0.258559f,
    958   -0.417603f,  1.570705f,  0.092459f,  -0.340974f, -0.284754f, -0.007801f,
    959   -0.324610f,  -0.004734f, -0.207716f, -0.057175f, 0.055467f,  -0.210830f,
    960   -0.113005f,  -0.299177f, 0.068074f,  0.017929f,  -2.897598f, -0.260074f,
    961   -0.014422f,  -0.206467f, 1.246997f,  -0.372863f, -0.214160f, -0.114035f,
    962   5.805862f,   0.003611f,  -1.340990f, -0.021085f, -0.260431f, -0.002720f,
    963   -1.251640f,  -0.353531f, -0.304009f, -0.153376f,
    964 };
    965 
    966 static const float av1_ab_partition_nn_bias_32_layer1[LABEL_SIZE] = {
    967   -0.521497f, -1.061572f, -0.078756f, -0.660662f, -0.403741f, -0.960163f,
    968   0.001427f,  0.523607f,  0.225068f,  -0.055273f, 1.019519f,  1.181880f,
    969   -0.010198f, 0.130597f,  1.276752f,  2.028188f,
    970 };
    971 
    972 static const NN_CONFIG av1_ab_partition_nnconfig_32 = {
    973   FEATURE_SIZE,  // num_inputs
    974   LABEL_SIZE,    // num_outputs
    975   1,             // num_hidden_layers
    976   {
    977       64,  // num_hidden_nodes
    978   },
    979   {
    980       av1_ab_partition_nn_weights_32_layer0,
    981       av1_ab_partition_nn_weights_32_layer1,
    982   },
    983   {
    984       av1_ab_partition_nn_bias_32_layer0,
    985       av1_ab_partition_nn_bias_32_layer1,
    986   },
    987 };
    988 
    989 // nn model for ab partition pruning, 16x16.
    990 static const float av1_ab_partition_nn_weights_16_layer0[FEATURE_SIZE * 64] = {
    991   0.151902f,  0.007947f,  -1.788454f, 0.431869f,  -2.971387f, 0.923566f,
    992   1.632542f,  -1.665136f, -0.338632f, -5.075884f, 0.398267f,  0.030467f,
    993   2.263534f,  -0.045532f, -1.066128f, 0.915139f,  -0.560500f, -3.293125f,
    994   2.072793f,  -1.011414f, 0.122716f,  -0.060169f, -0.388860f, 0.031019f,
    995   -0.381861f, 0.001551f,  -0.328472f, 0.038296f,  -0.060398f, -0.375556f,
    996   0.209226f,  0.014764f,  -1.443469f, -0.345486f, 2.409269f,  1.524846f,
    997   -0.640666f, 1.322139f,  -2.074771f, -0.580944f, -0.203960f, -0.072893f,
    998   0.329701f,  0.115339f,  -1.339542f, 0.249024f,  -0.421545f, -0.409151f,
    999   -0.258293f, 0.836288f,  -0.073685f, -0.009624f, 0.895712f,  0.320639f,
   1000   0.451002f,  -1.544558f, 0.193709f,  -1.389012f, 1.305451f,  0.089795f,
   1001   0.050338f,  -0.017433f, -0.304667f, 0.500729f,  0.504346f,  0.073757f,
   1002   0.582649f,  -0.993623f, 1.766766f,  -3.067265f, -0.415774f, -0.006036f,
   1003   -1.245281f, 0.253205f,  -0.591245f, -0.626238f, 0.551852f,  0.593755f,
   1004   0.491023f,  1.099384f,  -0.348448f, 0.054564f,  -0.451422f, -0.375781f,
   1005   -0.248390f, -0.052548f, -0.380069f, -0.165391f, -0.297968f, -0.052142f,
   1006   -0.316381f, -0.045246f, -0.243905f, -0.034169f, -0.247523f, -0.180773f,
   1007   0.068066f,  -0.374920f, 0.057536f,  -0.189748f, 0.058375f,  -0.267749f,
   1008   -0.147286f, -0.246153f, 0.006183f,  -0.202029f, -0.059128f, 0.116852f,
   1009   0.134719f,  -0.126900f, -0.064646f, -0.196458f, -0.182331f, 0.108029f,
   1010   -0.264499f, 0.155816f,  -0.107255f, -0.056983f, -0.209771f, -0.099070f,
   1011   0.007313f,  -0.254124f, -0.231964f, -0.275972f, 0.032098f,  -0.264564f,
   1012   -0.208743f, 0.155599f,  -0.121511f, -0.156145f, -0.162315f, -0.059788f,
   1013   -0.257073f, -0.076654f, -0.110616f, -0.321675f, -0.051952f, 0.006301f,
   1014   -0.154114f, 0.017032f,  -0.017364f, -0.233247f, 0.009918f,  -0.179289f,
   1015   -0.190722f, 0.147106f,  -0.063910f, -0.396872f, -0.263123f, -0.003850f,
   1016   -0.040718f, -0.324699f, 0.118660f,  -0.170727f, -0.316788f, 0.100886f,
   1017   -0.202842f, 0.045371f,  0.150561f,  -0.057054f, -0.308150f, 0.028346f,
   1018   -0.381473f, -0.195365f, 0.026221f,  -0.281795f, 0.087204f,  0.047689f,
   1019   -0.027643f, -0.104724f, -0.089030f, -0.117661f, -0.349160f, 0.056982f,
   1020   -0.340273f, 0.048086f,  0.046103f,  -0.121527f, 0.021697f,  0.054109f,
   1021   -0.002768f, -0.008461f, -2.297240f, 0.124651f,  3.621661f,  -0.057120f,
   1022   -1.151656f, 2.296894f,  -3.678720f, -0.290240f, 0.087683f,  -0.186389f,
   1023   0.007656f,  -0.090236f, -0.245217f, 0.110389f,  -0.251719f, -0.029084f,
   1024   -0.128203f, -0.100005f, -0.032779f, 0.007281f,  -0.366596f, -0.267870f,
   1025   -0.215620f, 0.047687f,  0.010303f,  0.097980f,  -0.191569f, -0.341162f,
   1026   0.119249f,  0.026279f,  -2.161546f, 0.459591f,  1.290566f,  1.791797f,
   1027   -0.409835f, 0.127081f,  -1.156367f, 0.198286f,  0.099561f,  -0.067445f,
   1028   -0.034352f, 0.017966f,  -0.277380f, -0.057220f, -0.174198f, -0.014164f,
   1029   0.146090f,  -0.357530f, 0.097644f,  -0.000932f, 0.446603f,  -0.066793f,
   1030   2.448620f,  0.937617f,  -1.232922f, 0.313183f,  0.816827f,  -0.275115f,
   1031   -0.245205f, -0.126895f, 0.156668f,  -0.186977f, -0.273505f, 0.013315f,
   1032   0.168629f,  -0.089084f, 0.006166f,  -0.116107f, -0.199316f, -0.024010f,
   1033   -0.242303f, 0.011612f,  -0.218485f, -0.229661f, -0.123922f, 0.136699f,
   1034   0.006732f,  -0.148718f, -0.164225f, 0.116063f,  1.587898f,  0.690519f,
   1035   0.360566f,  0.009739f,  -0.678702f, -0.046003f, 0.126984f,  0.605212f,
   1036   1.240663f,  -0.000228f, -1.119369f, -0.415589f, -0.721003f, 0.097936f,
   1037   -1.410586f, -2.358833f, -2.773129f, -3.983361f, -0.087144f, -0.050029f,
   1038   -0.242255f, 0.137424f,  -0.307490f, -0.084637f, -0.023812f, -0.196582f,
   1039   -0.078695f, 0.038257f,  -0.012110f, -0.263521f, 0.009839f,  -0.109125f,
   1040   -0.226036f, 0.060712f,  0.093671f,  0.153143f,  0.039116f,  -0.290891f,
   1041   0.227057f,  -0.204633f, -0.207539f, -0.148242f, 0.046204f,  -0.231268f,
   1042   -0.209315f, -0.307579f, -0.436556f, 0.023475f,  0.131793f,  -0.038301f,
   1043   1.650584f,  0.392570f,  1.446576f,  1.254380f,  -0.516867f, -0.057116f,
   1044   0.149320f,  0.414424f,  -0.246309f, 0.003877f,  -0.480238f, -1.037035f,
   1045   -0.830779f, -1.122244f, -0.408267f, -0.253956f, 0.382005f,  0.940609f,
   1046   -1.113370f, -0.018554f, 0.141064f,  -0.182504f, 1.270707f,  0.414904f,
   1047   -0.216036f, 0.203831f,  0.450716f,  -0.452909f, 0.139358f,  -0.027143f,
   1048   1.956892f,  1.643732f,  -0.867839f, -0.620520f, -0.334607f, -0.519982f,
   1049   0.205023f,  0.661159f,  -0.000809f, 0.049033f,  -0.348579f, -0.200338f,
   1050   -0.362144f, -0.346590f, -0.230096f, 0.180746f,  -0.149954f, -0.253429f,
   1051   -0.378170f, -0.040724f, -0.041597f, 0.243659f,  -0.472181f, 0.015401f,
   1052   -0.180376f, 0.153139f,  -0.247738f, -0.010485f, -0.157158f, 0.016825f,
   1053   -0.238925f, -0.265798f, -0.318374f, 0.142352f,  -0.210520f, 0.051928f,
   1054   -0.352190f, -0.179052f, -0.185498f, 0.025540f,  -0.111667f, -0.235187f,
   1055   -0.215454f, 0.010931f,  -0.238372f, -0.126659f, 0.075691f,  -0.091167f,
   1056   -2.462379f, -0.007950f, -0.637990f, 0.285554f,  -0.051275f, 0.282279f,
   1057   -0.744083f, -0.570646f, 0.592198f,  1.421332f,  -0.256027f, -0.140315f,
   1058   0.160247f,  -0.063185f, -0.055895f, -0.199864f, -0.287353f, -0.074561f,
   1059   -0.071228f, 0.055864f,  -1.084764f, -0.263409f, 0.779266f,  0.228187f,
   1060   0.375013f,  0.121204f,  -0.656948f, 0.533561f,  0.272671f,  -0.015423f,
   1061   -0.124180f, -0.009127f, 2.934838f,  -0.150998f, 1.163152f,  0.081997f,
   1062   -4.715939f, -3.676595f, -1.524886f, -0.167593f, 0.281186f,  0.024046f,
   1063   -1.451709f, 0.332558f,  0.990504f,  0.376290f,  -1.466773f, -0.448439f,
   1064   -2.929108f, -4.255188f, 0.065238f,  0.019950f,  1.372393f,  0.444052f,
   1065   -2.538772f, 1.579767f,  -0.464911f, -1.866114f, 1.053958f,  0.434467f,
   1066   -0.125964f, 0.034671f,  0.077116f,  -0.138466f, -0.413395f, -0.223453f,
   1067   -0.172127f, -0.251265f, -0.048239f, -0.395519f, 0.023141f,  0.037459f,
   1068   -0.249593f, -0.062215f, -0.047209f, -0.435189f, -0.164155f, -0.077590f,
   1069   -0.241164f, -0.126128f, -0.038243f, -0.180888f, 0.198840f,  -0.328036f,
   1070   -0.169790f, 0.036506f,  0.052572f,  -0.183570f, -0.073617f, -0.244959f,
   1071   0.266498f,  0.032846f,  -1.902106f, 0.486078f,  2.414993f,  0.975182f,
   1072   -0.382875f, 1.647810f,  -2.197017f, -0.890107f, 0.221287f,  0.010889f,
   1073   3.817042f,  0.572728f,  0.092466f,  0.473337f,  -1.634659f, -1.069455f,
   1074   1.486776f,  -1.023850f, 0.088184f,  0.008842f,  0.518202f,  0.270259f,
   1075   1.757191f,  -0.121839f, -2.912229f, -1.250866f, -2.381808f, 0.335309f,
   1076   -0.120079f, -0.061294f, -0.058725f, -0.315169f, -0.262443f, 0.072434f,
   1077   -0.267836f, -0.319354f, -0.274975f, 0.068970f,  -0.406467f, 0.044074f,
   1078   -0.152311f, -0.333656f, -0.228355f, -0.185613f, 0.017346f,  -0.177674f,
   1079   -0.090675f, -0.102047f, -0.011768f, -0.025280f, -0.271661f, 0.098099f,
   1080   -0.312272f, -0.222217f, -0.100548f, 0.106260f,  -0.034655f, 0.135109f,
   1081   -0.021276f, 0.018177f,  -0.353097f, -0.011128f, 0.061136f,  -0.511662f,
   1082   -0.223236f, -0.308841f, 0.118789f,  -0.154628f, -0.053178f, -0.055973f,
   1083   0.013175f,  -0.368337f, -0.090863f, -0.116920f, 0.178990f,  -0.025278f,
   1084   -0.190553f, -0.238092f, 0.303943f,  -0.024944f, 0.719373f,  0.384332f,
   1085   -0.378480f, -0.423316f, 0.709922f,  0.758514f,  -1.559023f, -2.503173f,
   1086   0.068652f,  -0.234741f, -0.182932f, 0.037878f,  0.020684f,  -0.174142f,
   1087   -0.182300f, -0.052796f, -0.219145f, 0.113028f,  -1.041826f, 0.035317f,
   1088   0.919904f,  -0.676011f, 0.652297f,  1.456447f,  -0.166904f, -0.861823f,
   1089   0.895827f,  0.429821f,  -0.180376f, -0.076587f, -0.273945f, -0.288990f,
   1090   -0.206692f, -0.080745f, -0.085444f, 0.186953f,  -0.050135f, 0.044243f,
   1091   -0.391706f, -0.160498f, -0.292268f, 0.164060f,  0.412649f,  0.211611f,
   1092   -0.327294f, -0.919399f, 0.320297f,  0.385284f,  -0.088848f, -0.072556f,
   1093   -0.384813f, -0.176267f, -0.065918f, 0.134724f,  -0.231104f, -0.337707f,
   1094   -0.195442f, -0.263569f, 0.098090f,  -0.341411f, -0.189211f, -0.439276f,
   1095   -0.404046f, 0.262491f,  -0.311093f, -0.086454f, -0.013400f, -0.061447f,
   1096   -0.026945f, -0.112036f, -0.322985f, 0.078500f,  -0.230205f, -0.344535f,
   1097   -0.021087f, 0.110220f,  -0.128671f, 0.044219f,
   1098 };
   1099 
   1100 static const float av1_ab_partition_nn_bias_16_layer0[64] = {
   1101   2.936406f,  -0.396539f, -0.110456f, -1.254954f, 0.785350f,  0.516290f,
   1102   -0.172341f, 0.254386f,  -0.192465f, -0.106751f, -0.055518f, -0.094994f,
   1103   0.000000f,  -0.065018f, -0.004908f, -0.130483f, -0.119580f, -0.142072f,
   1104   0.457446f,  -0.125051f, -0.107712f, 0.714607f,  -0.140809f, -1.788650f,
   1105   -0.087199f, 0.000000f,  -1.290050f, 0.443930f,  -0.110634f, -0.109380f,
   1106   -0.188213f, -1.414179f, 1.193579f,  0.388775f,  -0.873193f, -0.110050f,
   1107   -0.072565f, -0.117050f, -0.119132f, 0.456959f,  -0.132069f, 0.131974f,
   1108   1.160474f,  1.746465f,  0.442628f,  -0.188849f, -0.207794f, -0.108364f,
   1109   -0.856655f, -2.141620f, 0.335476f,  -0.105508f, -0.212162f, -0.109319f,
   1110   -0.237213f, -0.109980f, -0.291044f, -0.137877f, 0.470191f,  -0.023908f,
   1111   0.123809f,  -0.109797f, 0.200510f,  -0.147542f,
   1112 };
   1113 
   1114 static const float av1_ab_partition_nn_weights_16_layer1[64 * LABEL_SIZE] = {
   1115   -6.823716f, 1.406568f,  -0.144009f, 2.228765f,  0.838336f,  0.738107f,
   1116   -0.319014f, -0.148756f, 0.240862f,  -0.111089f, -0.004241f, 0.025758f,
   1117   -0.193820f, -0.246362f, -0.181363f, -0.201556f, 0.024268f,  0.252994f,
   1118   -0.289443f, 0.194932f,  0.057467f,  0.724735f,  0.014063f,  1.361352f,
   1119   0.025191f,  0.024274f,  0.231462f,  -7.227959f, -0.094515f, 0.039946f,
   1120   0.412719f,  0.812318f,  3.038903f,  -0.286289f, 0.647482f,  -0.115114f,
   1121   0.053590f,  0.066069f,  0.153134f,  0.996250f,  -0.125700f, 0.951365f,
   1122   -6.243494f, -4.827697f, 0.566320f,  0.239515f,  -0.099702f, 0.054546f,
   1123   1.847330f,  3.680076f,  -3.049829f, -0.127709f, 0.068469f,  -0.017794f,
   1124   0.223864f,  -0.106778f, -0.020425f, -0.040226f, -0.251890f, -0.168673f,
   1125   -0.552073f, 0.043311f,  0.218668f,  0.033209f,  -3.199210f, 0.193079f,
   1126   0.321406f,  0.718307f,  -0.181418f, -0.459612f, -1.981170f, 0.968496f,
   1127   -0.029757f, -0.130065f, 0.043782f,  0.072394f,  -0.088686f, 0.025322f,
   1128   0.129882f,  0.101324f,  0.335707f,  0.072714f,  -2.079774f, 0.203997f,
   1129   0.239321f,  -0.301757f, 0.257845f,  1.288382f,  -0.031275f, -0.234194f,
   1130   0.310722f,  2.045469f,  0.034716f,  0.135638f,  -0.251388f, 0.320071f,
   1131   -1.065301f, -0.322731f, -0.545028f, 0.226276f,  0.090799f,  0.019289f,
   1132   0.048950f,  -1.079300f, 0.231938f,  0.083683f,  4.762127f,  0.145037f,
   1133   -0.145549f, 0.075592f,  0.172336f,  0.108175f,  0.333751f,  1.090501f,
   1134   1.056114f,  0.047073f,  0.182052f,  -0.081587f, 0.089900f,  0.339286f,
   1135   2.049988f,  0.073585f,  0.537355f,  -0.243322f, -0.010179f, -0.052601f,
   1136   -0.174915f, 0.117793f,  2.222990f,  -2.520837f, -0.092699f, 1.199887f,
   1137   0.138720f,  0.679918f,  -0.463155f, -0.659496f, -0.109913f, -0.003398f,
   1138   0.114633f,  -0.128377f, 0.092970f,  -0.107489f, -0.191078f, 0.185182f,
   1139   0.216980f,  -0.019343f, 3.443133f,  0.287953f,  0.099314f,  0.985958f,
   1140   0.157268f,  -0.606516f, 0.049418f,  -0.221809f, -0.453081f, -0.344796f,
   1141   -0.003735f, -0.107269f, -0.128541f, -0.259543f, -0.934806f, -0.542456f,
   1142   -1.011192f, 0.022795f,  0.186363f,  -0.076356f, -0.050932f, -0.165098f,
   1143   0.168177f,  -0.101596f, -5.270886f, 2.553943f,  -0.440870f, -0.017494f,
   1144   0.215208f,  -0.017032f, 1.495915f,  -4.304677f, 0.762211f,  0.182937f,
   1145   0.254406f,  -0.029433f, -0.088364f, -0.110160f, -0.108257f, -0.036538f,
   1146   0.737697f,  -0.234989f, 0.168095f,  0.245118f,  -0.077262f, 0.195718f,
   1147   0.753302f,  -1.637869f, 0.126227f,  0.982129f,  -0.121444f, -0.295570f,
   1148   -1.215799f, 0.147867f,  -0.068496f, 0.132726f,  -0.005772f, -0.181774f,
   1149   0.126513f,  0.204723f,  -0.366123f, 0.103906f,  -0.148053f, -0.075272f,
   1150   0.243884f,  -0.104828f, 0.198988f,  0.501034f,  -0.112671f, 0.111421f,
   1151   0.167508f,  -0.117803f, -0.738624f, 2.046292f,  0.124011f,  0.057983f,
   1152   -0.359154f, -0.648883f, -0.259462f, -0.459041f, -2.501223f, -0.065138f,
   1153   0.122417f,  0.060291f,  -0.129033f, -0.843086f, 0.268241f,  -0.399927f,
   1154   1.585888f,  1.816393f,  -0.631427f, 0.127826f,  0.088105f,  0.073488f,
   1155   0.717694f,  -1.497362f, 2.608528f,  0.066896f,  -0.079230f, 0.223436f,
   1156   -0.010530f, 0.175310f,  1.120365f,  0.034391f,  0.835312f,  0.071652f,
   1157   -0.080615f, 0.111395f,  0.162742f,  0.079927f,  -3.859582f, -0.638431f,
   1158   -0.167880f, -0.992659f, -0.885355f, -1.276197f, 1.334344f,  0.931940f,
   1159   -0.078244f, -0.149030f, -0.070974f, -0.133566f, 0.200034f,  0.102793f,
   1160   -0.048546f, 0.063545f,  0.023864f,  -0.190863f, 1.934257f,  -0.136286f,
   1161   -0.107916f, -0.637468f, 0.066449f,  1.089693f,  -0.214047f, -0.265780f,
   1162   0.899660f,  -0.130333f, 0.288311f,  -0.049024f, 0.090202f,  0.487969f,
   1163   0.339704f,  0.858479f,  0.841253f,  -0.184100f, -0.637070f, -0.125071f,
   1164   -0.077650f, -0.087877f, 0.202268f,  -0.027300f, 2.842862f,  -0.100698f,
   1165   -0.259080f, 0.260556f,  0.157912f,  -0.070364f, 0.467190f,  1.200037f,
   1166   1.419317f,  -0.033588f, -0.227824f, 0.292617f,  0.228574f,  0.213839f,
   1167   -1.091099f, -0.022258f, -1.294681f, 0.136118f,  0.081652f,  -0.185359f,
   1168   -0.039706f, 0.191407f,  -2.053219f, -0.261934f, 0.047812f,  -0.029536f,
   1169   -0.823869f, -1.090534f, -0.755890f, 0.441035f,  -0.167945f, 0.231441f,
   1170   -0.135013f, -0.260762f, 0.256872f,  0.130339f,  -0.243751f, 0.189760f,
   1171   -0.288454f, 0.145363f,  0.338490f,  0.403898f,  -0.022814f, -1.263598f,
   1172   -0.101315f, 0.860135f,  0.136511f,  0.028942f,  0.574047f,  2.656370f,
   1173   0.037587f,  -0.188690f, -0.125312f, 1.100435f,  -1.080402f, 0.380905f,
   1174   0.004635f,  0.097144f,  -0.214309f, 0.085552f,  -0.285066f, -0.705134f,
   1175   -0.054704f, -0.319951f, 5.486626f,  0.958158f,  -1.380585f, 0.223340f,
   1176   -0.169167f, -0.170697f, -0.216748f, 0.324232f,  2.684204f,  -0.008490f,
   1177   -0.211052f, -0.201190f, 0.123466f,  -0.000234f, 0.579907f,  0.096938f,
   1178   -0.042745f, 0.201855f,  0.157195f,  -0.261440f, 0.029699f,  -0.046599f,
   1179   1.618216f,  -2.596280f, -0.377420f, -0.526725f, -0.493592f, -0.579615f,
   1180   0.579699f,  -0.100392f, 0.150694f,  0.061794f,  0.200425f,  -0.062515f,
   1181   -0.179122f, 0.250112f,  -0.344675f, -0.118359f, -0.095670f, 0.152311f,
   1182   3.662276f,  -0.154921f, -0.312991f, 0.972008f,  -0.308596f, -0.190426f,
   1183   0.133889f,  -0.238673f, -0.094726f, 1.683835f,  -0.215629f, -0.198890f,
   1184   -0.035278f, -0.367973f, -0.822435f, 0.240848f,  -0.194656f, 0.034655f,
   1185   -0.079424f, 0.146670f,  0.026646f,  -0.034507f, 0.059467f,  -0.153109f,
   1186   -0.431033f, 2.552991f,  -1.894091f, -0.180462f, -0.306839f, -0.025648f,
   1187   1.026326f,  -3.096230f, 1.346935f,  0.033633f,  -0.181827f, 0.094376f,
   1188   0.001696f,  -0.379264f, -1.069503f, -0.140972f, -0.208769f, -0.195239f,
   1189   0.281795f,  -0.127251f, 0.180776f,  0.067763f,  0.697124f,  -1.040779f,
   1190   0.111280f,  0.188351f,  -0.340234f, -0.207790f, -0.720075f, -0.137409f,
   1191   -0.070310f, -0.032918f, -0.060787f, 0.131484f,  -0.077845f, -0.258652f,
   1192   0.056911f,  -0.062034f, 0.007663f,  -0.185100f, 1.340361f,  0.014096f,
   1193   -0.124602f, 0.194241f,  0.128383f,  0.360465f,  0.082979f,  -0.050475f,
   1194   -0.519294f, 3.323262f,  0.067014f,  0.221203f,  -0.085082f, -0.228606f,
   1195   -0.916668f, -0.022643f, -1.386737f, -0.131902f, -0.349952f, -0.032874f,
   1196   -0.189190f, -0.898790f, -0.102394f, -1.017387f, 2.214050f,  1.790253f,
   1197   -1.913561f, -0.043716f, -0.214924f, -0.194598f, -0.064723f, -1.671793f,
   1198   2.251166f,  -0.146007f, 0.138527f,  -0.003134f, 0.103665f,  0.006928f,
   1199   -0.240253f, -0.227464f, 0.578437f,  -0.214724f, 0.503085f,  0.158093f,
   1200   0.033091f,  0.008061f,  4.815371f,  2.132264f,  0.281850f,  -2.288560f,
   1201   -0.145012f, 1.296832f,  -0.362401f, -0.403252f, 0.109873f,  0.185746f,
   1202   0.244764f,  0.172367f,  -0.185588f, 0.139801f,  -0.178254f, 0.068629f,
   1203   0.358488f,  -0.153969f, -6.433524f, 0.225983f,  -0.138123f, -0.095971f,
   1204   -0.036089f, -1.400083f, 0.265908f,  0.257787f,  0.181144f,  -1.647228f,
   1205   -0.136289f, -0.074206f, 0.122988f,  -0.088895f, -1.266717f, 0.006010f,
   1206   0.536681f,  0.263061f,  -0.032207f, -0.155136f, 0.086431f,  0.441950f,
   1207   -0.060755f, -0.280683f, -0.783475f, -2.567033f, 1.093221f,  0.117667f,
   1208   -0.000408f, 0.225719f,  -2.199698f, 0.141447f,  -1.459051f, 0.051315f,
   1209   0.203228f,  0.354432f,  -0.005775f, -0.028073f, -0.965817f, 0.231083f,
   1210   -0.666884f, 0.026283f,  -0.317486f, 0.210754f,  0.123897f,  0.223827f,
   1211   4.214405f,  1.457334f,  -0.253945f, -1.306733f, -0.391235f, 0.451154f,
   1212   -1.553888f, -0.353429f, 0.069533f,  0.159278f,  -0.173836f, -0.004952f,
   1213   -0.137033f, 0.127012f,  0.143600f,  0.051587f,  -0.070549f, 0.066509f,
   1214   -5.776547f, 0.180021f,  -0.189183f, -1.288504f, -0.233575f, -1.473873f,
   1215   0.140940f,  0.144451f,  -0.104534f, 2.089873f,  -0.168168f, 0.110726f,
   1216   0.132134f,  -0.215223f, -1.682754f, 0.157757f,  -0.146163f, 0.064882f,
   1217   0.117313f,  -0.038780f, -0.124720f, -0.501697f, 0.092047f,  -0.233992f,
   1218   3.324976f,  0.516601f,  1.294202f,  0.119989f,  0.061055f,  0.043420f,
   1219   -2.750727f, -0.382812f, -0.648496f, -0.115353f, -0.334205f, 0.024354f,
   1220   -0.282998f, -0.282705f, 0.073798f,  0.169851f,  0.135651f,  0.182677f,
   1221   -0.040220f, 0.132462f,  -0.303120f, -0.230113f, 6.165739f,  -0.258596f,
   1222   0.024127f,  -1.388283f, -0.006042f, 0.572600f,  0.348411f,  -0.387376f,
   1223   -0.075845f, 0.122319f,  -0.029616f, 0.077873f,  0.154763f,  0.049073f,
   1224   0.018597f,  0.102688f,  -0.204165f, 0.020734f,  -1.389133f, -0.032854f,
   1225   -0.147561f, 0.853944f,  0.132100f,  -3.259659f, 0.243745f,  0.181529f,
   1226   -0.738414f, 1.509994f,  0.023470f,  -0.005329f, 0.066115f,  -1.345081f,
   1227   -1.455402f, -0.172023f, -0.194625f, 0.071885f,  -0.201742f, -0.262402f,
   1228   0.077601f,  -0.048938f, 0.257993f,  -0.504029f, -2.032415f, 1.158880f,
   1229   0.448647f,  -0.025633f, 0.117586f,  -0.072275f, -0.673744f, -3.854342f,
   1230   -0.983843f, 0.047766f,  -0.017193f, -0.215775f, -0.158743f, -0.232042f,
   1231   -0.509112f, 0.148812f,  0.130122f,  0.006486f,  -0.099016f, 0.022514f,
   1232   -0.486850f, -0.059623f, 4.012731f,  0.025454f,  0.029059f,  -0.783546f,
   1233   -0.295260f, 0.322521f,  -0.473201f, -0.172100f, -0.100087f, -0.076516f,
   1234   -0.258367f, -0.112897f, 0.269364f,  -0.065912f, 0.169022f,  -0.178783f,
   1235   -0.095114f, 0.122089f,  -2.790099f, -0.100431f, -0.087963f, -0.009431f,
   1236   -0.087819f, -2.774399f, -0.100757f, 0.013005f,  -0.964533f, 3.236665f,
   1237   -0.354903f, -0.144169f, -0.166869f, -1.396513f, -0.931271f, -0.046261f,
   1238   -1.799262f, -0.365269f, 0.108611f,  0.037994f,  0.024747f,  -1.073639f,
   1239   -0.203158f, -0.935006f, 1.880891f,  1.578385f,  0.726272f,  -0.024546f,
   1240   -0.011626f, -0.151363f, -1.121716f, -1.787484f, 0.232806f,  0.075451f,
   1241   0.182899f,  0.092215f,  -0.207347f, -0.030111f, 0.054316f,  0.192481f,
   1242   0.594639f,  -0.247694f, 0.547471f,  -0.032094f, -0.065000f, 0.007198f,
   1243   1.605377f,  -0.155945f, -0.066200f, -2.343716f, -1.016283f, -0.079321f,
   1244   0.919365f,  0.599980f,  0.125545f,  0.265813f,  0.246884f,  0.095385f,
   1245   -0.260374f, -0.202916f, -0.042770f, 0.234967f,  -0.233139f, -0.326994f,
   1246   -1.375256f, 0.121766f,  0.077433f,  -1.103569f, 0.019497f,  -1.029185f,
   1247   0.253905f,  0.206569f,  0.187334f,  -0.237089f, -0.294351f, 0.164137f,
   1248   0.149696f,  -0.749787f, -0.413433f, 0.976587f,  1.027976f,  -0.285264f,
   1249   0.209273f,  -0.124762f, 0.050884f,  0.250764f,  -0.082031f, -0.646520f,
   1250   4.116680f,  0.437336f,  0.671684f,  0.129509f,  -0.078462f, 0.014072f,
   1251   -0.678232f, 0.094831f,  1.125624f,  0.207070f,  -0.154750f, -0.025780f,
   1252   -0.103030f, 0.118019f,  -0.908186f, -0.263546f, -1.555324f, -0.236887f,
   1253   -0.217854f, -0.051790f, 0.017915f,  0.171001f,  1.355562f,  0.094603f,
   1254   -0.233929f, -1.282169f, -0.773183f, -0.161682f, -0.834565f, -0.286776f,
   1255   -0.298901f, 0.038162f,  0.251899f,  0.039612f,  -0.022935f, -0.232308f,
   1256   -0.043855f, -0.192892f, -0.279009f, -0.182234f, -1.272808f, -0.070344f,
   1257   -0.092432f, -1.915946f, -0.134373f, -1.405496f, -0.067071f, -0.131922f,
   1258   0.185269f,  1.465082f,  0.040240f,  0.112665f,  0.144329f,  -0.286112f,
   1259   -0.617649f, 0.916177f,  0.221044f,  -0.079867f, 0.170251f,  -0.093638f,
   1260   -0.212620f, -0.305945f, -0.234356f, -0.482501f, 3.928472f,  1.241179f,
   1261   0.355922f,  -0.170848f, -0.189168f, 0.080225f,  -1.357793f, 0.190890f,
   1262   0.976800f,  -0.068070f, -0.016295f, -0.088623f, -0.129560f, -0.212267f,
   1263   -0.071537f, -0.219501f, -0.655198f, -0.225188f, -0.116024f, 0.224174f,
   1264   -0.049715f, -0.178005f, 3.029985f,  -1.141546f, 0.080066f,  -1.932316f,
   1265   -0.641137f, -0.189564f, 0.935080f,  0.136119f,  0.015558f,  -0.179331f,
   1266   0.204571f,  0.020350f,  0.009362f,  0.108478f,  0.037076f,  -0.049009f,
   1267   0.081090f,  -0.180202f, 1.455561f,  -0.081559f, 0.059361f,  0.484971f,
   1268   0.160923f,  -2.170744f, -0.013204f, 0.126561f,  -0.407122f, 1.223661f,
   1269   0.044262f,  0.118044f,  0.058274f,  -1.747100f, -0.171318f, 0.971374f,
   1270   0.306995f,  -0.103268f, -0.319443f, -0.333176f, -0.038608f, 0.119674f,
   1271   -0.106479f, -0.907933f, 1.121231f,  1.673840f,  -0.421458f, -0.021146f,
   1272   -0.254838f, 0.097632f,  0.235109f,  -2.901782f, 0.289518f,  -0.355459f,
   1273   -0.068264f, -0.179121f, 0.068560f,  -0.047570f, -0.522523f, -0.228963f,
   1274   -1.037158f, -0.163723f, 0.280563f,  -0.000868f, -0.197220f, -0.239329f,
   1275   1.985274f,  -0.256181f, -0.064341f, -0.822417f, -0.465140f, -0.010942f,
   1276   -0.792024f, -0.114290f, 0.060969f,  0.104106f,  -0.252123f, -0.150400f,
   1277   -0.133277f, 0.267147f,  0.274413f,  0.223744f,  -0.180223f, -0.345415f,
   1278   -0.104883f, 0.119210f,  -0.095041f, -0.301635f, 0.013175f,  -2.128121f,
   1279   -0.147208f, -0.151509f, -0.692013f, 3.418555f,  -0.016541f, 0.171511f,
   1280   0.107159f,  -1.516672f, 0.127408f,  0.687035f,  -0.906486f, -0.145463f,
   1281   -0.169382f, -0.143906f, 0.125091f,  -0.960645f, -0.180869f, -0.716908f,
   1282   2.840951f,  1.904919f,  -0.416268f, -0.425181f, -0.194697f, -0.075932f,
   1283   -0.950604f, -1.599800f, 0.943671f,  -0.022744f, -0.270492f, 0.080843f,
   1284   -0.372916f, 0.047838f,  -0.100300f, -0.026600f, 0.011733f,  -0.226051f,
   1285   0.172790f,  -0.172982f, 0.041258f,  -0.299379f,
   1286 };
   1287 
   1288 static const float av1_ab_partition_nn_bias_16_layer1[LABEL_SIZE] = {
   1289   -0.053805f, -1.248639f, 0.520965f, -0.904962f, -0.126425f, -0.118798f,
   1290   0.748430f,  0.203096f,  0.059317f, 0.418219f,  0.841294f,  0.402693f,
   1291   -0.658522f, 0.723479f,  0.544264f, 1.035225f,
   1292 };
   1293 
   1294 static const NN_CONFIG av1_ab_partition_nnconfig_16 = {
   1295   FEATURE_SIZE,  // num_inputs
   1296   LABEL_SIZE,    // num_outputs
   1297   1,             // num_hidden_layers
   1298   {
   1299       64,  // num_hidden_nodes
   1300   },
   1301   {
   1302       av1_ab_partition_nn_weights_16_layer0,
   1303       av1_ab_partition_nn_weights_16_layer1,
   1304   },
   1305   {
   1306       av1_ab_partition_nn_bias_16_layer0,
   1307       av1_ab_partition_nn_bias_16_layer1,
   1308   },
   1309 };
   1310 
   1311 #undef FEATURE_SIZE
   1312 #undef LABEL_SIZE
   1313 
   1314 #define FEATURE_SIZE 18
   1315 #define LABEL_SIZE 4
   1316 
   1317 static const float av1_4_partition_nn_weights_16_layer0[FEATURE_SIZE * 24] = {
   1318   -2.032866f, 0.056691f,  0.495960f,  0.778785f,  0.548153f,  -0.806942f,
   1319   0.481155f,  0.282298f,  0.584980f,  0.504688f,  0.209648f,  0.234616f,
   1320   0.213484f,  0.221969f,  0.205862f,  0.235054f,  0.317863f,  0.257139f,
   1321   0.529478f,  0.098122f,  -0.657532f, 0.036296f,  0.327728f,  1.323180f,
   1322   -0.813082f, 0.160216f,  -0.702030f, 0.722733f,  -0.270576f, -0.347416f,
   1323   -0.264700f, -0.254248f, 0.159820f,  0.087995f,  -0.184163f, 0.117357f,
   1324   0.074194f,  -0.667369f, 0.498246f,  0.420506f,  0.072409f,  -0.121581f,
   1325   0.315788f,  0.000525f,  0.414986f,  0.678166f,  -0.011230f, 0.188131f,
   1326   -0.227749f, 0.009564f,  0.108672f,  0.106923f,  -0.080695f, -0.279382f,
   1327   -0.061339f, -0.297835f, -0.134707f, 0.145865f,  -0.009655f, -0.000842f,
   1328   -0.047436f, -0.159149f, -0.320353f, -0.089646f, -0.344765f, 0.313416f,
   1329   -0.143413f, 0.279668f,  0.000885f,  -0.022380f, -0.140194f, -0.310473f,
   1330   0.252699f,  0.066204f,  0.477568f,  0.994609f,  -0.276000f, 1.213182f,
   1331   0.277028f,  -0.411570f, -0.211559f, 0.377815f,  0.121488f,  -0.100559f,
   1332   -0.317082f, -0.251039f, -0.335181f, -0.154114f, -0.052726f, -0.332558f,
   1333   -0.143196f, -0.334035f, 0.162305f,  0.142279f,  -0.001210f, -0.135252f,
   1334   -0.033562f, 0.204307f,  -0.039757f, -0.394174f, 0.126617f,  -0.128648f,
   1335   -0.410979f, 0.107641f,  -0.117573f, -0.326512f, 0.235166f,  0.084959f,
   1336   0.290063f,  -0.005838f, 0.459894f,  1.023709f,  -0.196145f, 1.100137f,
   1337   -0.319815f, -0.308526f, -0.443389f, -0.272769f, -0.035259f, -0.026932f,
   1338   -0.029743f, 0.125113f,  -0.131024f, -0.321458f, -0.143996f, 0.008714f,
   1339   -0.101234f, 0.079706f,  -1.128615f, -0.467381f, 0.220563f,  -0.409900f,
   1340   -0.435353f, 0.759499f,  -0.465799f, -0.394309f, 0.176282f,  -0.086275f,
   1341   -0.161225f, -0.354814f, 0.562871f,  0.418253f,  0.414361f,  0.445480f,
   1342   -0.995903f, -0.086632f, -0.230645f, 0.354656f,  -0.317576f, 0.079926f,
   1343   0.424369f,  0.997232f,  -0.304388f, 1.071667f,  -0.023540f, 0.029677f,
   1344   0.108564f,  0.183581f,  -0.201395f, -0.054854f, -0.193039f, -0.049899f,
   1345   -0.271949f, -0.358483f, 0.304930f,  0.023823f,  -0.009319f, -0.214247f,
   1346   0.100712f,  -0.050162f, 0.327103f,  -0.212999f, -0.030496f, 0.316380f,
   1347   -0.439589f, -0.249959f, 0.229777f,  -0.353664f, -0.384559f, 0.114236f,
   1348   0.023119f,  0.007927f,  0.618368f,  0.957759f,  -0.019780f, -1.002389f,
   1349   0.564277f,  -0.839531f, 1.040445f,  0.054340f,  0.031908f,  -0.032893f,
   1350   -0.019170f, -0.042011f, 0.568928f,  0.362567f,  -0.559999f, -0.605344f,
   1351   -0.586146f, -0.290778f, 0.195943f,  -0.109580f, -0.088898f, -0.113054f,
   1352   0.293282f,  0.429019f,  0.306136f,  0.863025f,  0.021234f,  0.125770f,
   1353   -0.097108f, -0.072659f, -0.137053f, -0.191631f, 0.106281f,  0.064151f,
   1354   0.029883f,  0.076287f,  0.757543f,  0.276713f,  -2.529775f, -0.351727f,
   1355   -1.832316f, 0.544780f,  -0.944529f, 0.509705f,  -0.010236f, -0.016181f,
   1356   0.021520f,  0.086417f,  0.041312f,  0.296853f,  -0.372378f, 0.354446f,
   1357   -1.366762f, 0.048875f,  0.464918f,  -0.007450f, 0.750013f,  -0.360261f,
   1358   0.518532f,  0.753776f,  0.641448f,  0.710746f,  0.250866f,  0.257063f,
   1359   0.283421f,  0.253585f,  0.170303f,  0.210426f,  0.208842f,  0.158000f,
   1360   -0.033144f, 0.130748f,  0.907147f,  0.409248f,  -0.854301f, -0.981307f,
   1361   0.294427f,  -0.507137f, 1.079967f,  0.203203f,  0.383890f,  0.368278f,
   1362   0.305122f,  0.449288f,  -0.044507f, -0.547263f, -0.298245f, -0.497834f,
   1363   0.007016f,  -0.101982f, -0.073488f, -0.096111f, -0.479418f, -0.045497f,
   1364   0.033502f,  -0.018578f, -0.231531f, 0.177949f,  0.099564f,  -0.010233f,
   1365   -0.333055f, -0.078586f, -0.417867f, 0.171271f,  0.013662f,  -0.143599f,
   1366   -0.117296f, 0.135382f,  0.048321f,  0.000924f,  -0.055024f, -0.405595f,
   1367   -0.068260f, -0.271011f, -0.436425f, 0.206751f,  -0.899890f, 0.605510f,
   1368   0.535649f,  -0.238919f, -0.037619f, -0.213734f, -0.391360f, -0.132344f,
   1369   0.004660f,  0.176644f,  -1.008475f, -0.038895f, 0.155429f,  -0.095229f,
   1370   -0.680124f, -0.258063f, -0.261901f, 0.110380f,  -0.337649f, -0.505870f,
   1371   -1.428536f, 0.610629f,  0.254905f,  0.045098f,  0.044109f,  0.172329f,
   1372   0.060001f,  -0.234009f, -0.184855f, -0.153028f, -0.140897f, -0.152006f,
   1373   -0.312134f, 0.081261f,  0.160166f,  0.112690f,  0.266081f,  0.030175f,
   1374   -0.242746f, 0.000754f,  -0.341811f, -0.149774f, -0.017484f, -0.301342f,
   1375   -0.121466f, 0.067300f,  0.342176f,  0.474538f,  0.085441f,  -0.263935f,
   1376   0.479235f,  -0.003713f, -0.784840f, 0.119480f,  0.456632f,  -0.640082f,
   1377   -0.080575f, -0.744403f, 0.259970f,  0.034667f,  -0.274641f, -0.257594f,
   1378   -1.121124f, -0.003745f, -0.420693f, 0.300441f,  -0.100976f, -1.049016f,
   1379   0.201960f,  0.113054f,  0.187010f,  1.237427f,  0.054803f,  -0.028673f,
   1380   0.003596f,  -0.034724f, 0.117246f,  0.190977f,  0.278915f,  0.224307f,
   1381   0.017852f,  -0.336233f, -0.372311f, -0.182284f, -0.143510f, 0.331466f,
   1382   0.045698f,  -0.301095f, 0.184447f,  0.348240f,  -0.017021f, -0.145064f,
   1383   -0.000221f, -0.382256f, -0.302683f, -0.083927f, -0.008070f, 0.217907f,
   1384   0.647597f,  -0.050490f, -0.572736f, -0.985748f, -0.289943f, 0.041391f,
   1385   -0.795464f, -0.186680f, -0.354062f, -0.617400f, -0.282783f, -0.170450f,
   1386   -0.197197f, -0.146496f, -0.173692f, -0.106277f, -0.071004f, -0.124405f,
   1387   -0.971412f, 0.038542f,  0.705204f,  0.887113f,  0.150430f,  -0.243676f,
   1388   0.638410f,  0.320953f,  0.776676f,  0.527584f,  0.070389f,  0.051554f,
   1389   0.177519f,  0.140451f,  0.128892f,  0.087771f,  0.197660f,  0.194764f,
   1390 };
   1391 
   1392 static const float av1_4_partition_nn_bias_16_layer0[24] = {
   1393   0.614063f,  -0.384872f, 0.084884f,  -0.023980f, -0.378765f, -0.082312f,
   1394   -0.458271f, 0.189578f,  -0.046169f, -0.073308f, -0.372322f, 0.162793f,
   1395   0.148803f,  0.829214f,  -0.221162f, -0.111157f, -0.017484f, -0.280596f,
   1396   -0.031905f, -0.143459f, 0.078823f,  -0.021940f, 0.026834f,  0.257472f,
   1397 };
   1398 
   1399 static const float av1_4_partition_nn_weights_16_layer1[24 * LABEL_SIZE] = {
   1400   -0.985391f, 0.587616f,  0.740683f,  0.192066f,  0.447080f,  -0.016585f,
   1401   0.680449f,  0.028983f,  0.643111f,  0.234338f,  0.107148f,  0.328456f,
   1402   -0.216394f, 1.106838f,  -0.179062f, -0.129108f, -0.121655f, -0.151340f,
   1403   -0.306017f, -0.350989f, 0.859284f,  -0.372831f, -0.954419f, 0.250495f,
   1404   1.046732f,  0.287923f,  -0.421088f, 0.326613f,  -0.314396f, -0.084757f,
   1405   -0.474228f, 0.687999f,  0.052334f,  0.441708f,  -0.630698f, -0.350348f,
   1406   -0.602067f, -0.434161f, -0.489824f, -0.313193f, 0.315568f,  0.603119f,
   1407   0.120245f,  0.182920f,  -1.117797f, -0.239594f, -0.296296f, -0.718093f,
   1408   0.489497f,  -0.527019f, 0.102453f,  0.426731f,  0.034606f,  0.311461f,
   1409   -0.012723f, -0.229877f, -0.284290f, 0.383227f,  0.065696f,  -0.222400f,
   1410   1.279248f,  -0.862190f, 0.629766f,  -0.250011f, -0.325060f, -0.360115f,
   1411   -0.159540f, -0.291856f, -0.038348f, 0.224639f,  0.600934f,  0.030205f,
   1412   1.337615f,  -0.286409f, -0.473710f, -0.418995f, -1.035249f, 0.004359f,
   1413   -0.481860f, 0.563625f,  -0.154709f, -0.101198f, -0.758796f, -0.507616f,
   1414   -0.095253f, -0.711135f, 0.207759f,  0.076313f,  -0.056087f, -0.162719f,
   1415   -0.232918f, -0.128402f, -0.444620f, -0.447344f, 1.126012f,  -1.504446f,
   1416 };
   1417 
   1418 static const float av1_4_partition_nn_bias_16_layer1[LABEL_SIZE] = {
   1419   -0.462133f,
   1420   0.465060f,
   1421   0.062211f,
   1422   0.401786f,
   1423 };
   1424 
   1425 static const NN_CONFIG av1_4_partition_nnconfig_16 = {
   1426   FEATURE_SIZE,  // num_inputs
   1427   LABEL_SIZE,    // num_outputs
   1428   1,             // num_hidden_layers
   1429   {
   1430       24,  // num_hidden_nodes
   1431   },
   1432   {
   1433       av1_4_partition_nn_weights_16_layer0,
   1434       av1_4_partition_nn_weights_16_layer1,
   1435   },
   1436   {
   1437       av1_4_partition_nn_bias_16_layer0,
   1438       av1_4_partition_nn_bias_16_layer1,
   1439   },
   1440 };
   1441 
   1442 static const float av1_4_partition_nn_weights_32_layer0[FEATURE_SIZE * 32] = {
   1443   -0.219494f, -0.428273f, 0.471006f,  0.448210f,  -0.152935f, 0.440435f,
   1444   0.922857f,  -0.074436f, 1.002195f,  0.414176f,  -0.327202f, -0.380066f,
   1445   -0.212346f, 0.061868f,  -0.056620f, 0.594134f,  0.617995f,  0.308358f,
   1446   0.232484f,  0.129849f,  1.483593f,  -0.071460f, 1.984515f,  1.116422f,
   1447   -1.141762f, -0.306220f, 0.089075f,  -0.271845f, 0.187524f,  0.050396f,
   1448   -0.061025f, 0.030809f,  0.172799f,  -0.458151f, -0.318357f, 0.122052f,
   1449   -0.414329f, 0.089366f,  0.118898f,  -0.376213f, -0.206151f, -0.519946f,
   1450   -0.463252f, -0.206694f, -0.254383f, -0.379487f, 0.093059f,  -0.245280f,
   1451   -0.205044f, -0.280060f, -0.171229f, -0.045389f, -0.179481f, -0.306245f,
   1452   -0.500856f, 0.003388f,  -0.527397f, -0.449330f, -0.174272f, 0.123769f,
   1453   0.023005f,  0.157273f,  0.073400f,  0.019099f,  -0.113848f, -0.098601f,
   1454   -0.290946f, -0.046770f, -0.314592f, -0.179914f, -0.391411f, -0.235631f,
   1455   -1.282604f, 0.048505f,  -0.746382f, 0.093740f,  -0.706583f, -0.085729f,
   1456   0.947382f,  -0.002961f, 1.175362f,  1.007309f,  0.141638f,  -0.037608f,
   1457   -0.118807f, -0.021474f, -0.146763f, 0.069363f,  -0.074372f, -0.215713f,
   1458   -0.004134f, -0.114110f, -0.330438f, -0.031136f, 0.111821f,  -0.534598f,
   1459   -0.357759f, -0.455950f, 0.139469f,  0.036582f,  -0.384743f, -0.168828f,
   1460   -0.239250f, 0.003520f,  -0.049003f, 0.075702f,  -0.025809f, -0.225972f,
   1461   -0.228905f, -0.412489f, 0.060570f,  -0.328819f, -0.206446f, -0.080231f,
   1462   -0.372008f, -0.218118f, -0.011954f, 0.024155f,  0.156014f,  0.020679f,
   1463   0.194398f,  -0.283491f, -0.024463f, -0.275099f, 0.028031f,  0.026340f,
   1464   -0.254668f, 0.103637f,  2.178693f,  0.552284f,  0.109366f,  -0.474806f,
   1465   -0.379286f, -0.026315f, 2.487924f,  -0.089466f, 0.206428f,  0.114578f,
   1466   0.152248f,  0.184050f,  -0.631948f, -0.014793f, -0.283782f, -0.830353f,
   1467   0.009343f,  -0.021029f, -0.060534f, -0.025164f, 1.841311f,  1.842748f,
   1468   -1.979708f, 0.450985f,  -1.606357f, -0.785454f, -0.212679f, -0.344342f,
   1469   0.198991f,  -0.258070f, 0.055974f,  0.224069f,  0.453051f,  0.408053f,
   1470   0.027873f,  -0.180538f, 0.056609f,  0.207654f,  0.104086f,  -0.194426f,
   1471   -0.359789f, -0.381143f, -0.331212f, -0.203973f, -0.324313f, -0.160825f,
   1472   -0.160439f, -0.044856f, -0.346647f, 0.044859f,  0.231398f,  -0.023643f,
   1473   -0.140316f, -0.260177f, 0.206965f,  -0.425386f, -0.420268f, -0.409748f,
   1474   0.006971f,  0.066186f,  -0.034950f, -0.345518f, 0.018633f,  -0.122489f,
   1475   -0.038506f, -0.330942f, 0.161236f,  -0.314119f, -0.050202f, -0.179597f,
   1476   0.731897f,  -0.184481f, 0.153598f,  -0.539501f, -0.301493f, -0.184967f,
   1477   -0.883754f, -0.586959f, -0.136292f, -1.772065f, -0.196276f, -0.053272f,
   1478   -0.101083f, -0.064142f, 0.161190f,  0.430826f,  0.355647f,  0.138266f,
   1479   0.051114f,  -0.028893f, -0.477673f, -0.238663f, -0.354117f, -0.056747f,
   1480   -0.334273f, -0.497688f, -0.486004f, -0.092033f, -0.241304f, -0.373250f,
   1481   0.120193f,  0.011360f,  -0.010475f, -0.092739f, -0.159650f, -0.033129f,
   1482   -0.259893f, -0.073217f, 0.200128f,  0.103407f,  -0.229233f, 0.128831f,
   1483   -0.063450f, -0.241732f, -0.408428f, -0.342239f, -0.264326f, -0.105403f,
   1484   -0.442879f, -0.310456f, -0.112881f, 0.263696f,  -0.205014f, -0.497936f,
   1485   -0.261734f, -0.382312f, -0.426807f, -0.021995f, -0.152794f, -0.301494f,
   1486   0.117232f,  -0.577809f, 0.154596f,  -0.409522f, -0.413113f, -0.359199f,
   1487   0.307294f,  -0.008746f, -0.310522f, 0.347620f,  -0.384845f, -0.451398f,
   1488   -0.226199f, 0.054154f,  -0.167608f, 0.046836f,  -0.013285f, -0.408119f,
   1489   -0.177973f, -0.248293f, -0.465830f, 0.035827f,  -0.222208f, -0.221717f,
   1490   0.066392f,  -0.349769f, -0.428029f, -0.516692f, 0.022398f,  -0.251682f,
   1491   0.134746f,  0.011167f,  -2.078787f, 0.173592f,  -1.948348f, 0.330060f,
   1492   1.993785f,  -0.052859f, -0.004795f, -3.703177f, 0.013450f,  -0.011687f,
   1493   0.073079f,  0.034803f,  0.025515f,  0.005994f,  0.101731f,  0.074303f,
   1494   -0.109962f, -0.270825f, -0.068273f, -0.163268f, -0.252826f, 0.137190f,
   1495   0.007667f,  -0.358453f, 0.027412f,  0.033492f,  0.021197f,  -0.049991f,
   1496   0.104468f,  -0.012157f, -0.056252f, -0.380756f, -0.338483f, 0.233235f,
   1497   -0.048631f, -0.441209f, -0.158482f, -0.148108f, -0.263453f, 0.138847f,
   1498   -0.304073f, -0.336312f, -0.017941f, -0.135563f, 0.075137f,  -0.246475f,
   1499   -0.229144f, -0.087744f, -0.346909f, 0.172611f,  0.004377f,  -0.009386f,
   1500   -0.023104f, 0.008000f,  -0.029390f, -0.317842f, 0.549674f,  -0.195337f,
   1501   -0.863979f, 0.160889f,  -0.269014f, -0.442104f, -1.799191f, 1.396533f,
   1502   -0.112837f, 0.881303f,  0.000764f,  -0.035415f, -0.141877f, 0.184831f,
   1503   -0.363566f, -0.178569f, 0.254134f,  -0.326893f, 0.127325f,  0.310620f,
   1504   -0.384621f, 0.146058f,  -0.287682f, -0.373447f, 0.026930f,  0.251650f,
   1505   0.053817f,  0.227509f,  0.121396f,  0.396514f,  -0.278381f, -0.038969f,
   1506   -1.538756f, -0.002856f, -0.892900f, 0.363426f,  -1.257922f, 0.743795f,
   1507   0.941177f,  0.219345f,  0.684189f,  1.396858f,  0.026299f,  -0.093433f,
   1508   -0.066182f, 0.057868f,  -0.089278f, -0.159680f, -0.262035f, -0.236656f,
   1509   0.005349f,  -0.031314f, 0.027917f,  -0.182113f, -0.212086f, -0.160774f,
   1510   0.051468f,  0.036787f,  0.183881f,  -0.288205f, -0.349691f, 0.162511f,
   1511   0.117878f,  -0.294534f, -