| /* |
| * Copyright (c) 2021, Alliance for Open Media. All rights reserved |
| * |
| * This source code is subject to the terms of the BSD 3-Clause Clear License |
| * and the Alliance for Open Media Patent License 1.0. If the BSD 3-Clause Clear |
| * License was not distributed with this source code in the LICENSE file, you |
| * can obtain it at aomedia.org/license/software-license/bsd-3-c-c/. If the |
| * Alliance for Open Media Patent License 1.0 was not distributed with this |
| * source code in the PATENTS file, you can obtain it at |
| * aomedia.org/license/patent-license/. |
| */ |
| |
| #ifndef AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_ |
| #define AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_ |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| #include "av1/encoder/ml.h" |
| |
| // TODO(chiyotsai@google.com): The performance of these models are getting worse |
| // due the changes in the encoder. We should retrain the models here to get |
| // better performance once we have the time. |
| |
| #define FEATURE_SIZE 10 |
| #define LABEL_SIZE 16 |
| // nn model for ab partition pruning, 128x128. |
| static const float av1_ab_partition_nn_weights_128_layer0[FEATURE_SIZE * 64] = { |
| -0.715251f, -0.015767f, -0.667353f, -0.345255f, 0.177887f, -0.469759f, |
| 0.426152f, 0.489798f, 0.469865f, 0.773821f, 0.088517f, 0.074585f, |
| 0.838754f, 0.048449f, -0.007584f, 0.638968f, 0.233305f, -0.319236f, |
| -0.257124f, -0.170869f, 0.137180f, 0.114852f, -0.721241f, -0.947962f, |
| -0.411298f, 0.494306f, -0.060435f, -0.648421f, -0.126624f, 0.072686f, |
| -0.143904f, -0.115839f, -0.175527f, -0.117728f, 0.040686f, -0.189925f, |
| 0.134361f, -0.258070f, -0.177558f, 0.158049f, 0.168668f, -0.062919f, |
| 0.341986f, 0.038100f, -0.435577f, -0.321255f, 0.203213f, 0.213061f, |
| 0.533304f, 0.359296f, -0.079558f, 0.004637f, 0.663904f, 0.043779f, |
| 0.383018f, 1.136559f, -0.084155f, 0.333057f, -0.199011f, 0.152059f, |
| -0.078419f, -0.167752f, -0.093651f, 0.083171f, -0.190143f, 0.086195f, |
| -0.280632f, -0.160663f, -0.017298f, 0.122628f, -0.138116f, 0.062927f, |
| 0.222462f, 0.626979f, 0.426928f, 0.117170f, -0.240457f, 0.053750f, |
| 0.038017f, 0.007359f, -0.017595f, 0.101407f, 0.332891f, 0.074933f, |
| 0.306498f, 0.219380f, -0.151638f, -0.247976f, 0.343405f, 0.121256f, |
| 0.049173f, 0.171474f, -0.139608f, -1.016599f, -0.345553f, -0.901138f, |
| 0.243401f, 0.059928f, -0.089396f, -0.195565f, 0.364705f, -0.020400f, |
| -1.383672f, 0.413018f, 0.536950f, -0.020904f, -1.335306f, -0.732290f, |
| 0.102885f, 0.315290f, -0.208521f, -0.081811f, 0.182300f, 0.125712f, |
| -0.593833f, -0.220639f, -0.314155f, 0.188327f, 0.118503f, 0.524427f, |
| -1.083859f, -1.130640f, 0.390352f, -0.045591f, 0.113160f, -0.009149f, |
| -0.096183f, 0.115829f, 0.377752f, 0.318396f, -0.591983f, 0.004797f, |
| -0.497377f, -0.342248f, 0.079546f, -0.025249f, -0.295972f, 0.615501f, |
| -0.464372f, 0.418315f, -0.173556f, 0.105217f, 0.298073f, 0.082478f, |
| 0.033223f, 0.977341f, -0.372982f, -0.052337f, 0.154124f, 0.396787f, |
| 0.536654f, -0.139061f, -0.223702f, 0.229666f, -0.846766f, 0.107723f, |
| 0.563839f, -0.483141f, 0.304813f, -0.765283f, 0.070964f, 0.151101f, |
| 0.275188f, 0.490303f, 1.175892f, 0.085377f, -0.191200f, 0.544532f, |
| -0.365075f, 0.167546f, 0.052183f, -0.220529f, -0.212227f, -0.144988f, |
| -0.273356f, -0.062023f, 0.103993f, -0.238493f, -0.161204f, -0.054611f, |
| -0.166672f, 0.128327f, 0.461751f, -0.545822f, 0.739798f, 0.594386f, |
| -0.163192f, -0.332501f, 0.363834f, -0.065043f, 0.474812f, -0.138811f, |
| 0.170924f, -0.778142f, -0.316474f, -0.508065f, -0.039986f, -0.478001f, |
| 0.340591f, 0.041783f, 0.055419f, 0.015155f, -0.981830f, -1.355237f, |
| 0.347516f, 1.155327f, 0.081319f, 0.274163f, -0.327230f, -0.113478f, |
| 0.556552f, -0.055986f, 0.217318f, -0.445351f, 0.325759f, 0.526547f, |
| -0.657434f, -0.572214f, -0.037087f, 0.081384f, 0.064518f, 0.014892f, |
| 0.215279f, 1.834504f, -0.242107f, 0.079810f, 0.129558f, 0.079588f, |
| -0.035189f, -0.221745f, -0.163414f, 0.043978f, -1.028662f, -0.623609f, |
| 1.130336f, 0.664661f, -0.063975f, -0.415863f, 0.018581f, 0.157758f, |
| 0.200570f, 0.063420f, 0.901039f, -0.746286f, 0.196230f, -0.290592f, |
| 0.042373f, -0.502500f, 0.183638f, 0.103394f, -0.298858f, 0.145436f, |
| 0.196916f, 0.108319f, -0.448572f, -0.881385f, 0.302497f, 0.121679f, |
| -0.021327f, 0.025150f, 0.481306f, -0.359634f, 0.350257f, -0.228647f, |
| -0.669860f, 0.260025f, -0.034182f, 0.619247f, -0.158826f, -0.405864f, |
| 0.674112f, -0.027885f, -0.325274f, -0.241492f, 0.036024f, -0.437685f, |
| -0.091458f, -0.109295f, -0.350676f, 0.044706f, 0.297059f, 0.016290f, |
| 1.121203f, 1.289062f, -1.299476f, -1.129221f, 0.103752f, 0.131302f, |
| -0.263265f, 0.222155f, -0.229908f, 0.013922f, -0.226001f, -0.248383f, |
| -0.004415f, -0.020958f, 0.055634f, 0.086200f, 0.114556f, -0.184061f, |
| -0.096210f, -0.146466f, -0.249618f, -0.195998f, 0.088758f, 0.023781f, |
| -0.264460f, 0.157026f, -0.235228f, -0.102564f, 0.043463f, -0.187823f, |
| -0.257500f, -0.199049f, -0.242210f, 0.030448f, 0.221604f, 0.151804f, |
| -0.100404f, -0.073931f, 0.144749f, -0.001572f, -1.438079f, -0.233716f, |
| 0.733422f, 1.727080f, -0.036397f, 0.027551f, 0.425321f, 0.085703f, |
| 0.031186f, 0.032333f, -0.675130f, 1.437733f, -0.202392f, -0.525003f, |
| 0.087048f, 0.328194f, -0.079989f, -0.391088f, -0.238732f, -0.120660f, |
| -0.139600f, 0.154665f, 0.026202f, -0.233501f, -0.009046f, -0.149187f, |
| -0.199646f, 0.115375f, 0.209762f, -0.014875f, 0.124038f, -0.119985f, |
| 1.079625f, -0.461513f, 0.614114f, 0.021003f, 0.439449f, -0.824834f, |
| -0.299701f, 0.193817f, -0.870551f, -1.262313f, -0.079517f, 0.341570f, |
| 0.305310f, -0.089721f, -0.317314f, -0.075631f, 0.127172f, -0.208635f, |
| 1.191922f, 0.163141f, 0.564285f, 0.286352f, 0.480865f, 0.173094f, |
| -0.094034f, -0.071339f, -0.328992f, -0.006382f, 0.314705f, 0.090258f, |
| -0.016099f, 0.193230f, 0.188061f, 0.398144f, 0.722781f, 0.769949f, |
| 0.025442f, -0.162016f, 0.070192f, -0.056946f, -0.100957f, -0.219934f, |
| -0.203492f, -0.015454f, -0.013272f, -0.098008f, 0.051707f, -0.017493f, |
| 0.527446f, 0.083605f, 0.588318f, 0.878215f, 0.028747f, -0.146479f, |
| -0.345170f, -0.136059f, -0.152005f, -0.203634f, 0.232702f, -0.101340f, |
| -0.027733f, -0.282611f, 0.265366f, 0.082362f, -0.265420f, -0.131124f, |
| 0.166303f, 0.040194f, -0.100710f, 0.579151f, -0.530136f, 0.163422f, |
| -0.998821f, -1.565311f, -1.774785f, -2.493372f, 0.116970f, -0.090302f, |
| 1.723272f, 0.552370f, -0.295954f, -0.439095f, -0.266730f, 0.027936f, |
| 0.539616f, -0.234902f, -0.167601f, -0.149877f, -0.242983f, 0.122353f, |
| -0.121620f, -0.205517f, -0.180144f, -0.264208f, 0.151500f, -0.159378f, |
| 0.029145f, -0.050892f, -0.223407f, -0.246239f, 0.043152f, -0.018460f, |
| 0.169972f, -0.187769f, -0.034670f, -0.238330f, 0.288070f, -0.093243f, |
| -0.437105f, -0.573376f, 0.660073f, 0.285727f, 0.408470f, 0.158475f, |
| 0.032699f, 0.056280f, -0.237176f, -0.083003f, 0.105598f, -0.169522f, |
| -0.260420f, -0.121100f, -0.173983f, -0.195693f, -0.232028f, 0.224940f, |
| 0.029124f, 0.009580f, -0.252034f, 0.103087f, 1.156561f, 0.603848f, |
| -0.562805f, -1.652742f, -0.568288f, -1.829395f, 0.046169f, 0.076095f, |
| 1.490819f, 0.415893f, -0.277788f, -0.115787f, 0.093750f, 0.270726f, |
| -0.395983f, -0.353742f, 0.034605f, 0.005342f, 0.184537f, 0.086445f, |
| 0.156417f, 1.476367f, 0.122587f, 0.002145f, 0.431057f, -0.381184f, |
| -1.646457f, -0.014009f, -0.671224f, 0.193726f, -0.019247f, -0.031267f, |
| -0.046208f, 0.298733f, 0.064734f, 0.616984f, 0.039381f, 0.182722f, |
| -0.116670f, 0.233093f, -1.214374f, -0.817970f, -0.064394f, -0.584783f, |
| 0.077697f, -0.266720f, 0.130875f, -0.235295f, -0.265754f, -0.159999f, |
| -0.250114f, -0.183017f, 0.194403f, -0.105808f, -0.169215f, -0.240866f, |
| -0.026662f, -0.045123f, -0.036175f, -0.167471f, -0.192908f, -0.232602f, |
| -0.267036f, -0.112500f, -0.257944f, -0.111909f, -0.802226f, -0.008800f, |
| 0.881460f, -0.678603f, 0.008666f, -0.252053f, -0.341035f, -0.175290f, |
| 0.183012f, 0.385991f, 0.079888f, -0.014039f, -0.148653f, 0.671778f, |
| -0.130219f, 1.086467f, 0.129267f, -0.040400f, -0.201221f, -0.077005f, |
| 0.015890f, 0.000781f, 0.137764f, 1.389546f, 0.172152f, 0.047279f, |
| -0.042783f, 0.127740f, 0.141467f, -0.335738f, -1.396392f, 0.031496f, |
| 0.357385f, 0.343602f, -0.714553f, 0.311014f, 0.132845f, 0.061149f, |
| 0.006796f, 0.568106f, -0.255949f, 0.104134f, -0.993447f, 0.298135f, |
| -0.406590f, -0.049228f, -0.578570f, -0.188561f, -0.107046f, 0.374095f, |
| 0.068481f, 0.036240f, -0.495801f, 0.180574f, -0.766129f, 0.886967f, |
| -0.568868f, -0.936062f, -0.418886f, -0.058735f, -0.511964f, -0.438596f, |
| 0.019016f, -0.015837f, 0.600197f, 0.429773f, 0.315026f, 0.319667f, |
| 0.214617f, -0.017316f, 0.270257f, -0.040524f, 0.695803f, -0.015223f, |
| -1.554965f, 0.356997f, -1.472428f, 0.024637f, -0.562958f, 0.870351f, |
| 0.193635f, 0.036063f, 0.328638f, 0.200274f, -1.634707f, 0.110534f, |
| 0.420104f, -0.072042f, -0.006404f, 0.171680f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_128_layer0[64] = { |
| 0.643147f, -1.348826f, 0.431627f, 0.000000f, 0.102717f, -0.772628f, |
| -0.034351f, -0.761977f, -0.638397f, 0.541969f, -0.391311f, 0.563076f, |
| 0.148553f, 0.267217f, -0.788092f, 0.544573f, -0.546280f, 0.000000f, |
| -0.446945f, 0.127732f, 0.270624f, -0.219435f, -1.220203f, 0.324584f, |
| 0.110885f, 0.276547f, 0.179726f, -0.375160f, 0.026401f, -0.032595f, |
| 0.000000f, -0.047932f, -0.648602f, -0.512637f, -0.031661f, -0.236761f, |
| 0.476453f, -0.028021f, -0.013673f, -0.015578f, -0.920077f, 0.000000f, |
| 0.915351f, -0.209962f, 0.000000f, -0.025731f, 0.218288f, 0.000000f, |
| 0.047726f, -0.813077f, -1.263281f, 0.239087f, 0.278614f, -0.030753f, |
| 0.000000f, 0.346744f, -0.948543f, -1.174211f, 0.216377f, 0.498913f, |
| 0.853918f, 0.002504f, -0.190403f, 0.452050f, |
| }; |
| |
| static const float av1_ab_partition_nn_weights_128_layer1[64 * LABEL_SIZE] = { |
| 0.179769f, 1.499417f, -0.445135f, -0.142278f, -0.337661f, 0.682064f, |
| -0.203213f, 0.302171f, 0.226877f, -0.422169f, 1.687586f, 0.783773f, |
| 0.220995f, 0.253482f, 0.370435f, -1.342775f, 0.337229f, -0.271473f, |
| 0.291796f, 1.362227f, -1.751397f, -0.086178f, 0.725496f, -0.118597f, |
| 0.227963f, -0.501577f, 0.223849f, -0.122421f, -0.123437f, -0.051045f, |
| -0.020115f, 0.212711f, 0.246025f, 0.088120f, -0.168995f, 1.740190f, |
| -0.195098f, 0.680339f, -0.589572f, -0.075244f, 0.878766f, 0.064092f, |
| -3.548527f, 0.001660f, 0.107926f, -0.169501f, -0.455212f, 0.123045f, |
| -1.836998f, 0.330365f, 1.301475f, 0.454761f, -0.576552f, -0.190761f, |
| 0.208459f, 0.618483f, 1.383364f, 0.970718f, 0.390174f, 0.406252f, |
| -0.564519f, -0.312062f, 1.345712f, -0.151873f, 0.109290f, 0.408847f, |
| 0.391243f, 0.152024f, 0.181764f, -0.036263f, -0.160466f, 0.153595f, |
| 0.049163f, -0.753012f, -1.804062f, 0.347475f, -2.746580f, 0.575618f, |
| 0.261799f, 0.210505f, -0.302054f, -0.109872f, 0.199506f, -1.182971f, |
| 0.723668f, 0.177758f, -0.338202f, 0.254396f, -0.220023f, 0.043504f, |
| 0.669866f, -0.040816f, -0.402730f, 0.017990f, 0.215523f, -0.216816f, |
| 0.454826f, -0.726067f, -0.018750f, -0.928679f, 0.154315f, -0.465641f, |
| 0.144566f, -0.030064f, -0.054667f, -0.154055f, 0.625384f, 1.323795f, |
| -0.159496f, 0.097072f, -0.463197f, -0.057938f, 0.750290f, -0.233061f, |
| 0.412631f, -0.535223f, -0.151423f, -0.154583f, 0.024721f, -0.494448f, |
| 0.230594f, -0.980138f, -0.653968f, 0.126079f, 0.051814f, -0.053219f, |
| -0.421708f, -0.228853f, 0.237885f, 0.888157f, 0.059655f, 0.241295f, |
| 0.210443f, 0.228238f, 0.119127f, -0.051989f, -0.355408f, 0.182215f, |
| 0.244277f, -0.104577f, -0.558035f, -0.023270f, 0.054571f, 0.700646f, |
| -0.223006f, 0.115523f, 0.023391f, 0.437264f, 0.709477f, -0.531212f, |
| -0.094731f, 0.328161f, -0.105418f, -0.133511f, 0.497168f, -0.030948f, |
| -0.407132f, -0.043943f, 0.155505f, 0.251945f, 0.205010f, 0.167160f, |
| 0.083654f, -0.636810f, 0.401315f, -0.398414f, 0.290046f, 0.206846f, |
| 0.042218f, 0.168150f, 0.843181f, -0.671242f, -0.202392f, -0.073301f, |
| 0.142895f, 0.237466f, 0.212145f, -0.091828f, 0.187038f, -0.720841f, |
| -0.616069f, -0.238021f, 0.065365f, 0.434119f, 0.179023f, -0.040107f, |
| -0.430734f, -0.297368f, 0.575954f, 0.382619f, -0.709787f, -0.320810f, |
| 0.242342f, -0.047614f, 0.705216f, 0.098077f, 0.357179f, 0.046017f, |
| 0.115074f, -0.412305f, -0.272304f, 0.048096f, -0.803811f, 0.275000f, |
| 0.642198f, 0.180286f, -0.087178f, -0.112707f, -0.394443f, 0.201989f, |
| 0.241759f, -1.038870f, 0.728124f, 0.800559f, -1.296268f, 0.198612f, |
| -0.053478f, 0.414344f, -0.510529f, 0.124179f, -2.219115f, -0.074583f, |
| -0.143055f, 0.001697f, 0.810811f, -0.657140f, 0.186818f, -0.936414f, |
| 0.539578f, -0.308244f, -0.126624f, -0.204767f, 0.091145f, -0.049340f, |
| 0.252014f, 0.394582f, 0.018764f, -0.060377f, -0.019133f, 0.064083f, |
| 0.069211f, -0.526693f, 0.209850f, -0.481466f, -0.468302f, -0.100407f, |
| 0.241018f, -1.037781f, 0.038539f, -2.113840f, -0.974895f, 0.163187f, |
| 0.425132f, -0.772546f, -1.261254f, -0.217488f, -0.971748f, -0.805640f, |
| -0.745175f, -0.177077f, 0.217658f, 0.381431f, -0.052338f, 0.087176f, |
| -0.165972f, 0.085937f, 0.472564f, -0.796627f, -2.453307f, 0.569664f, |
| -0.233010f, -0.192134f, 0.064339f, -0.111411f, -0.262469f, -0.410022f, |
| 0.519993f, -0.684620f, 0.393460f, -0.277753f, -0.153624f, 0.528984f, |
| -0.415558f, -0.445863f, 0.588512f, -0.142439f, -0.132127f, 0.199776f, |
| -0.579284f, 0.119488f, -0.033590f, -0.503846f, -0.674979f, 0.335125f, |
| 0.020519f, 0.233973f, -0.297998f, -0.051511f, 0.518626f, -0.412782f, |
| -0.074045f, 0.130523f, 0.465751f, -0.117795f, 2.535813f, 0.352108f, |
| -0.499228f, 0.379784f, 0.056699f, 0.173142f, -0.076519f, -0.026666f, |
| 0.017834f, 0.492333f, 0.093364f, 0.037867f, -0.165420f, -0.356429f, |
| -0.562334f, 0.057656f, -0.307544f, 0.085857f, -0.559851f, 0.107230f, |
| -0.398633f, 0.152618f, -0.216835f, -0.024539f, 0.026044f, -0.249519f, |
| -0.563594f, -0.746025f, 0.025265f, -0.298888f, -0.185243f, 0.058794f, |
| 0.233696f, -0.115223f, 0.144617f, -0.864390f, 0.619944f, -0.023980f, |
| 0.019481f, 0.225252f, 0.416552f, -0.115993f, 0.935387f, 0.744386f, |
| 0.053353f, -0.052582f, -0.065650f, 0.228488f, -0.032042f, -0.371252f, |
| -0.003638f, -0.736984f, -0.203776f, 0.030922f, -0.065577f, -0.031643f, |
| -0.049253f, -0.054640f, 0.787134f, 0.545414f, -0.140297f, -0.124274f, |
| -0.110011f, -0.029552f, 0.657005f, 0.214973f, -0.374300f, 0.251642f, |
| 0.276591f, 0.030566f, -0.145470f, 0.350579f, -0.356436f, -0.052694f, |
| -0.063966f, -0.751008f, -1.042392f, 0.328892f, -0.425058f, -0.421571f, |
| -0.571889f, -1.141472f, -0.125216f, 0.212713f, -0.485170f, -0.088791f, |
| 0.124589f, 0.023237f, 0.077635f, 0.020901f, -0.271402f, -0.321424f, |
| -0.513946f, -0.867872f, -0.284593f, 0.106276f, 0.220192f, -0.143532f, |
| -0.014648f, 0.073402f, 0.327256f, -0.139803f, 0.168763f, 0.048199f, |
| -0.122526f, 0.111713f, -0.134257f, 0.810364f, -0.085222f, -0.259221f, |
| -0.239349f, 0.044448f, 0.205031f, 0.413113f, -0.107720f, -0.018816f, |
| -0.247741f, -0.004963f, 0.041170f, -0.158019f, 0.134839f, 0.129502f, |
| 0.800488f, -1.041584f, -0.129336f, 0.170834f, 0.566586f, -0.230443f, |
| 0.437937f, -0.149922f, -0.046665f, -0.094646f, 0.200070f, 0.072943f, |
| -0.076943f, -0.084971f, -0.515843f, -0.146720f, 0.472869f, -0.444731f, |
| -0.100877f, 0.545196f, -1.786626f, -0.482946f, 0.500509f, -0.843257f, |
| 0.200374f, 0.045103f, -0.575718f, -0.164335f, -0.232522f, -0.021825f, |
| -0.139490f, 0.356058f, -0.352075f, 0.061751f, -0.200616f, -1.180921f, |
| -0.181355f, -0.137459f, 0.247574f, 0.181541f, 0.184314f, -0.961482f, |
| 0.493615f, 0.910261f, -2.279238f, 0.648631f, -0.055526f, -0.037137f, |
| 0.038643f, 0.136609f, -0.819373f, -0.040840f, -0.265989f, 0.006877f, |
| 0.454651f, -0.595323f, -0.099500f, -0.263717f, 0.150456f, 0.245077f, |
| -0.268666f, 0.162232f, -0.516451f, -0.024501f, 0.188046f, -0.002262f, |
| 0.261319f, 0.004173f, 0.746982f, 0.174761f, 0.470447f, -0.159558f, |
| -0.385240f, 0.023084f, -0.133520f, -0.220607f, -0.018731f, -0.373558f, |
| -0.707763f, -1.850150f, -0.807404f, -0.168063f, -0.071435f, -0.160740f, |
| -0.478789f, -1.070674f, -0.489740f, -0.255796f, 0.100486f, -0.153361f, |
| 0.334394f, -0.569472f, -0.198118f, 0.255922f, 0.104717f, -0.065179f, |
| 0.111879f, -0.447237f, 1.373623f, -0.190191f, -0.063311f, 0.337529f, |
| -0.138800f, 0.057009f, -0.137006f, 0.641378f, 0.883147f, -0.679655f, |
| 0.267717f, -0.351602f, -0.135225f, 0.229398f, -0.513225f, -1.120345f, |
| 0.528786f, -0.051081f, 0.086653f, 0.140141f, -0.563969f, 0.333402f, |
| -0.174745f, 0.321093f, -0.438641f, -0.005131f, 0.247415f, 0.110120f, |
| -0.076308f, -0.083244f, 0.838944f, -0.113043f, -0.013258f, -0.175028f, |
| -0.179941f, 0.272676f, -0.047946f, -0.088076f, -0.450031f, 0.053929f, |
| -0.083549f, -0.089952f, -0.186253f, 0.257483f, 0.011019f, 0.586435f, |
| 0.060580f, -0.052078f, 0.090277f, -0.780869f, 0.969811f, -0.025349f, |
| -0.281917f, 0.014857f, 0.231863f, -0.228601f, -0.003861f, 0.226550f, |
| 0.141825f, -0.102171f, -0.010387f, 0.220378f, -2.561975f, -0.497071f, |
| -0.315117f, 0.371981f, 0.138247f, 0.625031f, -0.308133f, -0.217876f, |
| 0.005615f, -0.860179f, 0.747491f, 0.006356f, -0.057024f, -0.483189f, |
| 0.055592f, -0.316834f, 0.069858f, 0.218788f, -0.200044f, 0.227588f, |
| 0.215496f, -0.055324f, -0.393147f, -0.394062f, -0.253264f, -0.075619f, |
| -0.152512f, -0.332995f, 0.129053f, 0.178668f, -0.302694f, 0.030678f, |
| 0.925896f, 0.964375f, 0.169021f, -0.218657f, -0.627204f, 0.206437f, |
| -0.521336f, 0.176206f, 0.142733f, 0.139248f, 0.411682f, 0.181544f, |
| 0.224850f, -0.935547f, -0.558208f, 0.348096f, 0.342129f, -0.389340f, |
| -0.236308f, -0.132099f, 0.073642f, 0.089391f, -0.306901f, -0.397842f, |
| 0.444282f, 0.074623f, -0.051075f, -0.106617f, -0.184037f, -0.239046f, |
| -0.138761f, 0.120794f, -0.647577f, -0.336471f, 0.527899f, -0.164234f, |
| -0.028354f, 1.083678f, -0.251534f, -0.145903f, -0.182783f, 0.070976f, |
| -0.199590f, -0.400306f, -0.029763f, -0.548042f, -0.266270f, -0.118084f, |
| -1.152632f, 0.383685f, -0.105895f, -0.096829f, 0.118382f, 0.047447f, |
| -0.019051f, 0.310180f, -0.162793f, -0.029574f, 0.058054f, -0.636017f, |
| 0.490639f, 0.158347f, -0.385701f, -0.147057f, 1.285825f, -1.276083f, |
| -0.021795f, -0.101600f, 0.163254f, 0.267160f, -2.317864f, -0.098598f, |
| -0.296337f, -0.309017f, 0.164127f, -0.270012f, -0.071187f, -0.262270f, |
| 0.075415f, -0.368328f, 0.186728f, -0.158031f, 0.481663f, 0.515950f, |
| -0.162551f, 0.497981f, 0.262196f, 0.168479f, 0.726066f, -0.243856f, |
| -0.058998f, 0.140168f, 0.053242f, -0.624623f, -0.249480f, 0.055197f, |
| -1.376804f, 0.417571f, 0.203784f, 0.174370f, -0.155531f, -0.029400f, |
| -0.491473f, 0.079811f, -0.080123f, 1.345900f, 0.637077f, 0.434862f, |
| -1.787438f, 0.005756f, -0.362706f, 0.179458f, -0.288263f, 0.516788f, |
| -0.921248f, 0.043794f, -0.137729f, -0.196171f, -0.046295f, -0.793781f, |
| -0.156532f, -0.132566f, 0.517989f, -0.154321f, -0.054174f, -0.077900f, |
| -0.373316f, -0.117718f, 0.188986f, -0.476188f, -0.245312f, 0.181439f, |
| -0.161024f, -0.229059f, -3.079907f, -0.225452f, -0.594355f, -0.558027f, |
| -0.135429f, 0.125766f, -0.081314f, -0.350894f, -0.163165f, -1.936507f, |
| -0.205966f, 0.031472f, 0.744446f, -0.006680f, -0.837551f, 0.605862f, |
| -0.854929f, -1.543750f, -0.307704f, -0.240517f, 0.178240f, -0.183586f, |
| -0.010307f, 0.099373f, -0.228278f, 0.175236f, -0.000133f, 0.104491f, |
| -1.540545f, -0.570971f, -0.252885f, 0.483036f, 0.052531f, 0.260214f, |
| -0.515016f, -0.602081f, -0.485690f, -0.730710f, 0.163719f, -1.775975f, |
| -0.298634f, 0.323626f, -0.373579f, -0.872977f, 0.619574f, 0.026862f, |
| -0.122531f, -0.084698f, -2.436297f, 0.483996f, -0.203640f, -0.302157f, |
| -0.150666f, -0.238320f, 0.089250f, 0.236485f, -0.668654f, -0.122863f, |
| 0.491152f, -0.226444f, -0.181248f, 0.120158f, 0.294027f, 0.250056f, |
| 0.307601f, 0.357875f, -1.746455f, -0.175670f, 0.385447f, -0.108808f, |
| -0.090235f, -0.642504f, -0.486004f, -0.055160f, -0.068692f, 0.009736f, |
| 0.607555f, -0.489426f, 0.150624f, 0.598114f, -0.128816f, -0.445793f, |
| -0.066524f, -0.254380f, 0.227106f, -0.406495f, -0.121632f, -0.275960f, |
| -0.136494f, 0.339457f, -1.318132f, -0.417572f, -2.614077f, 0.324603f, |
| -0.001211f, 0.375192f, -0.473448f, -0.162510f, 0.099329f, -0.277965f, |
| 0.101221f, -0.060263f, 0.121867f, -1.042140f, 0.440851f, 0.078898f, |
| -0.209007f, -0.243699f, 0.715197f, -0.093997f, 0.086022f, -0.178203f, |
| -2.275496f, -0.098413f, 0.199352f, -0.526791f, -0.162086f, -0.197806f, |
| -0.231657f, -0.269202f, -0.794294f, -0.223461f, 0.503584f, 0.416236f, |
| 0.064082f, 0.197655f, 0.340871f, -0.186645f, -0.291498f, 0.433938f, |
| -1.110063f, 0.003751f, 0.392738f, 0.069360f, 0.102088f, -0.302128f, |
| -1.518457f, 0.106939f, 0.404527f, -0.306868f, -0.286928f, 0.729276f, |
| -0.531710f, 0.745048f, -0.168837f, -1.953886f, -0.258828f, -0.190252f, |
| 0.241877f, -0.916744f, -0.030326f, -0.070541f, -0.271037f, 0.211303f, |
| -0.489957f, 0.100850f, 0.323999f, -0.802837f, -0.462408f, -0.079350f, |
| -0.029374f, 0.131213f, -0.825032f, 0.040202f, 0.351821f, 0.002869f, |
| -0.132516f, -0.471264f, -0.297002f, 0.263913f, 0.033478f, 0.146161f, |
| 0.533229f, -0.228608f, -0.200639f, -0.170955f, -0.915037f, 0.724491f, |
| 0.005151f, 0.018584f, -0.029771f, -0.396038f, -0.159236f, 0.038691f, |
| -1.197056f, 0.146302f, 0.226840f, -0.852126f, 0.031214f, 0.108880f, |
| 0.562000f, -0.134633f, -0.713343f, -0.342252f, -1.764521f, -0.114653f, |
| 0.515073f, -0.080515f, -0.121155f, -0.865139f, -0.833694f, -0.368553f, |
| 0.347673f, 0.623379f, 0.722067f, -0.492458f, -0.513263f, 0.585167f, |
| 0.721518f, -0.693499f, 0.343725f, -0.273861f, -0.040230f, -0.785664f, |
| -0.157500f, -0.308445f, 0.054062f, 0.600131f, -0.860887f, 0.434470f, |
| -0.191382f, -0.306150f, -0.243965f, 0.705444f, 0.007789f, -0.146154f, |
| -0.054499f, -0.073500f, -1.067364f, 0.404936f, -2.864590f, 0.182323f, |
| 0.326126f, 0.102405f, -0.135800f, 1.128095f, -0.012267f, -0.023996f, |
| -0.264834f, -0.108967f, -1.176746f, -0.926666f, 0.082999f, -0.498361f, |
| 0.083560f, -0.210074f, 0.019225f, -0.201614f, -0.904760f, 0.181421f, |
| 0.586384f, -0.177706f, 0.065471f, 0.168552f, 0.054705f, 0.045241f, |
| 0.048057f, -0.410957f, -2.188854f, -0.169812f, 0.015521f, 0.176856f, |
| -0.179331f, -0.352640f, -0.491735f, -1.743206f, 0.044227f, 0.010454f, |
| 0.823643f, -0.119781f, -0.098359f, 0.093119f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_128_layer1[LABEL_SIZE] = { |
| -0.433195f, -0.120488f, -0.116721f, 0.112134f, 0.118170f, -0.259769f, |
| -0.077530f, 0.394044f, 0.279167f, -0.317988f, 0.189538f, 0.314776f, |
| 0.325655f, -0.107123f, 0.591049f, 0.358744f, |
| }; |
| |
| static const NN_CONFIG av1_ab_partition_nnconfig_128 = { |
| FEATURE_SIZE, // num_inputs |
| LABEL_SIZE, // num_outputs |
| 1, // num_hidden_layers |
| { |
| 64, // num_hidden_nodes |
| }, |
| { |
| av1_ab_partition_nn_weights_128_layer0, |
| av1_ab_partition_nn_weights_128_layer1, |
| }, |
| { |
| av1_ab_partition_nn_bias_128_layer0, |
| av1_ab_partition_nn_bias_128_layer1, |
| }, |
| }; |
| |
| // nn model for ab partition pruning, 64x64. |
| static const float av1_ab_partition_nn_weights_64_layer0[FEATURE_SIZE * 64] = { |
| -0.495347f, -0.049498f, -0.026804f, 0.030474f, -0.289308f, -0.264193f, |
| -0.141121f, -0.072562f, -0.391665f, -0.051491f, -0.234761f, 0.027155f, |
| -0.038217f, 0.014872f, -0.289728f, -0.233577f, -0.415875f, -0.343615f, |
| -0.442543f, -0.482492f, 0.073510f, 0.007503f, 2.162329f, -0.362849f, |
| 2.145915f, -0.883135f, 0.185636f, -0.062859f, -0.465574f, -0.486205f, |
| -0.056710f, -0.330642f, -0.321860f, 0.042321f, -0.348965f, 0.003542f, |
| -0.291365f, -0.078164f, -0.345093f, -0.220272f, -0.471270f, -0.763853f, |
| 0.246622f, 0.199651f, -0.663420f, -0.154152f, -1.220383f, 0.047138f, |
| 0.816811f, 0.083247f, -0.218839f, 0.038143f, -0.063436f, 0.015517f, |
| -0.307320f, -0.166956f, -0.169499f, -0.399005f, -0.234638f, -0.162266f, |
| 0.050425f, -0.221723f, -0.256942f, -0.287285f, 0.144011f, -0.033245f, |
| 0.083649f, 0.119428f, -0.056706f, -0.117805f, 0.021866f, -0.257300f, |
| -0.201378f, -0.217484f, -0.413780f, -0.145793f, 0.082792f, -0.347247f, |
| 0.042539f, -0.302697f, 1.652316f, 0.000701f, -0.482843f, -0.160332f, |
| -0.450099f, 0.212399f, -4.715360f, -5.336774f, -5.375758f, -6.048339f, |
| 0.085956f, -0.037767f, 1.052409f, -0.931924f, -2.221907f, 0.268946f, |
| 0.015512f, 1.237094f, -1.092185f, 0.418247f, -0.082143f, -0.076914f, |
| -0.060749f, -0.325440f, -0.296960f, -0.066815f, -0.158477f, -0.373945f, |
| -0.122322f, -0.113495f, -0.097978f, -0.192816f, -0.270418f, 0.035840f, |
| -0.015458f, -0.121071f, -0.279582f, -0.067683f, 0.097855f, 0.019839f, |
| 0.451127f, 0.004376f, 1.410392f, 3.255835f, -0.344815f, 0.145202f, |
| 0.204132f, 0.171948f, -0.527736f, -0.110353f, 0.901448f, 0.003238f, |
| -3.822090f, 0.235462f, 1.024823f, -0.821244f, 0.876056f, 2.553762f, |
| -3.478597f, -2.076582f, -0.265515f, -0.055923f, -0.156980f, -0.164097f, |
| -0.246040f, 0.039430f, -0.071769f, -0.118847f, -0.304053f, -0.281541f, |
| -0.226021f, -0.263091f, -0.127359f, -0.249410f, -0.051023f, 0.083911f, |
| 0.084721f, 0.168089f, -0.272169f, -0.204998f, -0.008303f, -0.173998f, |
| 0.079376f, -0.197426f, -0.199052f, -0.118794f, -0.063753f, -0.094769f, |
| 0.066176f, -0.175832f, -0.238752f, -0.287960f, -0.134307f, -0.185953f, |
| -0.385845f, 0.119769f, -0.006567f, -0.382126f, -0.214221f, 0.038449f, |
| -0.253484f, -0.282766f, -0.020249f, -0.193929f, 0.016281f, -0.114423f, |
| -0.145940f, -0.281621f, -0.007588f, -0.131470f, -0.189012f, -0.185699f, |
| -0.279011f, -0.008132f, 0.208463f, 0.020569f, -0.206803f, -0.213408f, |
| -0.206131f, -0.290245f, 0.069701f, -0.000371f, -0.307572f, -0.451785f, |
| -0.300838f, -0.453186f, -0.301691f, 0.046327f, -0.312668f, 0.058272f, |
| -0.303131f, -0.376252f, 0.108384f, -0.086623f, -0.100630f, -0.027330f, |
| -0.003969f, 0.089502f, -0.200722f, -0.107889f, 0.061843f, -0.008478f, |
| -0.265057f, -0.271132f, -0.073562f, 0.129337f, -0.283698f, -0.353414f, |
| 0.076420f, -0.244280f, -0.119537f, -0.105366f, -0.184692f, -0.038817f, |
| -0.478507f, -0.118808f, -0.472979f, -0.305884f, -0.462813f, -0.189581f, |
| -0.011932f, -0.585700f, 0.253212f, -1.061900f, -0.205116f, -0.336407f, |
| -0.762199f, 0.577737f, 0.230832f, 0.434440f, -0.096713f, 0.038552f, |
| -0.147800f, -0.213553f, 0.041740f, -0.281907f, -0.026154f, -0.082356f, |
| -0.331871f, -0.408247f, -0.129022f, -0.037550f, -0.310233f, -0.320883f, |
| -0.391963f, -0.467392f, 0.027453f, -0.394761f, -0.045544f, 0.076052f, |
| 0.483985f, 0.067093f, 0.141361f, 0.576772f, 0.859718f, 2.566515f, |
| -0.025476f, 0.769738f, -0.680235f, -1.683309f, -2.394131f, -0.000714f, |
| -0.615021f, -0.195856f, -0.434035f, -0.295010f, -0.668659f, -0.245959f, |
| 0.551148f, 1.777227f, -0.461630f, 0.043093f, 0.012293f, -0.255841f, |
| -0.097070f, -0.371156f, -0.146323f, -0.015508f, -0.103873f, -0.087476f, |
| -0.297266f, -0.128699f, -0.149555f, 0.016534f, -0.375498f, -0.346759f, |
| -0.455156f, -0.147509f, -0.427076f, -0.354431f, -0.158025f, -0.164604f, |
| -0.237038f, -0.010314f, -0.092884f, -0.397084f, -0.217980f, -0.127184f, |
| -0.048421f, -0.144133f, 0.889073f, 0.012606f, 3.007608f, -0.602584f, |
| -1.849480f, -0.373159f, -1.890695f, -3.609938f, 0.811923f, -1.867208f, |
| -0.244326f, -0.018012f, -0.211192f, -0.220196f, 0.169363f, 0.119141f, |
| -0.230715f, 0.083247f, 0.020367f, -0.128629f, -0.217455f, -0.159640f, |
| 1.815952f, -0.369238f, -1.186447f, -0.658753f, -0.511026f, -0.096934f, |
| 0.662971f, 0.486475f, 0.159746f, -0.018932f, 3.692397f, 1.384353f, |
| -0.401984f, -0.248380f, -0.140861f, 0.215248f, -0.023711f, 0.059679f, |
| -0.072260f, 0.004271f, 0.039545f, -0.347971f, -0.081851f, -0.474896f, |
| -0.181572f, 0.066736f, -0.157822f, -0.163760f, -0.171113f, -0.089935f, |
| -0.338281f, -0.421444f, -0.306687f, -0.085283f, -0.377953f, -0.138750f, |
| -0.102701f, -0.312336f, 0.149831f, 0.007229f, -0.155700f, -0.173611f, |
| 4.074261f, 1.342306f, -1.272712f, 1.570899f, -0.545093f, -0.317605f, |
| -0.189440f, -0.133910f, -0.273190f, -0.108020f, -0.166107f, 0.021413f, |
| -0.239130f, -0.067211f, 0.041957f, -0.039234f, -1.003587f, -0.094412f, |
| 0.532512f, -0.870538f, -1.118023f, -1.160983f, -0.736307f, -0.418752f, |
| 0.419466f, 0.492122f, -0.004368f, -0.022096f, -1.115132f, 0.150886f, |
| 2.396852f, 2.660000f, -0.376537f, 0.468628f, 0.149413f, -0.074898f, |
| -0.067154f, 0.021245f, 0.127857f, 0.294189f, 0.508056f, 0.390232f, |
| -3.899177f, -3.414681f, -3.929195f, -4.160545f, -0.274323f, -0.052583f, |
| -0.003545f, -0.433084f, -0.404891f, -0.145051f, -0.312367f, 0.004579f, |
| -0.398724f, -0.372068f, -0.234279f, 0.017799f, -0.424760f, -0.646717f, |
| -0.047568f, 2.924664f, -0.644165f, 0.359349f, -0.294800f, 0.591746f, |
| -0.404710f, -0.092358f, -0.250729f, 0.030829f, -0.147149f, -0.476023f, |
| -0.071803f, -0.482516f, -0.293117f, -0.215923f, -0.373122f, -0.085315f, |
| -0.377052f, -0.449899f, -0.056452f, 0.138081f, -0.085350f, -0.308391f, |
| 0.106661f, 0.176234f, 0.258869f, -0.230172f, -0.233029f, -0.241208f, |
| -0.067509f, -0.223172f, -0.118353f, -0.302478f, -0.579632f, -0.561326f, |
| -0.158114f, -0.223167f, -0.026689f, 0.051863f, 0.212834f, -0.304714f, |
| -0.169071f, -0.193695f, -0.075682f, -0.170860f, -0.241008f, -0.044648f, |
| 0.280815f, -0.002585f, -0.283552f, -0.037701f, -0.681169f, -0.274535f, |
| -0.380595f, 0.109504f, -0.111141f, -0.437685f, -0.094459f, 0.144206f, |
| -0.106139f, -0.211832f, -0.054742f, -0.172813f, -0.295905f, -0.071907f, |
| -0.418429f, -0.183240f, 0.031319f, -0.095785f, -0.315447f, 0.069404f, |
| -0.422910f, -0.029867f, -0.357321f, -0.199976f, -0.337707f, -0.070188f, |
| -0.178198f, 0.177208f, 0.134688f, -0.081933f, -0.229452f, -0.208872f, |
| 0.026287f, -0.364040f, -0.063696f, -0.227443f, -0.234401f, -0.205699f, |
| -0.267238f, -0.494125f, -0.056255f, 0.053715f, -0.487754f, 0.014818f, |
| 0.087383f, -0.077556f, -0.168085f, -0.436851f, -0.276286f, -0.137845f, |
| -0.107606f, -0.103653f, -0.233766f, -0.419083f, 0.169185f, 0.010186f, |
| -0.001587f, 0.086735f, -2.465718f, 1.482185f, 1.621193f, -2.081680f, |
| 1.386553f, -3.204335f, -0.267111f, -0.004508f, 0.164712f, 0.274147f, |
| 1.724306f, -2.273659f, 0.749574f, -0.891905f, 0.105965f, -0.030428f, |
| -0.416018f, -0.300762f, 0.122911f, -0.316908f, -0.292504f, 0.138666f, |
| -0.161327f, -0.042143f, -0.249128f, 0.149210f, -0.088987f, -0.654101f, |
| -1.501843f, 0.216777f, 0.955914f, 0.524158f, -1.642561f, -1.643626f, |
| 0.864797f, -0.425451f, -2.115764f, -0.012502f, 0.065172f, 1.297270f, |
| 0.018845f, 1.167276f, -0.470970f, -0.244995f, 0.374782f, -1.811056f, |
| -0.055430f, -0.024102f, -0.376519f, -0.339640f, -0.119177f, -0.277995f, |
| -0.290095f, -0.081362f, -0.144139f, -0.118037f, -0.180357f, -0.217559f, |
| -0.370683f, 0.172816f, -0.265069f, 0.194321f, -0.273478f, 0.037442f, |
| -0.235552f, -0.078625f, -0.447541f, 0.016836f, -0.271123f, -0.171481f, |
| -0.321477f, -0.184826f, -0.442981f, -0.227273f, -0.370666f, -0.237232f, |
| -0.257493f, -0.225714f, -0.153716f, -0.283487f, -0.155399f, 0.067697f, |
| 0.230343f, -0.034318f, -0.022687f, -0.047090f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_64_layer0[64] = { |
| -0.212182f, -0.233725f, -0.758846f, -0.158162f, 0.614743f, -0.150944f, |
| -0.075727f, -0.208414f, 1.054996f, 0.713758f, -0.300051f, -0.151482f, |
| -2.443570f, 0.430590f, -0.129001f, -0.160733f, -0.230547f, -0.143228f, |
| -0.140577f, -0.086812f, -0.212298f, -0.159557f, -0.055647f, -0.211423f, |
| 0.578161f, -0.220318f, -0.210107f, -3.111584f, 0.604419f, -0.232622f, |
| -0.209924f, -0.130794f, -0.084097f, -0.036005f, 0.294594f, -2.535531f, |
| -0.209783f, -0.211189f, -2.766337f, 0.000000f, 0.450177f, -1.754884f, |
| 3.262664f, -0.209691f, -0.614886f, -0.211257f, -0.109096f, -0.190492f, |
| -0.109007f, -0.026910f, -0.136035f, -0.212321f, -0.139320f, -0.212233f, |
| -0.305430f, 0.739171f, 0.991277f, -0.088150f, 0.086313f, -0.023379f, |
| -0.125366f, -0.063576f, -0.212169f, -0.047463f, |
| }; |
| |
| static const float av1_ab_partition_nn_weights_64_layer1[64 * LABEL_SIZE] = { |
| -0.036800f, 0.528721f, 0.490767f, 0.144409f, 1.103640f, 0.361910f, |
| -0.180069f, 0.068033f, -14.868382f, 0.359013f, 0.322567f, -0.199212f, |
| 0.906164f, -0.488254f, 0.149653f, -0.216394f, -0.099347f, 0.004936f, |
| -0.111391f, 0.074848f, -0.041709f, 0.147627f, -0.018905f, 0.096116f, |
| 0.184817f, -0.016241f, 0.115739f, 2.376754f, 0.637097f, 0.052954f, |
| 0.136428f, 0.225267f, -0.181873f, -0.142876f, 0.684048f, 0.658791f, |
| 0.105795f, 0.241705f, 1.381114f, -0.209379f, 1.145949f, 0.795293f, |
| -9.361877f, 0.198302f, 0.539600f, 0.092317f, -0.081695f, 0.200777f, |
| 0.102334f, 0.081583f, 0.060948f, -0.025110f, 0.160951f, -0.020170f, |
| 0.234006f, -0.029369f, 0.375036f, 0.270209f, -0.556529f, 1.402949f, |
| 0.101777f, -0.027331f, 0.004502f, -0.153166f, -0.116651f, 0.151573f, |
| -0.022187f, 0.144044f, -0.108719f, -0.129942f, -0.270321f, 0.227363f, |
| 1.892330f, -0.661052f, -0.219398f, -0.229417f, -0.856438f, -1.196988f, |
| -0.081774f, 0.078847f, -0.207057f, -0.048947f, 0.152073f, -0.243056f, |
| -0.233329f, -0.288689f, -0.158333f, -0.141177f, -0.715436f, 0.016947f, |
| -0.093752f, 0.204984f, -1.209782f, 0.155683f, 0.092239f, 0.146495f, |
| 0.813146f, -0.027757f, 0.330982f, 2.173948f, -0.028867f, -0.141815f, |
| 0.292708f, -0.204794f, 0.014496f, 1.032799f, 1.312155f, 0.107020f, |
| 0.824752f, -0.013945f, 0.184829f, -0.041633f, 0.215300f, -0.476088f, |
| -0.053213f, 0.126862f, -0.020777f, 0.082893f, -0.223727f, -0.923063f, |
| 0.466529f, 0.082140f, -0.845758f, -1.140791f, -0.262033f, 0.138491f, |
| 0.151717f, -0.182479f, -0.131128f, 0.055411f, 0.106771f, 0.125552f, |
| 0.297184f, -0.257403f, -0.059884f, -0.274903f, 2.694357f, -0.108244f, |
| 0.025377f, 0.043092f, -0.558317f, 3.517159f, -0.270833f, -0.240676f, |
| 0.205100f, -0.057068f, -0.140445f, -0.193449f, -0.030061f, -0.286762f, |
| -0.467523f, -0.012647f, 0.190564f, 0.022394f, -0.101479f, 0.339684f, |
| -0.902743f, -0.169578f, -0.178029f, -0.041836f, -3.952108f, -0.028298f, |
| -0.221137f, -0.733895f, -0.223895f, 0.039012f, 0.687867f, 0.021423f, |
| 0.113063f, 0.676087f, -0.961000f, -0.064847f, 0.712856f, -0.192765f, |
| -0.001132f, 0.016689f, -0.236020f, -0.766186f, -0.175729f, 0.012879f, |
| -0.251064f, -0.105523f, -0.039212f, -0.347584f, 0.304352f, -0.034174f, |
| -0.364258f, -0.685252f, -0.266115f, -0.247345f, -0.155905f, 0.152283f, |
| -0.156315f, 0.174082f, -0.757654f, 0.102303f, -2.192316f, -0.245815f, |
| 0.119882f, -0.086542f, 1.987246f, -1.353163f, -0.374813f, -0.233504f, |
| -1.980895f, 0.692093f, -0.168351f, 0.172700f, -0.009052f, -0.015734f, |
| 0.106679f, -0.060472f, -0.256813f, -0.074874f, -0.207488f, -0.329515f, |
| -0.418268f, -0.017940f, -0.036081f, 0.064719f, -1.488016f, 0.020591f, |
| -0.176325f, -0.141074f, 0.944494f, 0.150237f, -0.249805f, -0.277280f, |
| 0.012686f, 0.132483f, 0.116123f, 0.013737f, -0.116091f, 0.750340f, |
| 3.251343f, -0.188864f, 1.096992f, 0.058467f, -0.041433f, -0.037937f, |
| -0.133294f, -0.137908f, -0.171132f, 0.106362f, 0.069383f, -0.052662f, |
| -0.177883f, -0.408049f, 0.680221f, -0.117035f, -0.904240f, -1.395228f, |
| 0.154527f, 0.134427f, 0.022767f, -0.158886f, -0.230316f, 0.161096f, |
| 0.362213f, -0.235060f, -0.941620f, 0.055912f, -0.049458f, -0.166632f, |
| 0.481418f, 0.930146f, 0.041108f, 0.033674f, 1.372066f, -1.847709f, |
| 0.003324f, 0.259534f, 0.177014f, -0.202761f, -0.262017f, -0.190852f, |
| -0.102839f, 0.028338f, 0.187193f, -0.041684f, 0.123973f, -0.198576f, |
| -0.110369f, -1.431400f, 0.208369f, -0.302370f, -0.248549f, 0.062985f, |
| 0.673409f, 0.036662f, -0.711340f, -0.120584f, -0.189789f, 0.098812f, |
| 2.947819f, 0.216567f, -0.414472f, -0.181742f, 1.873779f, -0.222726f, |
| -0.782870f, 0.007889f, 0.015062f, -0.554328f, 0.182928f, -0.191430f, |
| 0.123636f, -0.215460f, -0.225245f, 0.251516f, -0.013025f, -1.359595f, |
| -0.750602f, 0.342667f, -0.141899f, -0.687493f, -0.072639f, 0.048018f, |
| -0.242107f, -0.031917f, -0.287472f, -0.046088f, 0.832197f, -0.016576f, |
| -1.553349f, -0.216341f, 0.023077f, -0.410867f, 4.243743f, -0.514878f, |
| -0.066007f, -0.160696f, -0.262678f, -0.648790f, -0.430586f, 0.199940f, |
| -0.202496f, -0.222241f, -0.016406f, -0.121473f, 0.000828f, -0.081584f, |
| -0.152641f, -0.190166f, 0.644400f, 0.040196f, -0.302104f, -1.143654f, |
| -0.160327f, -0.320780f, -0.187006f, 0.037311f, 0.440618f, -0.070733f, |
| -0.117785f, 1.527539f, -0.419310f, 0.001300f, 1.389956f, -0.036366f, |
| -0.269203f, 0.612265f, 2.721897f, -0.086836f, -0.446999f, 0.012525f, |
| -0.078317f, -0.287052f, -0.111188f, -0.085181f, -0.164667f, -0.010466f, |
| -0.569722f, -0.018888f, -0.101663f, -1.147130f, -0.465204f, 0.114524f, |
| -2.192402f, -0.221325f, 0.375748f, 0.206284f, -0.261548f, -0.246257f, |
| -0.143004f, -0.069981f, -0.057306f, -0.116481f, -0.435903f, -0.314970f, |
| 0.013210f, -0.010175f, 4.630571f, -0.473226f, -0.197199f, -0.028204f, |
| 0.122907f, 2.475548f, 0.025011f, -0.092603f, -0.127561f, -0.151330f, |
| -0.077295f, 0.245016f, -0.045005f, 0.183396f, -0.330556f, -0.384887f, |
| 0.356374f, -0.016618f, -0.463353f, -1.291546f, -0.071986f, -0.311599f, |
| 0.072385f, -0.430786f, -2.094788f, 0.202733f, -0.910109f, -1.336543f, |
| -0.086800f, -0.096413f, 1.544383f, 0.031860f, -0.796211f, 0.762786f, |
| 3.250022f, -0.441798f, -0.698537f, 0.062839f, 0.033525f, -0.362996f, |
| 0.027022f, -1.131264f, -0.228926f, 0.053885f, -0.338628f, 0.155037f, |
| -0.046844f, -0.888172f, -0.241767f, 0.084965f, -0.617743f, -0.049896f, |
| -0.036894f, -0.304783f, -0.002639f, 0.137957f, 0.052121f, -0.131161f, |
| -0.117200f, -0.253380f, -0.205561f, -0.302450f, -0.047397f, -0.330518f, |
| 3.613420f, -1.525951f, -0.026738f, 0.209150f, -2.103534f, 2.019689f, |
| -0.366199f, -0.095260f, 0.027417f, -0.242512f, 0.162579f, 0.052113f, |
| -0.293851f, -0.068138f, -0.005799f, -0.344696f, -0.114824f, -0.431107f, |
| -0.120058f, -1.139926f, -1.048379f, 0.036446f, -0.323020f, -0.432945f, |
| 0.454151f, -0.140058f, 0.050649f, -0.094900f, -0.017278f, -0.238719f, |
| 1.193153f, 0.120447f, -0.496061f, 0.917431f, 2.936126f, -0.115521f, |
| -0.347397f, -0.435325f, -0.004383f, -0.211864f, 0.162383f, -1.040726f, |
| 0.089537f, -0.128579f, -0.133505f, 0.107129f, -0.435657f, -0.180388f, |
| 0.043650f, 0.018709f, -0.773242f, -0.687192f, -0.120633f, -0.063626f, |
| 0.029912f, 0.113972f, -0.403502f, -0.127640f, -0.269625f, 0.129794f, |
| -0.188539f, 0.041641f, 0.029769f, -0.198374f, 1.401407f, 0.353887f, |
| -0.219925f, 0.260515f, 1.157034f, -2.992044f, -0.097618f, -0.064417f, |
| -0.203626f, -0.008217f, -0.112339f, -0.227407f, -0.155118f, 0.247705f, |
| -0.012304f, -0.248447f, -0.913463f, -0.064788f, -0.214619f, -0.251761f, |
| -0.386861f, -0.040574f, -0.163219f, -0.100700f, 1.488274f, -0.071684f, |
| -0.033626f, -0.006497f, -0.246945f, -0.145221f, -3.747390f, 0.149609f, |
| -0.263326f, -0.297385f, -1.039896f, -0.083174f, -0.025473f, -0.235586f, |
| -0.001087f, 0.254286f, 0.265106f, 0.007325f, 0.199239f, 0.134103f, |
| -0.578211f, -0.259801f, -0.062373f, 2.368348f, 0.560556f, -0.252260f, |
| 0.889997f, -0.447872f, -0.059218f, -0.095315f, -0.061667f, 0.183580f, |
| -0.157479f, 0.055387f, -0.831734f, 0.007606f, -1.104906f, 0.301180f, |
| -0.117115f, 0.212959f, 4.727223f, -0.243833f, -0.397495f, -0.025021f, |
| -0.367587f, -2.082058f, -0.217699f, 0.148111f, 0.252430f, 0.111088f, |
| -0.260692f, 0.095124f, -0.407774f, -0.322169f, 0.002927f, 0.126169f, |
| -1.272325f, -0.279772f, -0.373680f, -0.485177f, -0.605458f, 0.021225f, |
| -0.092031f, -0.226585f, 1.895162f, 0.037866f, -0.275475f, 1.614360f, |
| -0.014972f, -0.277679f, -3.449082f, -0.092060f, -0.747873f, 0.020716f, |
| 2.776178f, -0.049963f, 0.183999f, -0.295259f, -0.028868f, 0.221895f, |
| 0.001265f, 0.336823f, 0.219372f, 0.112824f, 0.408132f, -0.017940f, |
| -0.311666f, 1.489606f, -0.058093f, -0.305659f, -0.491933f, -0.143847f, |
| 0.166115f, 0.042867f, -0.123447f, -0.087099f, -0.305395f, -0.365079f, |
| -0.755801f, -0.160649f, 0.736260f, -0.008611f, 0.095836f, -0.017345f, |
| 5.697515f, -0.498971f, -0.125280f, 0.199907f, 0.300053f, 0.605026f, |
| -0.228225f, -0.259523f, 0.016384f, 0.146973f, 0.210258f, 0.226766f, |
| -0.075178f, -0.050924f, 0.188496f, -0.415266f, -0.484880f, -0.236384f, |
| 0.071931f, -0.331863f, -0.601243f, -0.232479f, -0.285272f, 0.123789f, |
| -1.341333f, 0.037082f, -0.315202f, -1.587215f, -0.271576f, 0.003216f, |
| -4.437186f, -0.256205f, -0.576589f, -0.114147f, 2.153916f, -0.369618f, |
| 0.271415f, 0.145036f, -0.158731f, -0.240938f, -0.187369f, 0.036325f, |
| 0.254771f, 0.211488f, -0.240297f, 0.098417f, -0.415011f, 2.334793f, |
| -0.127252f, 0.020069f, -0.168755f, -0.448922f, -0.219207f, 0.016232f, |
| -0.221935f, -0.269500f, -0.100636f, 0.102545f, -0.809376f, -0.054979f, |
| 0.360713f, -0.326541f, 0.112933f, 0.138073f, 4.229404f, -0.763801f, |
| -0.305429f, 0.199955f, -1.787713f, 0.272866f, 0.109895f, 0.138466f, |
| -0.250259f, -0.167162f, -0.212588f, -0.217589f, -0.067125f, -0.077490f, |
| -0.208970f, -0.006863f, -0.671146f, -0.298320f, -0.165509f, 0.044597f, |
| -1.408624f, -0.213957f, -0.220947f, 0.129718f, 1.316777f, -0.098928f, |
| -0.008121f, -0.558293f, -0.297290f, -0.218873f, -4.346638f, -0.228174f, |
| -0.204710f, -0.388864f, 2.697919f, 0.025260f, 0.857020f, 0.009921f, |
| 0.036915f, -0.320275f, -0.087937f, 0.022636f, 0.236667f, 0.135496f, |
| -0.059616f, -0.192955f, 0.009470f, 2.139589f, -0.200449f, 0.129818f, |
| 1.017444f, -0.608299f, 0.257914f, -0.134306f, -0.033327f, 0.002855f, |
| -0.338598f, 0.015559f, 0.117362f, -0.166760f, 0.086903f, -0.167666f, |
| 0.193523f, 0.033852f, -1.147686f, 0.489468f, -0.006969f, 0.125630f, |
| 1.557907f, -1.604449f, -0.071114f, 0.096178f, 0.007065f, 0.200013f, |
| 0.213393f, 0.168466f, -0.100568f, -0.117861f, -0.161542f, -0.072561f, |
| -1.069871f, -0.470138f, -0.352578f, -1.503513f, -0.001394f, -0.380109f, |
| 0.065089f, -0.281668f, 0.988953f, -0.002778f, -0.659026f, -0.470692f, |
| -0.407292f, 0.011710f, -1.362085f, 0.184738f, -0.135786f, -1.374241f, |
| 4.487930f, -0.067274f, -0.956404f, -0.233995f, 0.224527f, -0.454556f, |
| 0.037900f, -0.281658f, 0.208224f, -0.254753f, 0.045740f, 0.051444f, |
| -0.388281f, 0.257112f, -0.485030f, -0.082659f, 0.148103f, -1.007456f, |
| -0.022295f, 0.036984f, -0.369401f, -0.076943f, -0.007636f, -0.293022f, |
| 0.470466f, 0.199012f, -2.158182f, 0.036577f, -0.014725f, -0.229516f, |
| 2.236929f, 0.030945f, -0.400045f, 0.109348f, 0.214691f, -0.891516f, |
| -0.251379f, -0.217358f, 0.013733f, 0.205573f, -0.151725f, -0.191782f, |
| -0.339630f, -0.163905f, -0.119191f, -0.032516f, 0.503015f, 0.025772f, |
| 0.029094f, -1.146153f, 0.216723f, -0.330023f, 0.064695f, -0.262521f, |
| 0.425612f, -0.093080f, -0.489648f, 1.051293f, -0.092332f, 0.095557f, |
| -0.874132f, 0.218483f, -0.127648f, -1.605802f, 2.763617f, -0.186734f, |
| -1.243166f, -0.193514f, -0.173748f, 0.337822f, 0.183873f, -0.251594f, |
| -0.211582f, 0.144081f, 0.029620f, -0.024853f, -0.385140f, 0.467341f, |
| -0.928316f, -0.195442f, 0.917783f, 0.357084f, 0.174445f, -0.073659f, |
| -0.012811f, -0.115420f, -0.181147f, -0.364449f, -0.567395f, -0.012969f, |
| -1.680714f, 0.065323f, 0.198063f, -0.244201f, 1.428545f, -0.432539f, |
| -0.208931f, -0.091205f, 0.957125f, 0.813519f, -0.262677f, 0.246852f, |
| 0.015536f, 0.055026f, 0.067054f, 0.262103f, -0.358115f, -0.095206f, |
| -0.267522f, -0.402710f, -0.680397f, -0.123627f, -0.385590f, -1.504680f, |
| -0.169513f, -0.215338f, 0.043633f, -0.079052f, -0.464410f, 0.122894f, |
| -0.278231f, -2.456445f, -0.159917f, -0.015597f, -0.735449f, -0.078854f, |
| -0.400290f, -1.153870f, 3.657228f, -0.287093f, -1.174355f, -0.102001f, |
| -0.288281f, 0.185209f, -0.145228f, -0.200449f, -0.099914f, -0.138354f, |
| 0.254428f, -0.161751f, -0.118206f, 0.296043f, -0.482613f, 0.080932f, |
| 1.097605f, -0.010190f, 0.232439f, 0.447617f, -0.133508f, 0.115763f, |
| -0.388589f, 0.174695f, -0.236014f, 0.006284f, -1.374129f, 0.092015f, |
| -0.241419f, -0.231667f, 2.763950f, -0.922932f, -0.061605f, 0.208740f, |
| -1.597190f, 1.353325f, -0.198528f, 0.250498f, -0.013950f, -0.203861f, |
| -0.254563f, 0.081931f, -0.413369f, 0.011844f, 0.080961f, -0.231161f, |
| -1.234909f, -0.440843f, -0.174980f, -0.315283f, -0.337474f, -0.123243f, |
| -0.310001f, -0.271028f, 0.364179f, 0.022845f, -0.535517f, -0.772936f, |
| -0.188435f, 0.039667f, -0.807463f, 0.266550f, -0.288857f, -1.630789f, |
| 1.280155f, 0.065712f, -0.279960f, -0.300056f, 0.258440f, -0.073781f, |
| 0.213878f, 0.042196f, 0.021360f, 0.211698f, -0.003751f, -0.192673f, |
| -0.137008f, 0.247878f, -0.470604f, 0.073164f, 1.523241f, 0.734755f, |
| -0.114126f, -0.193834f, -0.025759f, 0.263183f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_64_layer1[LABEL_SIZE] = { |
| -0.343508f, -0.706936f, -0.160676f, -0.877101f, -0.517567f, -0.253254f, |
| -0.148074f, 0.923430f, -0.364770f, 0.203550f, 0.401216f, 0.938246f, |
| -0.872737f, 0.718723f, 0.703398f, 2.560015f, |
| }; |
| |
| static const NN_CONFIG av1_ab_partition_nnconfig_64 = { |
| FEATURE_SIZE, // num_inputs |
| LABEL_SIZE, // num_outputs |
| 1, // num_hidden_layers |
| { |
| 64, // num_hidden_nodes |
| }, |
| { |
| av1_ab_partition_nn_weights_64_layer0, |
| av1_ab_partition_nn_weights_64_layer1, |
| }, |
| { |
| av1_ab_partition_nn_bias_64_layer0, |
| av1_ab_partition_nn_bias_64_layer1, |
| }, |
| }; |
| |
| // nn model for ab partition pruning, 32x32. |
| static const float av1_ab_partition_nn_weights_32_layer0[FEATURE_SIZE * 64] = { |
| -0.323723f, -0.214013f, -0.007772f, -0.458851f, -0.125542f, -0.123860f, |
| -0.410973f, -0.209389f, -0.087580f, -0.272881f, -0.168500f, -1.130845f, |
| 0.344916f, -0.475017f, -0.362262f, -0.195662f, -0.566124f, 0.782163f, |
| 0.411575f, -0.013378f, -0.318650f, -0.124678f, -0.612909f, -0.315788f, |
| -0.263990f, -0.508783f, -0.048938f, -0.416407f, -0.402648f, -0.156644f, |
| 0.225887f, -0.000493f, 2.682241f, 0.871204f, 0.059014f, 0.803542f, |
| -1.407028f, -1.154669f, 1.388148f, -0.293348f, -0.003669f, -0.009607f, |
| 1.330030f, -0.337841f, 2.118617f, 1.033059f, -0.084788f, 0.212904f, |
| 0.082405f, -0.070579f, -0.494005f, -0.173392f, 0.039546f, -0.463865f, |
| 0.077163f, -0.434066f, 0.030835f, -0.427139f, -0.560520f, -0.031606f, |
| -0.368541f, -0.027458f, 0.370574f, 0.461418f, 1.087682f, -0.572137f, |
| -1.509596f, -0.765697f, -0.499383f, -0.277998f, -0.106492f, -0.129564f, |
| -0.169133f, -0.269834f, -0.114270f, -0.275431f, 0.016339f, -0.156744f, |
| -0.267922f, 0.171216f, 0.110556f, 0.002954f, -0.200327f, -0.187663f, |
| 3.691601f, 1.234152f, 0.186315f, -0.125370f, -0.211235f, -0.554432f, |
| -0.131072f, -0.124982f, -0.130339f, -0.235350f, 0.018903f, 0.012896f, |
| -0.159372f, -0.269571f, -0.025709f, -0.221251f, 0.061919f, 0.016307f, |
| 0.384673f, -0.134525f, -1.599126f, -0.416459f, -0.743052f, 0.670249f, |
| -0.169709f, 0.421681f, -0.033360f, -0.072817f, 0.003647f, -0.110632f, |
| -0.158651f, -0.095136f, 0.223759f, 0.165767f, -0.269129f, -0.196075f, |
| -0.023183f, -0.293420f, 0.014875f, 0.018688f, -0.153407f, -0.172009f, |
| -0.259947f, -0.124015f, 0.173653f, -0.089103f, -0.021001f, -0.334230f, |
| 0.027177f, 0.103371f, -0.183860f, -0.204051f, -0.023721f, -0.192297f, |
| -0.143771f, -0.247106f, 0.218116f, -0.013240f, 2.831783f, 1.483928f, |
| -0.877025f, -0.313462f, -0.411320f, -0.447825f, 0.605977f, 0.234684f, |
| -0.119150f, -0.075182f, -0.330463f, 0.071503f, -0.254924f, -0.360071f, |
| -0.037022f, 0.063261f, -0.148759f, -0.238254f, -0.462018f, -0.027166f, |
| 0.065318f, -0.235743f, -0.257194f, -0.094784f, 0.022423f, 0.055925f, |
| 0.086672f, -0.021010f, 0.009965f, -0.001648f, -0.104917f, -0.387443f, |
| -0.102673f, -0.281706f, 0.145923f, -0.233391f, -0.378365f, -0.145584f, |
| -0.077751f, -0.121166f, 1.134565f, -0.097500f, -0.749202f, -0.544566f, |
| -1.361374f, -0.102494f, 1.089275f, 0.375299f, -0.105091f, 0.037641f, |
| -0.054248f, -0.282691f, -0.377797f, -0.066427f, -0.253815f, -0.329677f, |
| -0.339326f, -0.128217f, -0.282905f, 0.014937f, 1.067185f, -0.171764f, |
| 0.484458f, 0.396706f, -0.557055f, -0.891596f, -0.257839f, -0.720879f, |
| -0.218449f, -0.004755f, 1.572857f, 0.006229f, 1.962895f, -0.029746f, |
| -4.137691f, -2.185991f, -2.763477f, -0.520437f, -0.208708f, 0.006444f, |
| -1.263078f, -0.304560f, 1.072374f, 2.556429f, 0.312850f, 0.257488f, |
| -0.634264f, 0.156769f, -0.188943f, 0.040295f, -0.389915f, 0.085250f, |
| -0.248525f, 0.045667f, -0.776115f, -0.274680f, -0.448145f, -0.566161f, |
| -1.285316f, 0.079060f, 0.389124f, -0.510401f, -0.015299f, -0.664661f, |
| 0.099901f, -0.470694f, -0.051593f, -1.076381f, -0.442104f, -0.197867f, |
| -0.330011f, -0.448523f, -0.301018f, -0.442093f, -0.491953f, -0.582091f, |
| -0.064569f, -0.156516f, 0.543522f, -0.005924f, 0.161432f, 0.974793f, |
| 0.273712f, 1.104850f, -0.290312f, 0.313417f, -0.125370f, 0.136234f, |
| -0.191227f, -0.165054f, 0.011872f, -0.298871f, 0.095740f, 0.142760f, |
| -0.215771f, -0.031437f, 0.101041f, -0.085620f, 0.435387f, 0.002786f, |
| 1.971375f, 0.018392f, -1.771940f, -0.401433f, 0.808263f, -3.350013f, |
| 2.296952f, -1.024403f, -0.041645f, -0.034799f, -0.024078f, -0.347301f, |
| -0.276088f, -0.455907f, 0.266021f, 0.087348f, -0.146566f, 0.040492f, |
| -0.539866f, -0.206851f, -0.387874f, -0.125508f, -0.496676f, -0.373845f, |
| -0.472356f, -0.357082f, -0.081254f, -0.456466f, 0.554713f, 0.002185f, |
| -4.225019f, 0.344025f, 0.728796f, -0.262936f, 1.383924f, 1.577300f, |
| -2.653320f, -2.516156f, -0.301604f, -0.204105f, -0.138252f, -0.587536f, |
| -0.097889f, -0.352414f, -0.288276f, -0.184340f, -0.122741f, -0.243376f, |
| 0.031970f, -0.373402f, -0.396079f, 0.045566f, 0.072595f, -0.222681f, |
| -0.243802f, -0.340129f, -0.258494f, -0.192041f, -0.386112f, -0.240940f, |
| -0.047268f, -0.555802f, -0.032514f, -0.241341f, -0.167463f, -0.478308f, |
| -0.205936f, -0.316275f, 0.103729f, -0.197893f, -0.128029f, -0.218796f, |
| -0.167362f, -0.111814f, -0.126062f, -0.394260f, -0.025357f, -0.402697f, |
| -0.587395f, -0.400385f, -0.259664f, -0.415588f, -0.338503f, -0.399166f, |
| -0.270504f, 0.234505f, 0.272144f, 0.266938f, -0.392395f, -0.011717f, |
| -0.384221f, -0.473446f, -0.038420f, -0.241101f, -0.234402f, -0.275567f, |
| -0.410454f, -0.377599f, -0.179099f, -0.138432f, -0.248083f, -0.543026f, |
| -0.428043f, -0.239895f, -0.333193f, -0.103346f, -0.039038f, -0.171109f, |
| -0.119432f, -0.222351f, 0.000450f, 0.208724f, -0.510526f, -0.144656f, |
| -0.316721f, -0.344846f, -0.244794f, -0.129134f, -0.045634f, -0.400183f, |
| 0.043714f, -0.235414f, 0.115594f, -0.195616f, -0.106693f, -0.124242f, |
| 0.083990f, 0.049110f, -0.196130f, -0.059860f, -0.464235f, -0.516443f, |
| -0.101521f, -0.422379f, -0.413955f, -0.042991f, -0.345263f, -0.129264f, |
| -0.106911f, -0.140156f, -0.457841f, -0.199848f, -0.218954f, -0.329850f, |
| -0.364097f, -0.335262f, -0.312254f, -0.299331f, -0.052710f, -0.251019f, |
| -0.023459f, -0.222538f, 0.028849f, -0.088038f, -0.301550f, -0.273566f, |
| 0.067295f, -0.174608f, -0.445784f, -0.158366f, -0.567275f, -0.557652f, |
| -0.353503f, -0.302092f, -0.302049f, -0.551793f, -0.034535f, -0.225190f, |
| -0.210733f, -0.219377f, -0.057197f, -0.430933f, -0.025185f, -0.388150f, |
| -0.086147f, -0.430088f, 0.058466f, -0.152129f, -0.058411f, -0.236392f, |
| -0.547669f, -0.613849f, -0.893774f, -0.351715f, -0.399227f, -0.454909f, |
| -0.324501f, 0.000490f, -0.282167f, -0.073163f, -0.281452f, 0.047932f, |
| -0.175500f, 0.165220f, -0.276212f, 0.062153f, -0.217054f, -0.255487f, |
| -0.146416f, -0.097718f, -0.173809f, -0.559328f, -0.055695f, -0.391193f, |
| -0.132020f, -0.561184f, -0.308666f, -0.474053f, -0.219149f, -0.246558f, |
| -0.158325f, 0.151907f, -0.266835f, -0.144697f, -0.193960f, -0.046587f, |
| -0.220028f, -0.247355f, 0.135584f, 0.016511f, 0.367705f, -1.855877f, |
| 0.435622f, 0.444710f, -3.372301f, -3.030489f, 1.013267f, 0.380951f, |
| -0.170011f, -0.111415f, -0.456146f, -0.107254f, -0.095220f, -0.053078f, |
| -0.135864f, -0.591949f, -0.252810f, -0.324799f, -0.094796f, -0.260969f, |
| -0.391981f, -0.063170f, -0.336130f, -0.470127f, -0.405168f, -0.433219f, |
| -0.309563f, -0.295462f, -0.552270f, -0.012300f, -0.057793f, -0.034494f, |
| -0.446843f, -0.640160f, -1.188681f, -0.791361f, 0.543271f, 1.189112f, |
| 1.458468f, -0.005876f, -0.927475f, 0.062038f, -1.170818f, 0.338227f, |
| -3.007096f, -4.559296f, -4.045457f, -5.953635f, -0.228386f, -0.266890f, |
| -0.092595f, -0.377440f, -0.044534f, -0.053565f, -0.349268f, -0.415030f, |
| -0.310094f, 0.062721f, 0.251422f, -0.014350f, -1.282910f, 1.619560f, |
| 1.180566f, -0.032163f, -1.322951f, -0.603601f, 1.443710f, 0.654650f, |
| -0.393227f, 0.003536f, 0.029725f, -0.108925f, -0.053911f, 0.133977f, |
| -0.036145f, -0.168438f, 0.046989f, -0.331463f, -0.176983f, -0.311922f, |
| -0.272389f, -0.379592f, -0.399993f, -0.297873f, -0.193425f, -0.177524f, |
| -0.258309f, -0.567312f, -0.260217f, -0.241869f, 0.024010f, -0.032867f, |
| -0.039424f, -0.063670f, 0.193808f, -0.303514f, -0.013376f, -0.057761f, |
| 0.187922f, 0.006938f, 0.031810f, 0.180594f, -1.198427f, 2.820662f, |
| 0.154986f, -0.375518f, 0.116925f, -0.795782f, -0.085139f, -0.079365f, |
| -0.197936f, -0.321468f, -0.205271f, -0.558203f, -0.296235f, -0.151193f, |
| -0.158282f, -0.245402f, -0.208504f, -0.042335f, -0.087426f, -0.557129f, |
| -0.381427f, -0.441551f, -0.541011f, -0.060567f, -0.469305f, -0.032326f, |
| -2.453587f, -0.045568f, -0.296932f, 0.613061f, -0.320284f, 0.191620f, |
| -0.827145f, -0.225277f, 0.275800f, 1.696635f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_32_layer0[64] = { |
| -0.176206f, 0.660189f, -0.186156f, -2.481963f, -1.564218f, -0.280424f, |
| 0.732684f, -0.135581f, -2.193132f, -0.172771f, 0.605001f, -0.060392f, |
| -0.067190f, -0.132969f, -1.410812f, -0.298701f, -0.105963f, -0.086173f, |
| 0.632779f, 0.005585f, 1.310169f, 1.392136f, -0.563860f, -0.051053f, |
| 0.660998f, -0.214726f, -1.894342f, -0.128288f, -0.330721f, -0.053988f, |
| -0.177726f, 1.200859f, -0.178902f, -0.172620f, -0.184476f, -0.175559f, |
| 0.538503f, -0.322158f, -0.219080f, -0.058208f, -0.171347f, -0.216060f, |
| -0.174950f, -0.295740f, -0.184820f, -0.213896f, 1.317728f, -0.020116f, |
| -0.208096f, 0.000000f, 1.246166f, -0.225421f, -0.181555f, 0.861761f, |
| 1.172429f, -0.172892f, -0.737092f, -0.189904f, -0.179385f, -0.114618f, |
| -1.384604f, -0.201713f, -0.271948f, 0.372351f, |
| }; |
| |
| static const float av1_ab_partition_nn_weights_32_layer1[64 * 16] = { |
| -0.037828f, 1.529029f, 0.004927f, 1.475763f, 0.627172f, 0.325872f, |
| -0.990757f, 0.129476f, 0.889958f, -0.082031f, 0.332133f, 0.074422f, |
| -0.176212f, -0.074355f, 0.774378f, 0.110987f, -0.155469f, 0.253310f, |
| 0.882538f, 0.253605f, 0.332436f, -5.389474f, 0.278470f, 0.168644f, |
| 0.914611f, 0.154165f, 0.809262f, -0.174734f, 0.923673f, 0.064716f, |
| -0.070228f, -0.228735f, 0.002312f, 0.112222f, -0.045502f, -0.046004f, |
| 0.514101f, 0.306480f, 0.021232f, -0.015955f, -0.288260f, 0.189177f, |
| -0.104158f, 0.103273f, 0.096910f, -0.086328f, 1.327289f, -0.154247f, |
| 0.056676f, -0.243327f, -0.646676f, 0.177221f, -0.086761f, 0.729729f, |
| -14.710893f, -0.044881f, 0.339003f, -0.134737f, 0.073621f, -0.162913f, |
| 1.215237f, 0.140723f, 0.138630f, 1.241719f, 0.204092f, -0.463080f, |
| -0.176086f, 1.125868f, 1.034814f, 0.225455f, -0.203421f, -0.078787f, |
| -0.527498f, 0.012491f, -0.563307f, -0.170792f, 0.002679f, 0.116153f, |
| 0.211348f, -0.191900f, -0.212505f, 0.263445f, -0.074679f, -0.081441f, |
| -0.815405f, 2.448215f, 0.781299f, 0.149542f, -1.045162f, 0.043014f, |
| 0.217381f, -0.094500f, -0.090427f, 0.025784f, -0.228906f, -2.741798f, |
| 0.230475f, -0.256112f, -0.103297f, 0.159121f, -0.229793f, -0.014883f, |
| -0.104131f, -0.123816f, 0.164148f, -0.052279f, -0.071845f, -0.041197f, |
| 0.208527f, -0.234197f, -0.542336f, 0.020053f, 0.088870f, 0.014346f, |
| 2.502164f, -0.010244f, -0.267792f, 0.844394f, 2.711486f, -0.015262f, |
| -0.868053f, -0.295704f, 0.222289f, -0.000286f, -0.352098f, -0.079000f, |
| 0.021267f, -0.721739f, -0.240558f, -0.384775f, 0.065974f, -2.161058f, |
| 0.195889f, 0.268966f, -0.009329f, 0.014949f, 0.314943f, 0.235885f, |
| 0.072591f, -0.127120f, 0.150784f, 0.105697f, -1.297403f, -0.207509f, |
| -0.217688f, -0.076752f, 0.170952f, -0.294235f, 0.449973f, -1.712690f, |
| 0.860989f, 0.054757f, -0.812627f, -0.105316f, -0.736230f, -0.133192f, |
| -3.741608f, 0.495660f, -0.288936f, 4.654852f, -0.021305f, -0.308916f, |
| 0.049205f, -0.259996f, 0.114248f, -0.252647f, -0.253180f, -0.449314f, |
| 0.022979f, 0.063281f, -0.196154f, 0.078295f, -0.322317f, -0.145142f, |
| 0.300573f, 0.048385f, -0.254787f, 0.123939f, -1.263088f, -0.228565f, |
| -0.389061f, 0.391084f, 2.322438f, 0.075009f, 0.225743f, -0.198808f, |
| -0.280538f, -0.173939f, -0.120543f, -0.070792f, -0.417187f, -0.781056f, |
| -0.102756f, -1.760965f, 0.019149f, -0.867342f, 0.347141f, 0.031588f, |
| 0.302572f, -0.203573f, -0.357320f, -0.096078f, -0.527528f, 0.046699f, |
| -0.108561f, -0.167077f, -2.851509f, -0.307116f, 0.202720f, -0.160280f, |
| -0.215525f, 0.064355f, -0.427220f, 1.516230f, 0.634453f, 0.099400f, |
| -1.013887f, -0.029740f, -0.093426f, -0.044272f, -1.297636f, -0.237614f, |
| -0.160953f, 0.399036f, -0.030685f, -0.113619f, -0.184704f, 0.040519f, |
| -0.588252f, -0.210235f, -0.067623f, -0.031841f, -0.107261f, -0.192582f, |
| -0.253959f, -0.430821f, -0.103184f, -0.280185f, -0.357723f, 0.197761f, |
| -0.175087f, -0.055171f, 1.642014f, -0.192559f, -0.288147f, 0.610311f, |
| 4.688195f, -0.128728f, -0.914869f, -0.108286f, 0.013789f, 0.092125f, |
| 0.019770f, -0.178386f, 0.074164f, -1.152658f, -0.216738f, -0.277286f, |
| 0.012381f, 0.418259f, -0.680727f, -0.221917f, -0.485946f, 0.101672f, |
| 2.009457f, 0.054302f, 1.019838f, -0.116170f, 0.165134f, -0.112567f, |
| 0.852632f, -0.385796f, -0.108666f, 0.053181f, -0.311797f, -0.372875f, |
| -0.675717f, 2.409268f, -0.514720f, -0.214245f, -0.646596f, 0.009756f, |
| 0.203993f, 0.093617f, -0.301290f, 0.253551f, -0.128909f, -1.448442f, |
| -0.186823f, -0.278001f, -0.294993f, -0.176928f, -0.473605f, 0.062049f, |
| -0.212084f, -0.137326f, 0.012505f, 0.087850f, -0.200413f, -0.394119f, |
| -0.132224f, 0.146917f, 0.155746f, 0.198725f, -0.322541f, 0.196391f, |
| -0.945500f, 0.036736f, -0.155646f, -0.677341f, 1.130545f, -0.339554f, |
| 0.411628f, -0.355813f, -0.249843f, 0.213694f, -2.035607f, 0.055694f, |
| -0.111669f, 0.408696f, -0.067043f, -0.048182f, 0.398110f, -0.067542f, |
| 1.459801f, 0.236833f, -0.178806f, 0.168758f, 0.492387f, 0.099691f, |
| -0.776680f, -0.172865f, 0.204225f, 0.193982f, 0.575685f, -0.062248f, |
| 0.011486f, 0.058571f, -0.493391f, 0.026893f, -0.900467f, 3.793129f, |
| -0.634613f, -0.064660f, -0.048262f, 0.361905f, 0.033641f, 0.245171f, |
| -0.064671f, 0.034954f, 0.204358f, -0.904023f, -0.052714f, -0.250134f, |
| 0.136700f, 0.000734f, -0.371720f, 0.226483f, 0.217958f, 0.060559f, |
| 0.180111f, 0.000970f, 0.079556f, -0.096775f, 0.093855f, -0.026224f, |
| -0.243664f, 0.004290f, 0.123281f, -0.239476f, 1.230374f, -0.107826f, |
| -0.101982f, -0.153917f, 5.464427f, 0.304375f, -0.809957f, 0.090564f, |
| -0.278416f, -0.245555f, -2.078421f, 0.243093f, -0.127666f, 0.052451f, |
| -0.126662f, -0.783505f, 0.025149f, -1.422675f, -0.207769f, -0.362547f, |
| 0.115310f, 0.133390f, 1.264754f, -0.027055f, -0.485312f, -0.240717f, |
| -0.239722f, 0.146818f, -1.265043f, -0.235553f, 0.267104f, -0.021357f, |
| -0.435949f, -0.309371f, 0.049920f, 1.302721f, -0.233978f, -0.097551f, |
| -0.240631f, -0.287821f, -0.378380f, -0.273131f, -3.075169f, 0.226404f, |
| -0.029361f, 2.703590f, -0.430659f, 0.067927f, -0.387520f, -0.370630f, |
| -0.229236f, 0.085653f, -0.370956f, -0.065556f, -0.187859f, 0.068309f, |
| -0.109299f, -0.259898f, -0.103644f, -0.271199f, -0.209350f, 0.140993f, |
| -0.196713f, -0.135508f, -1.423209f, -0.406385f, -0.019956f, -0.864694f, |
| 5.963707f, -0.201157f, 0.726377f, -0.011076f, 0.010553f, -0.102918f, |
| -2.230088f, -0.258098f, -0.039547f, -0.029262f, -0.082324f, -0.860222f, |
| -0.094735f, -1.381839f, 0.587298f, -0.173048f, 0.721360f, 0.241900f, |
| 0.764302f, -0.023609f, -1.173755f, 0.103912f, -0.185363f, 0.078435f, |
| -2.245062f, -0.127269f, 0.202234f, 0.158975f, -0.260909f, 0.098608f, |
| -0.348247f, 1.732502f, -0.412298f, -0.269602f, -0.425771f, -0.146243f, |
| -0.530730f, 0.125716f, -1.004419f, 0.145109f, -0.059289f, 1.096304f, |
| 0.012891f, 0.045033f, -0.306875f, 0.003514f, -0.176110f, 0.037544f, |
| -0.441537f, -0.518921f, -0.262149f, -0.060407f, -0.379419f, -0.141245f, |
| -0.128894f, -0.176537f, -1.161318f, -0.249100f, -0.118330f, 0.042816f, |
| 1.173404f, 0.088312f, -0.393568f, -0.175134f, 6.529819f, -0.326652f, |
| -0.631917f, -0.393476f, 0.057781f, -0.217748f, -1.781139f, -0.012614f, |
| -0.212621f, -0.720322f, -0.218498f, -0.388556f, -0.254796f, -0.248399f, |
| -0.608744f, -0.265146f, 0.238517f, 0.066882f, -2.916806f, 0.054642f, |
| 0.282590f, 0.075248f, 0.010188f, -0.133486f, 0.985945f, -0.045849f, |
| -0.347564f, 0.057320f, -0.417920f, 0.063664f, 0.387062f, -2.692059f, |
| -0.535549f, 0.263736f, 0.327889f, -0.070273f, -0.775254f, 0.147250f, |
| 3.309425f, -0.212191f, -0.067204f, -2.912663f, -0.061496f, 0.084233f, |
| 0.022907f, 0.138421f, -0.112159f, -0.288447f, -0.010799f, 0.056049f, |
| -0.036527f, 0.021525f, 0.106649f, -0.291883f, 0.088424f, -0.057773f, |
| -0.086031f, 0.015277f, -0.318505f, -0.269049f, -1.008913f, -0.224785f, |
| -0.025820f, -0.649037f, 0.706381f, 0.096410f, 0.643776f, -0.046743f, |
| -0.009654f, -0.024246f, 1.469255f, -0.183536f, -0.370046f, -0.048442f, |
| -0.376527f, -0.431264f, -0.245109f, -0.093951f, 0.203683f, -0.099872f, |
| 0.087210f, 0.160692f, -3.527694f, -0.068891f, -0.228994f, -0.231817f, |
| -0.241949f, 0.193613f, 0.979597f, -0.091259f, 0.414424f, -0.047341f, |
| -0.209582f, -0.295134f, -0.016824f, 0.460327f, -0.072671f, 0.246234f, |
| 0.235896f, 0.127238f, -1.068683f, 0.035648f, 2.254888f, 0.180105f, |
| -0.260098f, -2.322120f, -0.184249f, -0.314801f, -0.099969f, -0.272117f, |
| -0.237916f, 0.031103f, -0.274063f, -0.049384f, -0.044917f, 0.102477f, |
| -0.342148f, -0.257558f, -0.346300f, 0.115333f, -0.115456f, 0.208354f, |
| -0.359301f, -0.167395f, 1.146514f, -0.177861f, -0.098658f, -0.444570f, |
| 6.759993f, -0.369772f, -0.831118f, 0.001866f, -0.073298f, -0.072095f, |
| 0.811902f, -0.431997f, -0.286587f, -0.269500f, 0.111492f, -0.525364f, |
| -0.351785f, -2.463474f, -1.852659f, 0.135325f, 0.138267f, 0.100643f, |
| -2.373278f, -0.285514f, -0.395388f, -0.185016f, -0.030249f, -0.005767f, |
| -0.716424f, -0.031674f, 0.011147f, 0.057405f, -0.215873f, -0.094401f, |
| 0.573528f, -1.223820f, 0.414852f, -0.059053f, -0.076488f, -0.287168f, |
| -0.842640f, 0.174084f, -0.567186f, 0.336629f, -0.062514f, 2.075448f, |
| -0.061680f, -0.131529f, -0.098994f, -0.204111f, -0.347865f, 0.108516f, |
| -0.049616f, -0.069212f, -0.273935f, -0.096545f, -0.210784f, -0.284698f, |
| 0.141501f, -0.176924f, -0.361341f, -0.251197f, -0.286694f, 0.245569f, |
| -1.521661f, -0.122639f, -0.015760f, -0.718912f, 5.877828f, 0.146916f, |
| 0.151767f, 0.220785f, -0.032298f, 0.230902f, 0.663943f, -0.252613f, |
| 0.057718f, -0.436038f, -0.323994f, -1.139787f, -0.042489f, -1.326298f, |
| -1.031206f, -0.104136f, 0.389897f, 0.127602f, -2.667789f, -0.212366f, |
| -0.506262f, -0.009115f, -0.213202f, 0.076167f, -1.629405f, 0.055129f, |
| 0.375393f, -0.150272f, -0.241515f, -0.326497f, 0.100069f, 0.410703f, |
| 0.340622f, 0.042437f, -0.349945f, 0.041176f, -1.178950f, 0.030992f, |
| 0.933908f, -0.035844f, -0.098660f, 1.030584f, -0.092043f, -0.355739f, |
| -0.305562f, 0.036161f, -0.049558f, -0.033225f, -0.403856f, -0.088276f, |
| 0.215493f, -0.149105f, -0.013363f, 0.025886f, -0.101306f, -0.205781f, |
| -1.072487f, -0.076019f, 0.077555f, 0.131003f, 1.267763f, -0.008954f, |
| -0.327617f, -0.246539f, 6.664081f, -0.404403f, -1.442489f, 0.191301f, |
| -0.336361f, 0.181156f, 0.833108f, 0.007879f, -0.194464f, -1.029408f, |
| -0.036268f, -0.927110f, -0.379190f, -0.293443f, -1.848579f, -0.242548f, |
| -0.065990f, 0.203160f, -0.291788f, 0.000680f, 0.587011f, -0.241289f, |
| 0.037034f, 0.000552f, 1.072308f, -0.387230f, -0.230050f, 0.292322f, |
| -0.720001f, 0.034109f, -0.467260f, 2.211644f, -1.839191f, -0.048797f, |
| -0.083469f, -0.334686f, -0.269056f, 0.051295f, 1.319904f, -0.035603f, |
| -0.018457f, -0.824915f, -0.212285f, -0.230516f, -0.035093f, -0.400843f, |
| -0.305469f, -0.099011f, 0.014225f, -0.452772f, 0.170331f, -0.389312f, |
| -0.115084f, -0.014770f, -0.429387f, -0.155961f, -0.568200f, -0.037853f, |
| -0.125137f, 0.067228f, -1.329271f, -0.117874f, -0.132499f, -0.218376f, |
| -0.588325f, -0.320024f, 0.085695f, -0.235047f, -0.217790f, 0.103015f, |
| -0.698644f, 0.017766f, -0.058299f, 0.199411f, -0.122485f, -0.563949f, |
| -0.349011f, -0.557045f, -0.131165f, 0.002281f, 0.118559f, -0.210302f, |
| -1.153815f, 0.116738f, -0.236007f, -0.003487f, -0.006885f, -0.244816f, |
| 0.953222f, 0.093748f, 0.266869f, 0.241869f, -0.860832f, -0.387012f, |
| -0.338986f, 2.097515f, -1.942512f, -0.298021f, 0.543911f, -0.043214f, |
| 0.082125f, -0.120242f, 0.712231f, 0.213327f, -0.301687f, -0.544011f, |
| -0.392131f, 0.004302f, 0.004825f, -0.317440f, -0.107518f, -0.293407f, |
| -0.159111f, -0.080367f, 0.132663f, -0.017726f, -0.237521f, -0.190297f, |
| -0.361633f, 0.200518f, -0.538296f, -0.027975f, -0.381704f, -0.016963f, |
| 0.630105f, -0.190997f, -0.287840f, -0.603488f, 3.605598f, -0.276614f, |
| -1.346383f, 0.186912f, -0.047575f, -0.189232f, -1.519072f, 0.097816f, |
| -0.223722f, 0.304924f, -0.213022f, -1.052433f, -0.322283f, -1.706734f, |
| -2.458027f, 0.237976f, 0.171050f, -0.103139f, -0.278689f, 0.329824f, |
| -0.262448f, -0.122916f, -0.236398f, -0.013848f, -0.969160f, -0.374907f, |
| 0.091018f, -0.386471f, -0.723940f, 0.064956f, -0.057652f, 1.321024f, |
| -1.397418f, -0.143136f, 0.272468f, -0.030749f, 0.037324f, 0.069316f, |
| -0.904925f, -0.333693f, -0.117709f, 2.279598f, -0.428065f, -0.131157f, |
| -0.014288f, -0.402862f, -0.666090f, 0.017070f, -0.028333f, 0.002481f, |
| 0.197156f, -0.038120f, -0.271062f, -0.188275f, -0.021370f, -0.070849f, |
| -0.905007f, -0.095886f, -0.093055f, -0.121821f, -1.239812f, -0.411799f, |
| -0.089948f, -0.936827f, 1.437569f, -0.388908f, 0.126170f, 0.186162f, |
| -0.018819f, -0.138364f, -1.066412f, -0.138222f, -0.022186f, 0.107331f, |
| -0.230436f, -1.352605f, -0.161323f, -1.081810f, -0.933825f, -0.136675f, |
| 0.378157f, 0.113377f, -0.850610f, 0.080245f, -0.087305f, -0.002852f, |
| 0.044408f, -0.188172f, -1.891998f, 0.092189f, 0.125325f, -0.105090f, |
| -0.848510f, -0.396308f, -0.384130f, 2.007509f, -1.480787f, -0.126946f, |
| 0.314767f, 0.000195f, -0.285628f, -0.110442f, -0.293948f, 0.258559f, |
| -0.417603f, 1.570705f, 0.092459f, -0.340974f, -0.284754f, -0.007801f, |
| -0.324610f, -0.004734f, -0.207716f, -0.057175f, 0.055467f, -0.210830f, |
| -0.113005f, -0.299177f, 0.068074f, 0.017929f, -2.897598f, -0.260074f, |
| -0.014422f, -0.206467f, 1.246997f, -0.372863f, -0.214160f, -0.114035f, |
| 5.805862f, 0.003611f, -1.340990f, -0.021085f, -0.260431f, -0.002720f, |
| -1.251640f, -0.353531f, -0.304009f, -0.153376f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_32_layer1[LABEL_SIZE] = { |
| -0.521497f, -1.061572f, -0.078756f, -0.660662f, -0.403741f, -0.960163f, |
| 0.001427f, 0.523607f, 0.225068f, -0.055273f, 1.019519f, 1.181880f, |
| -0.010198f, 0.130597f, 1.276752f, 2.028188f, |
| }; |
| |
| static const NN_CONFIG av1_ab_partition_nnconfig_32 = { |
| FEATURE_SIZE, // num_inputs |
| LABEL_SIZE, // num_outputs |
| 1, // num_hidden_layers |
| { |
| 64, // num_hidden_nodes |
| }, |
| { |
| av1_ab_partition_nn_weights_32_layer0, |
| av1_ab_partition_nn_weights_32_layer1, |
| }, |
| { |
| av1_ab_partition_nn_bias_32_layer0, |
| av1_ab_partition_nn_bias_32_layer1, |
| }, |
| }; |
| |
| // nn model for ab partition pruning, 16x16. |
| static const float av1_ab_partition_nn_weights_16_layer0[FEATURE_SIZE * 64] = { |
| 0.151902f, 0.007947f, -1.788454f, 0.431869f, -2.971387f, 0.923566f, |
| 1.632542f, -1.665136f, -0.338632f, -5.075884f, 0.398267f, 0.030467f, |
| 2.263534f, -0.045532f, -1.066128f, 0.915139f, -0.560500f, -3.293125f, |
| 2.072793f, -1.011414f, 0.122716f, -0.060169f, -0.388860f, 0.031019f, |
| -0.381861f, 0.001551f, -0.328472f, 0.038296f, -0.060398f, -0.375556f, |
| 0.209226f, 0.014764f, -1.443469f, -0.345486f, 2.409269f, 1.524846f, |
| -0.640666f, 1.322139f, -2.074771f, -0.580944f, -0.203960f, -0.072893f, |
| 0.329701f, 0.115339f, -1.339542f, 0.249024f, -0.421545f, -0.409151f, |
| -0.258293f, 0.836288f, -0.073685f, -0.009624f, 0.895712f, 0.320639f, |
| 0.451002f, -1.544558f, 0.193709f, -1.389012f, 1.305451f, 0.089795f, |
| 0.050338f, -0.017433f, -0.304667f, 0.500729f, 0.504346f, 0.073757f, |
| 0.582649f, -0.993623f, 1.766766f, -3.067265f, -0.415774f, -0.006036f, |
| -1.245281f, 0.253205f, -0.591245f, -0.626238f, 0.551852f, 0.593755f, |
| 0.491023f, 1.099384f, -0.348448f, 0.054564f, -0.451422f, -0.375781f, |
| -0.248390f, -0.052548f, -0.380069f, -0.165391f, -0.297968f, -0.052142f, |
| -0.316381f, -0.045246f, -0.243905f, -0.034169f, -0.247523f, -0.180773f, |
| 0.068066f, -0.374920f, 0.057536f, -0.189748f, 0.058375f, -0.267749f, |
| -0.147286f, -0.246153f, 0.006183f, -0.202029f, -0.059128f, 0.116852f, |
| 0.134719f, -0.126900f, -0.064646f, -0.196458f, -0.182331f, 0.108029f, |
| -0.264499f, 0.155816f, -0.107255f, -0.056983f, -0.209771f, -0.099070f, |
| 0.007313f, -0.254124f, -0.231964f, -0.275972f, 0.032098f, -0.264564f, |
| -0.208743f, 0.155599f, -0.121511f, -0.156145f, -0.162315f, -0.059788f, |
| -0.257073f, -0.076654f, -0.110616f, -0.321675f, -0.051952f, 0.006301f, |
| -0.154114f, 0.017032f, -0.017364f, -0.233247f, 0.009918f, -0.179289f, |
| -0.190722f, 0.147106f, -0.063910f, -0.396872f, -0.263123f, -0.003850f, |
| -0.040718f, -0.324699f, 0.118660f, -0.170727f, -0.316788f, 0.100886f, |
| -0.202842f, 0.045371f, 0.150561f, -0.057054f, -0.308150f, 0.028346f, |
| -0.381473f, -0.195365f, 0.026221f, -0.281795f, 0.087204f, 0.047689f, |
| -0.027643f, -0.104724f, -0.089030f, -0.117661f, -0.349160f, 0.056982f, |
| -0.340273f, 0.048086f, 0.046103f, -0.121527f, 0.021697f, 0.054109f, |
| -0.002768f, -0.008461f, -2.297240f, 0.124651f, 3.621661f, -0.057120f, |
| -1.151656f, 2.296894f, -3.678720f, -0.290240f, 0.087683f, -0.186389f, |
| 0.007656f, -0.090236f, -0.245217f, 0.110389f, -0.251719f, -0.029084f, |
| -0.128203f, -0.100005f, -0.032779f, 0.007281f, -0.366596f, -0.267870f, |
| -0.215620f, 0.047687f, 0.010303f, 0.097980f, -0.191569f, -0.341162f, |
| 0.119249f, 0.026279f, -2.161546f, 0.459591f, 1.290566f, 1.791797f, |
| -0.409835f, 0.127081f, -1.156367f, 0.198286f, 0.099561f, -0.067445f, |
| -0.034352f, 0.017966f, -0.277380f, -0.057220f, -0.174198f, -0.014164f, |
| 0.146090f, -0.357530f, 0.097644f, -0.000932f, 0.446603f, -0.066793f, |
| 2.448620f, 0.937617f, -1.232922f, 0.313183f, 0.816827f, -0.275115f, |
| -0.245205f, -0.126895f, 0.156668f, -0.186977f, -0.273505f, 0.013315f, |
| 0.168629f, -0.089084f, 0.006166f, -0.116107f, -0.199316f, -0.024010f, |
| -0.242303f, 0.011612f, -0.218485f, -0.229661f, -0.123922f, 0.136699f, |
| 0.006732f, -0.148718f, -0.164225f, 0.116063f, 1.587898f, 0.690519f, |
| 0.360566f, 0.009739f, -0.678702f, -0.046003f, 0.126984f, 0.605212f, |
| 1.240663f, -0.000228f, -1.119369f, -0.415589f, -0.721003f, 0.097936f, |
| -1.410586f, -2.358833f, -2.773129f, -3.983361f, -0.087144f, -0.050029f, |
| -0.242255f, 0.137424f, -0.307490f, -0.084637f, -0.023812f, -0.196582f, |
| -0.078695f, 0.038257f, -0.012110f, -0.263521f, 0.009839f, -0.109125f, |
| -0.226036f, 0.060712f, 0.093671f, 0.153143f, 0.039116f, -0.290891f, |
| 0.227057f, -0.204633f, -0.207539f, -0.148242f, 0.046204f, -0.231268f, |
| -0.209315f, -0.307579f, -0.436556f, 0.023475f, 0.131793f, -0.038301f, |
| 1.650584f, 0.392570f, 1.446576f, 1.254380f, -0.516867f, -0.057116f, |
| 0.149320f, 0.414424f, -0.246309f, 0.003877f, -0.480238f, -1.037035f, |
| -0.830779f, -1.122244f, -0.408267f, -0.253956f, 0.382005f, 0.940609f, |
| -1.113370f, -0.018554f, 0.141064f, -0.182504f, 1.270707f, 0.414904f, |
| -0.216036f, 0.203831f, 0.450716f, -0.452909f, 0.139358f, -0.027143f, |
| 1.956892f, 1.643732f, -0.867839f, -0.620520f, -0.334607f, -0.519982f, |
| 0.205023f, 0.661159f, -0.000809f, 0.049033f, -0.348579f, -0.200338f, |
| -0.362144f, -0.346590f, -0.230096f, 0.180746f, -0.149954f, -0.253429f, |
| -0.378170f, -0.040724f, -0.041597f, 0.243659f, -0.472181f, 0.015401f, |
| -0.180376f, 0.153139f, -0.247738f, -0.010485f, -0.157158f, 0.016825f, |
| -0.238925f, -0.265798f, -0.318374f, 0.142352f, -0.210520f, 0.051928f, |
| -0.352190f, -0.179052f, -0.185498f, 0.025540f, -0.111667f, -0.235187f, |
| -0.215454f, 0.010931f, -0.238372f, -0.126659f, 0.075691f, -0.091167f, |
| -2.462379f, -0.007950f, -0.637990f, 0.285554f, -0.051275f, 0.282279f, |
| -0.744083f, -0.570646f, 0.592198f, 1.421332f, -0.256027f, -0.140315f, |
| 0.160247f, -0.063185f, -0.055895f, -0.199864f, -0.287353f, -0.074561f, |
| -0.071228f, 0.055864f, -1.084764f, -0.263409f, 0.779266f, 0.228187f, |
| 0.375013f, 0.121204f, -0.656948f, 0.533561f, 0.272671f, -0.015423f, |
| -0.124180f, -0.009127f, 2.934838f, -0.150998f, 1.163152f, 0.081997f, |
| -4.715939f, -3.676595f, -1.524886f, -0.167593f, 0.281186f, 0.024046f, |
| -1.451709f, 0.332558f, 0.990504f, 0.376290f, -1.466773f, -0.448439f, |
| -2.929108f, -4.255188f, 0.065238f, 0.019950f, 1.372393f, 0.444052f, |
| -2.538772f, 1.579767f, -0.464911f, -1.866114f, 1.053958f, 0.434467f, |
| -0.125964f, 0.034671f, 0.077116f, -0.138466f, -0.413395f, -0.223453f, |
| -0.172127f, -0.251265f, -0.048239f, -0.395519f, 0.023141f, 0.037459f, |
| -0.249593f, -0.062215f, -0.047209f, -0.435189f, -0.164155f, -0.077590f, |
| -0.241164f, -0.126128f, -0.038243f, -0.180888f, 0.198840f, -0.328036f, |
| -0.169790f, 0.036506f, 0.052572f, -0.183570f, -0.073617f, -0.244959f, |
| 0.266498f, 0.032846f, -1.902106f, 0.486078f, 2.414993f, 0.975182f, |
| -0.382875f, 1.647810f, -2.197017f, -0.890107f, 0.221287f, 0.010889f, |
| 3.817042f, 0.572728f, 0.092466f, 0.473337f, -1.634659f, -1.069455f, |
| 1.486776f, -1.023850f, 0.088184f, 0.008842f, 0.518202f, 0.270259f, |
| 1.757191f, -0.121839f, -2.912229f, -1.250866f, -2.381808f, 0.335309f, |
| -0.120079f, -0.061294f, -0.058725f, -0.315169f, -0.262443f, 0.072434f, |
| -0.267836f, -0.319354f, -0.274975f, 0.068970f, -0.406467f, 0.044074f, |
| -0.152311f, -0.333656f, -0.228355f, -0.185613f, 0.017346f, -0.177674f, |
| -0.090675f, -0.102047f, -0.011768f, -0.025280f, -0.271661f, 0.098099f, |
| -0.312272f, -0.222217f, -0.100548f, 0.106260f, -0.034655f, 0.135109f, |
| -0.021276f, 0.018177f, -0.353097f, -0.011128f, 0.061136f, -0.511662f, |
| -0.223236f, -0.308841f, 0.118789f, -0.154628f, -0.053178f, -0.055973f, |
| 0.013175f, -0.368337f, -0.090863f, -0.116920f, 0.178990f, -0.025278f, |
| -0.190553f, -0.238092f, 0.303943f, -0.024944f, 0.719373f, 0.384332f, |
| -0.378480f, -0.423316f, 0.709922f, 0.758514f, -1.559023f, -2.503173f, |
| 0.068652f, -0.234741f, -0.182932f, 0.037878f, 0.020684f, -0.174142f, |
| -0.182300f, -0.052796f, -0.219145f, 0.113028f, -1.041826f, 0.035317f, |
| 0.919904f, -0.676011f, 0.652297f, 1.456447f, -0.166904f, -0.861823f, |
| 0.895827f, 0.429821f, -0.180376f, -0.076587f, -0.273945f, -0.288990f, |
| -0.206692f, -0.080745f, -0.085444f, 0.186953f, -0.050135f, 0.044243f, |
| -0.391706f, -0.160498f, -0.292268f, 0.164060f, 0.412649f, 0.211611f, |
| -0.327294f, -0.919399f, 0.320297f, 0.385284f, -0.088848f, -0.072556f, |
| -0.384813f, -0.176267f, -0.065918f, 0.134724f, -0.231104f, -0.337707f, |
| -0.195442f, -0.263569f, 0.098090f, -0.341411f, -0.189211f, -0.439276f, |
| -0.404046f, 0.262491f, -0.311093f, -0.086454f, -0.013400f, -0.061447f, |
| -0.026945f, -0.112036f, -0.322985f, 0.078500f, -0.230205f, -0.344535f, |
| -0.021087f, 0.110220f, -0.128671f, 0.044219f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_16_layer0[64] = { |
| 2.936406f, -0.396539f, -0.110456f, -1.254954f, 0.785350f, 0.516290f, |
| -0.172341f, 0.254386f, -0.192465f, -0.106751f, -0.055518f, -0.094994f, |
| 0.000000f, -0.065018f, -0.004908f, -0.130483f, -0.119580f, -0.142072f, |
| 0.457446f, -0.125051f, -0.107712f, 0.714607f, -0.140809f, -1.788650f, |
| -0.087199f, 0.000000f, -1.290050f, 0.443930f, -0.110634f, -0.109380f, |
| -0.188213f, -1.414179f, 1.193579f, 0.388775f, -0.873193f, -0.110050f, |
| -0.072565f, -0.117050f, -0.119132f, 0.456959f, -0.132069f, 0.131974f, |
| 1.160474f, 1.746465f, 0.442628f, -0.188849f, -0.207794f, -0.108364f, |
| -0.856655f, -2.141620f, 0.335476f, -0.105508f, -0.212162f, -0.109319f, |
| -0.237213f, -0.109980f, -0.291044f, -0.137877f, 0.470191f, -0.023908f, |
| 0.123809f, -0.109797f, 0.200510f, -0.147542f, |
| }; |
| |
| static const float av1_ab_partition_nn_weights_16_layer1[64 * LABEL_SIZE] = { |
| -6.823716f, 1.406568f, -0.144009f, 2.228765f, 0.838336f, 0.738107f, |
| -0.319014f, -0.148756f, 0.240862f, -0.111089f, -0.004241f, 0.025758f, |
| -0.193820f, -0.246362f, -0.181363f, -0.201556f, 0.024268f, 0.252994f, |
| -0.289443f, 0.194932f, 0.057467f, 0.724735f, 0.014063f, 1.361352f, |
| 0.025191f, 0.024274f, 0.231462f, -7.227959f, -0.094515f, 0.039946f, |
| 0.412719f, 0.812318f, 3.038903f, -0.286289f, 0.647482f, -0.115114f, |
| 0.053590f, 0.066069f, 0.153134f, 0.996250f, -0.125700f, 0.951365f, |
| -6.243494f, -4.827697f, 0.566320f, 0.239515f, -0.099702f, 0.054546f, |
| 1.847330f, 3.680076f, -3.049829f, -0.127709f, 0.068469f, -0.017794f, |
| 0.223864f, -0.106778f, -0.020425f, -0.040226f, -0.251890f, -0.168673f, |
| -0.552073f, 0.043311f, 0.218668f, 0.033209f, -3.199210f, 0.193079f, |
| 0.321406f, 0.718307f, -0.181418f, -0.459612f, -1.981170f, 0.968496f, |
| -0.029757f, -0.130065f, 0.043782f, 0.072394f, -0.088686f, 0.025322f, |
| 0.129882f, 0.101324f, 0.335707f, 0.072714f, -2.079774f, 0.203997f, |
| 0.239321f, -0.301757f, 0.257845f, 1.288382f, -0.031275f, -0.234194f, |
| 0.310722f, 2.045469f, 0.034716f, 0.135638f, -0.251388f, 0.320071f, |
| -1.065301f, -0.322731f, -0.545028f, 0.226276f, 0.090799f, 0.019289f, |
| 0.048950f, -1.079300f, 0.231938f, 0.083683f, 4.762127f, 0.145037f, |
| -0.145549f, 0.075592f, 0.172336f, 0.108175f, 0.333751f, 1.090501f, |
| 1.056114f, 0.047073f, 0.182052f, -0.081587f, 0.089900f, 0.339286f, |
| 2.049988f, 0.073585f, 0.537355f, -0.243322f, -0.010179f, -0.052601f, |
| -0.174915f, 0.117793f, 2.222990f, -2.520837f, -0.092699f, 1.199887f, |
| 0.138720f, 0.679918f, -0.463155f, -0.659496f, -0.109913f, -0.003398f, |
| 0.114633f, -0.128377f, 0.092970f, -0.107489f, -0.191078f, 0.185182f, |
| 0.216980f, -0.019343f, 3.443133f, 0.287953f, 0.099314f, 0.985958f, |
| 0.157268f, -0.606516f, 0.049418f, -0.221809f, -0.453081f, -0.344796f, |
| -0.003735f, -0.107269f, -0.128541f, -0.259543f, -0.934806f, -0.542456f, |
| -1.011192f, 0.022795f, 0.186363f, -0.076356f, -0.050932f, -0.165098f, |
| 0.168177f, -0.101596f, -5.270886f, 2.553943f, -0.440870f, -0.017494f, |
| 0.215208f, -0.017032f, 1.495915f, -4.304677f, 0.762211f, 0.182937f, |
| 0.254406f, -0.029433f, -0.088364f, -0.110160f, -0.108257f, -0.036538f, |
| 0.737697f, -0.234989f, 0.168095f, 0.245118f, -0.077262f, 0.195718f, |
| 0.753302f, -1.637869f, 0.126227f, 0.982129f, -0.121444f, -0.295570f, |
| -1.215799f, 0.147867f, -0.068496f, 0.132726f, -0.005772f, -0.181774f, |
| 0.126513f, 0.204723f, -0.366123f, 0.103906f, -0.148053f, -0.075272f, |
| 0.243884f, -0.104828f, 0.198988f, 0.501034f, -0.112671f, 0.111421f, |
| 0.167508f, -0.117803f, -0.738624f, 2.046292f, 0.124011f, 0.057983f, |
| -0.359154f, -0.648883f, -0.259462f, -0.459041f, -2.501223f, -0.065138f, |
| 0.122417f, 0.060291f, -0.129033f, -0.843086f, 0.268241f, -0.399927f, |
| 1.585888f, 1.816393f, -0.631427f, 0.127826f, 0.088105f, 0.073488f, |
| 0.717694f, -1.497362f, 2.608528f, 0.066896f, -0.079230f, 0.223436f, |
| -0.010530f, 0.175310f, 1.120365f, 0.034391f, 0.835312f, 0.071652f, |
| -0.080615f, 0.111395f, 0.162742f, 0.079927f, -3.859582f, -0.638431f, |
| -0.167880f, -0.992659f, -0.885355f, -1.276197f, 1.334344f, 0.931940f, |
| -0.078244f, -0.149030f, -0.070974f, -0.133566f, 0.200034f, 0.102793f, |
| -0.048546f, 0.063545f, 0.023864f, -0.190863f, 1.934257f, -0.136286f, |
| -0.107916f, -0.637468f, 0.066449f, 1.089693f, -0.214047f, -0.265780f, |
| 0.899660f, -0.130333f, 0.288311f, -0.049024f, 0.090202f, 0.487969f, |
| 0.339704f, 0.858479f, 0.841253f, -0.184100f, -0.637070f, -0.125071f, |
| -0.077650f, -0.087877f, 0.202268f, -0.027300f, 2.842862f, -0.100698f, |
| -0.259080f, 0.260556f, 0.157912f, -0.070364f, 0.467190f, 1.200037f, |
| 1.419317f, -0.033588f, -0.227824f, 0.292617f, 0.228574f, 0.213839f, |
| -1.091099f, -0.022258f, -1.294681f, 0.136118f, 0.081652f, -0.185359f, |
| -0.039706f, 0.191407f, -2.053219f, -0.261934f, 0.047812f, -0.029536f, |
| -0.823869f, -1.090534f, -0.755890f, 0.441035f, -0.167945f, 0.231441f, |
| -0.135013f, -0.260762f, 0.256872f, 0.130339f, -0.243751f, 0.189760f, |
| -0.288454f, 0.145363f, 0.338490f, 0.403898f, -0.022814f, -1.263598f, |
| -0.101315f, 0.860135f, 0.136511f, 0.028942f, 0.574047f, 2.656370f, |
| 0.037587f, -0.188690f, -0.125312f, 1.100435f, -1.080402f, 0.380905f, |
| 0.004635f, 0.097144f, -0.214309f, 0.085552f, -0.285066f, -0.705134f, |
| -0.054704f, -0.319951f, 5.486626f, 0.958158f, -1.380585f, 0.223340f, |
| -0.169167f, -0.170697f, -0.216748f, 0.324232f, 2.684204f, -0.008490f, |
| -0.211052f, -0.201190f, 0.123466f, -0.000234f, 0.579907f, 0.096938f, |
| -0.042745f, 0.201855f, 0.157195f, -0.261440f, 0.029699f, -0.046599f, |
| 1.618216f, -2.596280f, -0.377420f, -0.526725f, -0.493592f, -0.579615f, |
| 0.579699f, -0.100392f, 0.150694f, 0.061794f, 0.200425f, -0.062515f, |
| -0.179122f, 0.250112f, -0.344675f, -0.118359f, -0.095670f, 0.152311f, |
| 3.662276f, -0.154921f, -0.312991f, 0.972008f, -0.308596f, -0.190426f, |
| 0.133889f, -0.238673f, -0.094726f, 1.683835f, -0.215629f, -0.198890f, |
| -0.035278f, -0.367973f, -0.822435f, 0.240848f, -0.194656f, 0.034655f, |
| -0.079424f, 0.146670f, 0.026646f, -0.034507f, 0.059467f, -0.153109f, |
| -0.431033f, 2.552991f, -1.894091f, -0.180462f, -0.306839f, -0.025648f, |
| 1.026326f, -3.096230f, 1.346935f, 0.033633f, -0.181827f, 0.094376f, |
| 0.001696f, -0.379264f, -1.069503f, -0.140972f, -0.208769f, -0.195239f, |
| 0.281795f, -0.127251f, 0.180776f, 0.067763f, 0.697124f, -1.040779f, |
| 0.111280f, 0.188351f, -0.340234f, -0.207790f, -0.720075f, -0.137409f, |
| -0.070310f, -0.032918f, -0.060787f, 0.131484f, -0.077845f, -0.258652f, |
| 0.056911f, -0.062034f, 0.007663f, -0.185100f, 1.340361f, 0.014096f, |
| -0.124602f, 0.194241f, 0.128383f, 0.360465f, 0.082979f, -0.050475f, |
| -0.519294f, 3.323262f, 0.067014f, 0.221203f, -0.085082f, -0.228606f, |
| -0.916668f, -0.022643f, -1.386737f, -0.131902f, -0.349952f, -0.032874f, |
| -0.189190f, -0.898790f, -0.102394f, -1.017387f, 2.214050f, 1.790253f, |
| -1.913561f, -0.043716f, -0.214924f, -0.194598f, -0.064723f, -1.671793f, |
| 2.251166f, -0.146007f, 0.138527f, -0.003134f, 0.103665f, 0.006928f, |
| -0.240253f, -0.227464f, 0.578437f, -0.214724f, 0.503085f, 0.158093f, |
| 0.033091f, 0.008061f, 4.815371f, 2.132264f, 0.281850f, -2.288560f, |
| -0.145012f, 1.296832f, -0.362401f, -0.403252f, 0.109873f, 0.185746f, |
| 0.244764f, 0.172367f, -0.185588f, 0.139801f, -0.178254f, 0.068629f, |
| 0.358488f, -0.153969f, -6.433524f, 0.225983f, -0.138123f, -0.095971f, |
| -0.036089f, -1.400083f, 0.265908f, 0.257787f, 0.181144f, -1.647228f, |
| -0.136289f, -0.074206f, 0.122988f, -0.088895f, -1.266717f, 0.006010f, |
| 0.536681f, 0.263061f, -0.032207f, -0.155136f, 0.086431f, 0.441950f, |
| -0.060755f, -0.280683f, -0.783475f, -2.567033f, 1.093221f, 0.117667f, |
| -0.000408f, 0.225719f, -2.199698f, 0.141447f, -1.459051f, 0.051315f, |
| 0.203228f, 0.354432f, -0.005775f, -0.028073f, -0.965817f, 0.231083f, |
| -0.666884f, 0.026283f, -0.317486f, 0.210754f, 0.123897f, 0.223827f, |
| 4.214405f, 1.457334f, -0.253945f, -1.306733f, -0.391235f, 0.451154f, |
| -1.553888f, -0.353429f, 0.069533f, 0.159278f, -0.173836f, -0.004952f, |
| -0.137033f, 0.127012f, 0.143600f, 0.051587f, -0.070549f, 0.066509f, |
| -5.776547f, 0.180021f, -0.189183f, -1.288504f, -0.233575f, -1.473873f, |
| 0.140940f, 0.144451f, -0.104534f, 2.089873f, -0.168168f, 0.110726f, |
| 0.132134f, -0.215223f, -1.682754f, 0.157757f, -0.146163f, 0.064882f, |
| 0.117313f, -0.038780f, -0.124720f, -0.501697f, 0.092047f, -0.233992f, |
| 3.324976f, 0.516601f, 1.294202f, 0.119989f, 0.061055f, 0.043420f, |
| -2.750727f, -0.382812f, -0.648496f, -0.115353f, -0.334205f, 0.024354f, |
| -0.282998f, -0.282705f, 0.073798f, 0.169851f, 0.135651f, 0.182677f, |
| -0.040220f, 0.132462f, -0.303120f, -0.230113f, 6.165739f, -0.258596f, |
| 0.024127f, -1.388283f, -0.006042f, 0.572600f, 0.348411f, -0.387376f, |
| -0.075845f, 0.122319f, -0.029616f, 0.077873f, 0.154763f, 0.049073f, |
| 0.018597f, 0.102688f, -0.204165f, 0.020734f, -1.389133f, -0.032854f, |
| -0.147561f, 0.853944f, 0.132100f, -3.259659f, 0.243745f, 0.181529f, |
| -0.738414f, 1.509994f, 0.023470f, -0.005329f, 0.066115f, -1.345081f, |
| -1.455402f, -0.172023f, -0.194625f, 0.071885f, -0.201742f, -0.262402f, |
| 0.077601f, -0.048938f, 0.257993f, -0.504029f, -2.032415f, 1.158880f, |
| 0.448647f, -0.025633f, 0.117586f, -0.072275f, -0.673744f, -3.854342f, |
| -0.983843f, 0.047766f, -0.017193f, -0.215775f, -0.158743f, -0.232042f, |
| -0.509112f, 0.148812f, 0.130122f, 0.006486f, -0.099016f, 0.022514f, |
| -0.486850f, -0.059623f, 4.012731f, 0.025454f, 0.029059f, -0.783546f, |
| -0.295260f, 0.322521f, -0.473201f, -0.172100f, -0.100087f, -0.076516f, |
| -0.258367f, -0.112897f, 0.269364f, -0.065912f, 0.169022f, -0.178783f, |
| -0.095114f, 0.122089f, -2.790099f, -0.100431f, -0.087963f, -0.009431f, |
| -0.087819f, -2.774399f, -0.100757f, 0.013005f, -0.964533f, 3.236665f, |
| -0.354903f, -0.144169f, -0.166869f, -1.396513f, -0.931271f, -0.046261f, |
| -1.799262f, -0.365269f, 0.108611f, 0.037994f, 0.024747f, -1.073639f, |
| -0.203158f, -0.935006f, 1.880891f, 1.578385f, 0.726272f, -0.024546f, |
| -0.011626f, -0.151363f, -1.121716f, -1.787484f, 0.232806f, 0.075451f, |
| 0.182899f, 0.092215f, -0.207347f, -0.030111f, 0.054316f, 0.192481f, |
| 0.594639f, -0.247694f, 0.547471f, -0.032094f, -0.065000f, 0.007198f, |
| 1.605377f, -0.155945f, -0.066200f, -2.343716f, -1.016283f, -0.079321f, |
| 0.919365f, 0.599980f, 0.125545f, 0.265813f, 0.246884f, 0.095385f, |
| -0.260374f, -0.202916f, -0.042770f, 0.234967f, -0.233139f, -0.326994f, |
| -1.375256f, 0.121766f, 0.077433f, -1.103569f, 0.019497f, -1.029185f, |
| 0.253905f, 0.206569f, 0.187334f, -0.237089f, -0.294351f, 0.164137f, |
| 0.149696f, -0.749787f, -0.413433f, 0.976587f, 1.027976f, -0.285264f, |
| 0.209273f, -0.124762f, 0.050884f, 0.250764f, -0.082031f, -0.646520f, |
| 4.116680f, 0.437336f, 0.671684f, 0.129509f, -0.078462f, 0.014072f, |
| -0.678232f, 0.094831f, 1.125624f, 0.207070f, -0.154750f, -0.025780f, |
| -0.103030f, 0.118019f, -0.908186f, -0.263546f, -1.555324f, -0.236887f, |
| -0.217854f, -0.051790f, 0.017915f, 0.171001f, 1.355562f, 0.094603f, |
| -0.233929f, -1.282169f, -0.773183f, -0.161682f, -0.834565f, -0.286776f, |
| -0.298901f, 0.038162f, 0.251899f, 0.039612f, -0.022935f, -0.232308f, |
| -0.043855f, -0.192892f, -0.279009f, -0.182234f, -1.272808f, -0.070344f, |
| -0.092432f, -1.915946f, -0.134373f, -1.405496f, -0.067071f, -0.131922f, |
| 0.185269f, 1.465082f, 0.040240f, 0.112665f, 0.144329f, -0.286112f, |
| -0.617649f, 0.916177f, 0.221044f, -0.079867f, 0.170251f, -0.093638f, |
| -0.212620f, -0.305945f, -0.234356f, -0.482501f, 3.928472f, 1.241179f, |
| 0.355922f, -0.170848f, -0.189168f, 0.080225f, -1.357793f, 0.190890f, |
| 0.976800f, -0.068070f, -0.016295f, -0.088623f, -0.129560f, -0.212267f, |
| -0.071537f, -0.219501f, -0.655198f, -0.225188f, -0.116024f, 0.224174f, |
| -0.049715f, -0.178005f, 3.029985f, -1.141546f, 0.080066f, -1.932316f, |
| -0.641137f, -0.189564f, 0.935080f, 0.136119f, 0.015558f, -0.179331f, |
| 0.204571f, 0.020350f, 0.009362f, 0.108478f, 0.037076f, -0.049009f, |
| 0.081090f, -0.180202f, 1.455561f, -0.081559f, 0.059361f, 0.484971f, |
| 0.160923f, -2.170744f, -0.013204f, 0.126561f, -0.407122f, 1.223661f, |
| 0.044262f, 0.118044f, 0.058274f, -1.747100f, -0.171318f, 0.971374f, |
| 0.306995f, -0.103268f, -0.319443f, -0.333176f, -0.038608f, 0.119674f, |
| -0.106479f, -0.907933f, 1.121231f, 1.673840f, -0.421458f, -0.021146f, |
| -0.254838f, 0.097632f, 0.235109f, -2.901782f, 0.289518f, -0.355459f, |
| -0.068264f, -0.179121f, 0.068560f, -0.047570f, -0.522523f, -0.228963f, |
| -1.037158f, -0.163723f, 0.280563f, -0.000868f, -0.197220f, -0.239329f, |
| 1.985274f, -0.256181f, -0.064341f, -0.822417f, -0.465140f, -0.010942f, |
| -0.792024f, -0.114290f, 0.060969f, 0.104106f, -0.252123f, -0.150400f, |
| -0.133277f, 0.267147f, 0.274413f, 0.223744f, -0.180223f, -0.345415f, |
| -0.104883f, 0.119210f, -0.095041f, -0.301635f, 0.013175f, -2.128121f, |
| -0.147208f, -0.151509f, -0.692013f, 3.418555f, -0.016541f, 0.171511f, |
| 0.107159f, -1.516672f, 0.127408f, 0.687035f, -0.906486f, -0.145463f, |
| -0.169382f, -0.143906f, 0.125091f, -0.960645f, -0.180869f, -0.716908f, |
| 2.840951f, 1.904919f, -0.416268f, -0.425181f, -0.194697f, -0.075932f, |
| -0.950604f, -1.599800f, 0.943671f, -0.022744f, -0.270492f, 0.080843f, |
| -0.372916f, 0.047838f, -0.100300f, -0.026600f, 0.011733f, -0.226051f, |
| 0.172790f, -0.172982f, 0.041258f, -0.299379f, |
| }; |
| |
| static const float av1_ab_partition_nn_bias_16_layer1[LABEL_SIZE] = { |
| -0.053805f, -1.248639f, 0.520965f, -0.904962f, -0.126425f, -0.118798f, |
| 0.748430f, 0.203096f, 0.059317f, 0.418219f, 0.841294f, 0.402693f, |
| -0.658522f, 0.723479f, 0.544264f, 1.035225f, |
| }; |
| |
| static const NN_CONFIG av1_ab_partition_nnconfig_16 = { |
| FEATURE_SIZE, // num_inputs |
| LABEL_SIZE, // num_outputs |
| 1, // num_hidden_layers |
| { |
| 64, // num_hidden_nodes |
| }, |
| { |
| av1_ab_partition_nn_weights_16_layer0, |
| av1_ab_partition_nn_weights_16_layer1, |
| }, |
| { |
| av1_ab_partition_nn_bias_16_layer0, |
| av1_ab_partition_nn_bias_16_layer1, |
| }, |
| }; |
| |
| #undef FEATURE_SIZE |
| #undef LABEL_SIZE |
| |
| #define FEATURE_SIZE 18 |
| #define LABEL_SIZE 4 |
| |
| static const float av1_4_partition_nn_weights_16_layer0[FEATURE_SIZE * 24] = { |
| -2.032866f, 0.056691f, 0.495960f, 0.778785f, 0.548153f, -0.806942f, |
| 0.481155f, 0.282298f, 0.584980f, 0.504688f, 0.209648f, 0.234616f, |
| 0.213484f, 0.221969f, 0.205862f, 0.235054f, 0.317863f, 0.257139f, |
| 0.529478f, 0.098122f, -0.657532f, 0.036296f, 0.327728f, 1.323180f, |
| -0.813082f, 0.160216f, -0.702030f, 0.722733f, -0.270576f, -0.347416f, |
| -0.264700f, -0.254248f, 0.159820f, 0.087995f, -0.184163f, 0.117357f, |
| 0.074194f, -0.667369f, 0.498246f, 0.420506f, 0.072409f, -0.121581f, |
| 0.315788f, 0.000525f, 0.414986f, 0.678166f, -0.011230f, 0.188131f, |
| -0.227749f, 0.009564f, 0.108672f, 0.106923f, -0.080695f, -0.279382f, |
| -0.061339f, -0.297835f, -0.134707f, 0.145865f, -0.009655f, -0.000842f, |
| -0.047436f, -0.159149f, -0.320353f, -0.089646f, -0.344765f, 0.313416f, |
| -0.143413f, 0.279668f, 0.000885f, -0.022380f, -0.140194f, -0.310473f, |
| 0.252699f, 0.066204f, 0.477568f, 0.994609f, -0.276000f, 1.213182f, |
| 0.277028f, -0.411570f, -0.211559f, 0.377815f, 0.121488f, -0.100559f, |
| -0.317082f, -0.251039f, -0.335181f, -0.154114f, -0.052726f, -0.332558f, |
| -0.143196f, -0.334035f, 0.162305f, 0.142279f, -0.001210f, -0.135252f, |
| -0.033562f, 0.204307f, -0.039757f, -0.394174f, 0.126617f, -0.128648f, |
| -0.410979f, 0.107641f, -0.117573f, -0.326512f, 0.235166f, 0.084959f, |
| 0.290063f, -0.005838f, 0.459894f, 1.023709f, -0.196145f, 1.100137f, |
| -0.319815f, -0.308526f, -0.443389f, -0.272769f, -0.035259f, -0.026932f, |
| -0.029743f, 0.125113f, -0.131024f, -0.321458f, -0.143996f, 0.008714f, |
| -0.101234f, 0.079706f, -1.128615f, -0.467381f, 0.220563f, -0.409900f, |
| -0.435353f, 0.759499f, -0.465799f, -0.394309f, 0.176282f, -0.086275f, |
| -0.161225f, -0.354814f, 0.562871f, 0.418253f, 0.414361f, 0.445480f, |
| -0.995903f, -0.086632f, -0.230645f, 0.354656f, -0.317576f, 0.079926f, |
| 0.424369f, 0.997232f, -0.304388f, 1.071667f, -0.023540f, 0.029677f, |
| 0.108564f, 0.183581f, -0.201395f, -0.054854f, -0.193039f, -0.049899f, |
| -0.271949f, -0.358483f, 0.304930f, 0.023823f, -0.009319f, -0.214247f, |
| 0.100712f, -0.050162f, 0.327103f, -0.212999f, -0.030496f, 0.316380f, |
| -0.439589f, -0.249959f, 0.229777f, -0.353664f, -0.384559f, 0.114236f, |
| 0.023119f, 0.007927f, 0.618368f, 0.957759f, -0.019780f, -1.002389f, |
| 0.564277f, -0.839531f, 1.040445f, 0.054340f, 0.031908f, -0.032893f, |
| -0.019170f, -0.042011f, 0.568928f, 0.362567f, -0.559999f, -0.605344f, |
| -0.586146f, -0.290778f, 0.195943f, -0.109580f, -0.088898f, -0.113054f, |
| 0.293282f, 0.429019f, 0.306136f, 0.863025f, 0.021234f, 0.125770f, |
| -0.097108f, -0.072659f, -0.137053f, -0.191631f, 0.106281f, 0.064151f, |
| 0.029883f, 0.076287f, 0.757543f, 0.276713f, -2.529775f, -0.351727f, |
| -1.832316f, 0.544780f, -0.944529f, 0.509705f, -0.010236f, -0.016181f, |
| 0.021520f, 0.086417f, 0.041312f, 0.296853f, -0.372378f, 0.354446f, |
| -1.366762f, 0.048875f, 0.464918f, -0.007450f, 0.750013f, -0.360261f, |
| 0.518532f, 0.753776f, 0.641448f, 0.710746f, 0.250866f, 0.257063f, |
| 0.283421f, 0.253585f, 0.170303f, 0.210426f, 0.208842f, 0.158000f, |
| -0.033144f, 0.130748f, 0.907147f, 0.409248f, -0.854301f, -0.981307f, |
| 0.294427f, -0.507137f, 1.079967f, 0.203203f, 0.383890f, 0.368278f, |
| 0.305122f, 0.449288f, -0.044507f, -0.547263f, -0.298245f, -0.497834f, |
| 0.007016f, -0.101982f, -0.073488f, -0.096111f, -0.479418f, -0.045497f, |
| 0.033502f, -0.018578f, -0.231531f, 0.177949f, 0.099564f, -0.010233f, |
| -0.333055f, -0.078586f, -0.417867f, 0.171271f, 0.013662f, -0.143599f, |
| -0.117296f, 0.135382f, 0.048321f, 0.000924f, -0.055024f, -0.405595f, |
| -0.068260f, -0.271011f, -0.436425f, 0.206751f, -0.899890f, 0.605510f, |
| 0.535649f, -0.238919f, -0.037619f, -0.213734f, -0.391360f, -0.132344f, |
| 0.004660f, 0.176644f, -1.008475f, -0.038895f, 0.155429f, -0.095229f, |
| -0.680124f, -0.258063f, -0.261901f, 0.110380f, -0.337649f, -0.505870f, |
| -1.428536f, 0.610629f, 0.254905f, 0.045098f, 0.044109f, 0.172329f, |
| 0.060001f, -0.234009f, -0.184855f, -0.153028f, -0.140897f, -0.152006f, |
| -0.312134f, 0.081261f, 0.160166f, 0.112690f, 0.266081f, 0.030175f, |
| -0.242746f, 0.000754f, -0.341811f, -0.149774f, -0.017484f, -0.301342f, |
| -0.121466f, 0.067300f, 0.342176f, 0.474538f, 0.085441f, -0.263935f, |
| 0.479235f, -0.003713f, -0.784840f, 0.119480f, 0.456632f, -0.640082f, |
| -0.080575f, -0.744403f, 0.259970f, 0.034667f, -0.274641f, -0.257594f, |
| -1.121124f, -0.003745f, -0.420693f, 0.300441f, -0.100976f, -1.049016f, |
| 0.201960f, 0.113054f, 0.187010f, 1.237427f, 0.054803f, -0.028673f, |
| 0.003596f, -0.034724f, 0.117246f, 0.190977f, 0.278915f, 0.224307f, |
| 0.017852f, -0.336233f, -0.372311f, -0.182284f, -0.143510f, 0.331466f, |
| 0.045698f, -0.301095f, 0.184447f, 0.348240f, -0.017021f, -0.145064f, |
| -0.000221f, -0.382256f, -0.302683f, -0.083927f, -0.008070f, 0.217907f, |
| 0.647597f, -0.050490f, -0.572736f, -0.985748f, -0.289943f, 0.041391f, |
| -0.795464f, -0.186680f, -0.354062f, -0.617400f, -0.282783f, -0.170450f, |
| -0.197197f, -0.146496f, -0.173692f, -0.106277f, -0.071004f, -0.124405f, |
| -0.971412f, 0.038542f, 0.705204f, 0.887113f, 0.150430f, -0.243676f, |
| 0.638410f, 0.320953f, 0.776676f, 0.527584f, 0.070389f, 0.051554f, |
| 0.177519f, 0.140451f, 0.128892f, 0.087771f, 0.197660f, 0.194764f, |
| }; |
| |
| static const float av1_4_partition_nn_bias_16_layer0[24] = { |
| 0.614063f, -0.384872f, 0.084884f, -0.023980f, -0.378765f, -0.082312f, |
| -0.458271f, 0.189578f, -0.046169f, -0.073308f, -0.372322f, 0.162793f, |
| 0.148803f, 0.829214f, -0.221162f, -0.111157f, -0.017484f, -0.280596f, |
| -0.031905f, -0.143459f, 0.078823f, -0.021940f, 0.026834f, 0.257472f, |
| }; |
| |
| static const float av1_4_partition_nn_weights_16_layer1[24 * LABEL_SIZE] = { |
| -0.985391f, 0.587616f, 0.740683f, 0.192066f, 0.447080f, -0.016585f, |
| 0.680449f, 0.028983f, 0.643111f, 0.234338f, 0.107148f, 0.328456f, |
| -0.216394f, 1.106838f, -0.179062f, -0.129108f, -0.121655f, -0.151340f, |
| -0.306017f, -0.350989f, 0.859284f, -0.372831f, -0.954419f, 0.250495f, |
| 1.046732f, 0.287923f, -0.421088f, 0.326613f, -0.314396f, -0.084757f, |
| -0.474228f, 0.687999f, 0.052334f, 0.441708f, -0.630698f, -0.350348f, |
| -0.602067f, -0.434161f, -0.489824f, -0.313193f, 0.315568f, 0.603119f, |
| 0.120245f, 0.182920f, -1.117797f, -0.239594f, -0.296296f, -0.718093f |