• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2018, Alliance for Open Media. All rights reserved
3  *
4  * This source code is subject to the terms of the BSD 2 Clause License and
5  * the Alliance for Open Media Patent License 1.0. If the BSD 2 Clause License
6  * was not distributed with this source code in the LICENSE file, you can
7  * obtain it at www.aomedia.org/license/software. If the Alliance for Open
8  * Media Patent License 1.0 was not distributed with this source code in the
9  * PATENTS file, you can obtain it at www.aomedia.org/license/patent.
10  */
11 
12 #ifndef AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_
13 #define AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_
14 
15 #ifdef __cplusplus
16 extern "C" {
17 #endif
18 
19 #include "av1/encoder/ml.h"
20 
21 #define FEATURE_SIZE 10
22 #define LABEL_SIZE 16
23 // nn model for ab partition pruning, 128x128.
24 static const float av1_ab_partition_nn_weights_128_layer0[FEATURE_SIZE * 64] = {
25   -0.715251f, -0.015767f, -0.667353f, -0.345255f, 0.177887f,  -0.469759f,
26   0.426152f,  0.489798f,  0.469865f,  0.773821f,  0.088517f,  0.074585f,
27   0.838754f,  0.048449f,  -0.007584f, 0.638968f,  0.233305f,  -0.319236f,
28   -0.257124f, -0.170869f, 0.137180f,  0.114852f,  -0.721241f, -0.947962f,
29   -0.411298f, 0.494306f,  -0.060435f, -0.648421f, -0.126624f, 0.072686f,
30   -0.143904f, -0.115839f, -0.175527f, -0.117728f, 0.040686f,  -0.189925f,
31   0.134361f,  -0.258070f, -0.177558f, 0.158049f,  0.168668f,  -0.062919f,
32   0.341986f,  0.038100f,  -0.435577f, -0.321255f, 0.203213f,  0.213061f,
33   0.533304f,  0.359296f,  -0.079558f, 0.004637f,  0.663904f,  0.043779f,
34   0.383018f,  1.136559f,  -0.084155f, 0.333057f,  -0.199011f, 0.152059f,
35   -0.078419f, -0.167752f, -0.093651f, 0.083171f,  -0.190143f, 0.086195f,
36   -0.280632f, -0.160663f, -0.017298f, 0.122628f,  -0.138116f, 0.062927f,
37   0.222462f,  0.626979f,  0.426928f,  0.117170f,  -0.240457f, 0.053750f,
38   0.038017f,  0.007359f,  -0.017595f, 0.101407f,  0.332891f,  0.074933f,
39   0.306498f,  0.219380f,  -0.151638f, -0.247976f, 0.343405f,  0.121256f,
40   0.049173f,  0.171474f,  -0.139608f, -1.016599f, -0.345553f, -0.901138f,
41   0.243401f,  0.059928f,  -0.089396f, -0.195565f, 0.364705f,  -0.020400f,
42   -1.383672f, 0.413018f,  0.536950f,  -0.020904f, -1.335306f, -0.732290f,
43   0.102885f,  0.315290f,  -0.208521f, -0.081811f, 0.182300f,  0.125712f,
44   -0.593833f, -0.220639f, -0.314155f, 0.188327f,  0.118503f,  0.524427f,
45   -1.083859f, -1.130640f, 0.390352f,  -0.045591f, 0.113160f,  -0.009149f,
46   -0.096183f, 0.115829f,  0.377752f,  0.318396f,  -0.591983f, 0.004797f,
47   -0.497377f, -0.342248f, 0.079546f,  -0.025249f, -0.295972f, 0.615501f,
48   -0.464372f, 0.418315f,  -0.173556f, 0.105217f,  0.298073f,  0.082478f,
49   0.033223f,  0.977341f,  -0.372982f, -0.052337f, 0.154124f,  0.396787f,
50   0.536654f,  -0.139061f, -0.223702f, 0.229666f,  -0.846766f, 0.107723f,
51   0.563839f,  -0.483141f, 0.304813f,  -0.765283f, 0.070964f,  0.151101f,
52   0.275188f,  0.490303f,  1.175892f,  0.085377f,  -0.191200f, 0.544532f,
53   -0.365075f, 0.167546f,  0.052183f,  -0.220529f, -0.212227f, -0.144988f,
54   -0.273356f, -0.062023f, 0.103993f,  -0.238493f, -0.161204f, -0.054611f,
55   -0.166672f, 0.128327f,  0.461751f,  -0.545822f, 0.739798f,  0.594386f,
56   -0.163192f, -0.332501f, 0.363834f,  -0.065043f, 0.474812f,  -0.138811f,
57   0.170924f,  -0.778142f, -0.316474f, -0.508065f, -0.039986f, -0.478001f,
58   0.340591f,  0.041783f,  0.055419f,  0.015155f,  -0.981830f, -1.355237f,
59   0.347516f,  1.155327f,  0.081319f,  0.274163f,  -0.327230f, -0.113478f,
60   0.556552f,  -0.055986f, 0.217318f,  -0.445351f, 0.325759f,  0.526547f,
61   -0.657434f, -0.572214f, -0.037087f, 0.081384f,  0.064518f,  0.014892f,
62   0.215279f,  1.834504f,  -0.242107f, 0.079810f,  0.129558f,  0.079588f,
63   -0.035189f, -0.221745f, -0.163414f, 0.043978f,  -1.028662f, -0.623609f,
64   1.130336f,  0.664661f,  -0.063975f, -0.415863f, 0.018581f,  0.157758f,
65   0.200570f,  0.063420f,  0.901039f,  -0.746286f, 0.196230f,  -0.290592f,
66   0.042373f,  -0.502500f, 0.183638f,  0.103394f,  -0.298858f, 0.145436f,
67   0.196916f,  0.108319f,  -0.448572f, -0.881385f, 0.302497f,  0.121679f,
68   -0.021327f, 0.025150f,  0.481306f,  -0.359634f, 0.350257f,  -0.228647f,
69   -0.669860f, 0.260025f,  -0.034182f, 0.619247f,  -0.158826f, -0.405864f,
70   0.674112f,  -0.027885f, -0.325274f, -0.241492f, 0.036024f,  -0.437685f,
71   -0.091458f, -0.109295f, -0.350676f, 0.044706f,  0.297059f,  0.016290f,
72   1.121203f,  1.289062f,  -1.299476f, -1.129221f, 0.103752f,  0.131302f,
73   -0.263265f, 0.222155f,  -0.229908f, 0.013922f,  -0.226001f, -0.248383f,
74   -0.004415f, -0.020958f, 0.055634f,  0.086200f,  0.114556f,  -0.184061f,
75   -0.096210f, -0.146466f, -0.249618f, -0.195998f, 0.088758f,  0.023781f,
76   -0.264460f, 0.157026f,  -0.235228f, -0.102564f, 0.043463f,  -0.187823f,
77   -0.257500f, -0.199049f, -0.242210f, 0.030448f,  0.221604f,  0.151804f,
78   -0.100404f, -0.073931f, 0.144749f,  -0.001572f, -1.438079f, -0.233716f,
79   0.733422f,  1.727080f,  -0.036397f, 0.027551f,  0.425321f,  0.085703f,
80   0.031186f,  0.032333f,  -0.675130f, 1.437733f,  -0.202392f, -0.525003f,
81   0.087048f,  0.328194f,  -0.079989f, -0.391088f, -0.238732f, -0.120660f,
82   -0.139600f, 0.154665f,  0.026202f,  -0.233501f, -0.009046f, -0.149187f,
83   -0.199646f, 0.115375f,  0.209762f,  -0.014875f, 0.124038f,  -0.119985f,
84   1.079625f,  -0.461513f, 0.614114f,  0.021003f,  0.439449f,  -0.824834f,
85   -0.299701f, 0.193817f,  -0.870551f, -1.262313f, -0.079517f, 0.341570f,
86   0.305310f,  -0.089721f, -0.317314f, -0.075631f, 0.127172f,  -0.208635f,
87   1.191922f,  0.163141f,  0.564285f,  0.286352f,  0.480865f,  0.173094f,
88   -0.094034f, -0.071339f, -0.328992f, -0.006382f, 0.314705f,  0.090258f,
89   -0.016099f, 0.193230f,  0.188061f,  0.398144f,  0.722781f,  0.769949f,
90   0.025442f,  -0.162016f, 0.070192f,  -0.056946f, -0.100957f, -0.219934f,
91   -0.203492f, -0.015454f, -0.013272f, -0.098008f, 0.051707f,  -0.017493f,
92   0.527446f,  0.083605f,  0.588318f,  0.878215f,  0.028747f,  -0.146479f,
93   -0.345170f, -0.136059f, -0.152005f, -0.203634f, 0.232702f,  -0.101340f,
94   -0.027733f, -0.282611f, 0.265366f,  0.082362f,  -0.265420f, -0.131124f,
95   0.166303f,  0.040194f,  -0.100710f, 0.579151f,  -0.530136f, 0.163422f,
96   -0.998821f, -1.565311f, -1.774785f, -2.493372f, 0.116970f,  -0.090302f,
97   1.723272f,  0.552370f,  -0.295954f, -0.439095f, -0.266730f, 0.027936f,
98   0.539616f,  -0.234902f, -0.167601f, -0.149877f, -0.242983f, 0.122353f,
99   -0.121620f, -0.205517f, -0.180144f, -0.264208f, 0.151500f,  -0.159378f,
100   0.029145f,  -0.050892f, -0.223407f, -0.246239f, 0.043152f,  -0.018460f,
101   0.169972f,  -0.187769f, -0.034670f, -0.238330f, 0.288070f,  -0.093243f,
102   -0.437105f, -0.573376f, 0.660073f,  0.285727f,  0.408470f,  0.158475f,
103   0.032699f,  0.056280f,  -0.237176f, -0.083003f, 0.105598f,  -0.169522f,
104   -0.260420f, -0.121100f, -0.173983f, -0.195693f, -0.232028f, 0.224940f,
105   0.029124f,  0.009580f,  -0.252034f, 0.103087f,  1.156561f,  0.603848f,
106   -0.562805f, -1.652742f, -0.568288f, -1.829395f, 0.046169f,  0.076095f,
107   1.490819f,  0.415893f,  -0.277788f, -0.115787f, 0.093750f,  0.270726f,
108   -0.395983f, -0.353742f, 0.034605f,  0.005342f,  0.184537f,  0.086445f,
109   0.156417f,  1.476367f,  0.122587f,  0.002145f,  0.431057f,  -0.381184f,
110   -1.646457f, -0.014009f, -0.671224f, 0.193726f,  -0.019247f, -0.031267f,
111   -0.046208f, 0.298733f,  0.064734f,  0.616984f,  0.039381f,  0.182722f,
112   -0.116670f, 0.233093f,  -1.214374f, -0.817970f, -0.064394f, -0.584783f,
113   0.077697f,  -0.266720f, 0.130875f,  -0.235295f, -0.265754f, -0.159999f,
114   -0.250114f, -0.183017f, 0.194403f,  -0.105808f, -0.169215f, -0.240866f,
115   -0.026662f, -0.045123f, -0.036175f, -0.167471f, -0.192908f, -0.232602f,
116   -0.267036f, -0.112500f, -0.257944f, -0.111909f, -0.802226f, -0.008800f,
117   0.881460f,  -0.678603f, 0.008666f,  -0.252053f, -0.341035f, -0.175290f,
118   0.183012f,  0.385991f,  0.079888f,  -0.014039f, -0.148653f, 0.671778f,
119   -0.130219f, 1.086467f,  0.129267f,  -0.040400f, -0.201221f, -0.077005f,
120   0.015890f,  0.000781f,  0.137764f,  1.389546f,  0.172152f,  0.047279f,
121   -0.042783f, 0.127740f,  0.141467f,  -0.335738f, -1.396392f, 0.031496f,
122   0.357385f,  0.343602f,  -0.714553f, 0.311014f,  0.132845f,  0.061149f,
123   0.006796f,  0.568106f,  -0.255949f, 0.104134f,  -0.993447f, 0.298135f,
124   -0.406590f, -0.049228f, -0.578570f, -0.188561f, -0.107046f, 0.374095f,
125   0.068481f,  0.036240f,  -0.495801f, 0.180574f,  -0.766129f, 0.886967f,
126   -0.568868f, -0.936062f, -0.418886f, -0.058735f, -0.511964f, -0.438596f,
127   0.019016f,  -0.015837f, 0.600197f,  0.429773f,  0.315026f,  0.319667f,
128   0.214617f,  -0.017316f, 0.270257f,  -0.040524f, 0.695803f,  -0.015223f,
129   -1.554965f, 0.356997f,  -1.472428f, 0.024637f,  -0.562958f, 0.870351f,
130   0.193635f,  0.036063f,  0.328638f,  0.200274f,  -1.634707f, 0.110534f,
131   0.420104f,  -0.072042f, -0.006404f, 0.171680f,
132 };
133 
134 static const float av1_ab_partition_nn_bias_128_layer0[64] = {
135   0.643147f,  -1.348826f, 0.431627f,  0.000000f,  0.102717f,  -0.772628f,
136   -0.034351f, -0.761977f, -0.638397f, 0.541969f,  -0.391311f, 0.563076f,
137   0.148553f,  0.267217f,  -0.788092f, 0.544573f,  -0.546280f, 0.000000f,
138   -0.446945f, 0.127732f,  0.270624f,  -0.219435f, -1.220203f, 0.324584f,
139   0.110885f,  0.276547f,  0.179726f,  -0.375160f, 0.026401f,  -0.032595f,
140   0.000000f,  -0.047932f, -0.648602f, -0.512637f, -0.031661f, -0.236761f,
141   0.476453f,  -0.028021f, -0.013673f, -0.015578f, -0.920077f, 0.000000f,
142   0.915351f,  -0.209962f, 0.000000f,  -0.025731f, 0.218288f,  0.000000f,
143   0.047726f,  -0.813077f, -1.263281f, 0.239087f,  0.278614f,  -0.030753f,
144   0.000000f,  0.346744f,  -0.948543f, -1.174211f, 0.216377f,  0.498913f,
145   0.853918f,  0.002504f,  -0.190403f, 0.452050f,
146 };
147 
148 static const float av1_ab_partition_nn_weights_128_layer1[64 * LABEL_SIZE] = {
149   0.179769f,  1.499417f,  -0.445135f, -0.142278f, -0.337661f, 0.682064f,
150   -0.203213f, 0.302171f,  0.226877f,  -0.422169f, 1.687586f,  0.783773f,
151   0.220995f,  0.253482f,  0.370435f,  -1.342775f, 0.337229f,  -0.271473f,
152   0.291796f,  1.362227f,  -1.751397f, -0.086178f, 0.725496f,  -0.118597f,
153   0.227963f,  -0.501577f, 0.223849f,  -0.122421f, -0.123437f, -0.051045f,
154   -0.020115f, 0.212711f,  0.246025f,  0.088120f,  -0.168995f, 1.740190f,
155   -0.195098f, 0.680339f,  -0.589572f, -0.075244f, 0.878766f,  0.064092f,
156   -3.548527f, 0.001660f,  0.107926f,  -0.169501f, -0.455212f, 0.123045f,
157   -1.836998f, 0.330365f,  1.301475f,  0.454761f,  -0.576552f, -0.190761f,
158   0.208459f,  0.618483f,  1.383364f,  0.970718f,  0.390174f,  0.406252f,
159   -0.564519f, -0.312062f, 1.345712f,  -0.151873f, 0.109290f,  0.408847f,
160   0.391243f,  0.152024f,  0.181764f,  -0.036263f, -0.160466f, 0.153595f,
161   0.049163f,  -0.753012f, -1.804062f, 0.347475f,  -2.746580f, 0.575618f,
162   0.261799f,  0.210505f,  -0.302054f, -0.109872f, 0.199506f,  -1.182971f,
163   0.723668f,  0.177758f,  -0.338202f, 0.254396f,  -0.220023f, 0.043504f,
164   0.669866f,  -0.040816f, -0.402730f, 0.017990f,  0.215523f,  -0.216816f,
165   0.454826f,  -0.726067f, -0.018750f, -0.928679f, 0.154315f,  -0.465641f,
166   0.144566f,  -0.030064f, -0.054667f, -0.154055f, 0.625384f,  1.323795f,
167   -0.159496f, 0.097072f,  -0.463197f, -0.057938f, 0.750290f,  -0.233061f,
168   0.412631f,  -0.535223f, -0.151423f, -0.154583f, 0.024721f,  -0.494448f,
169   0.230594f,  -0.980138f, -0.653968f, 0.126079f,  0.051814f,  -0.053219f,
170   -0.421708f, -0.228853f, 0.237885f,  0.888157f,  0.059655f,  0.241295f,
171   0.210443f,  0.228238f,  0.119127f,  -0.051989f, -0.355408f, 0.182215f,
172   0.244277f,  -0.104577f, -0.558035f, -0.023270f, 0.054571f,  0.700646f,
173   -0.223006f, 0.115523f,  0.023391f,  0.437264f,  0.709477f,  -0.531212f,
174   -0.094731f, 0.328161f,  -0.105418f, -0.133511f, 0.497168f,  -0.030948f,
175   -0.407132f, -0.043943f, 0.155505f,  0.251945f,  0.205010f,  0.167160f,
176   0.083654f,  -0.636810f, 0.401315f,  -0.398414f, 0.290046f,  0.206846f,
177   0.042218f,  0.168150f,  0.843181f,  -0.671242f, -0.202392f, -0.073301f,
178   0.142895f,  0.237466f,  0.212145f,  -0.091828f, 0.187038f,  -0.720841f,
179   -0.616069f, -0.238021f, 0.065365f,  0.434119f,  0.179023f,  -0.040107f,
180   -0.430734f, -0.297368f, 0.575954f,  0.382619f,  -0.709787f, -0.320810f,
181   0.242342f,  -0.047614f, 0.705216f,  0.098077f,  0.357179f,  0.046017f,
182   0.115074f,  -0.412305f, -0.272304f, 0.048096f,  -0.803811f, 0.275000f,
183   0.642198f,  0.180286f,  -0.087178f, -0.112707f, -0.394443f, 0.201989f,
184   0.241759f,  -1.038870f, 0.728124f,  0.800559f,  -1.296268f, 0.198612f,
185   -0.053478f, 0.414344f,  -0.510529f, 0.124179f,  -2.219115f, -0.074583f,
186   -0.143055f, 0.001697f,  0.810811f,  -0.657140f, 0.186818f,  -0.936414f,
187   0.539578f,  -0.308244f, -0.126624f, -0.204767f, 0.091145f,  -0.049340f,
188   0.252014f,  0.394582f,  0.018764f,  -0.060377f, -0.019133f, 0.064083f,
189   0.069211f,  -0.526693f, 0.209850f,  -0.481466f, -0.468302f, -0.100407f,
190   0.241018f,  -1.037781f, 0.038539f,  -2.113840f, -0.974895f, 0.163187f,
191   0.425132f,  -0.772546f, -1.261254f, -0.217488f, -0.971748f, -0.805640f,
192   -0.745175f, -0.177077f, 0.217658f,  0.381431f,  -0.052338f, 0.087176f,
193   -0.165972f, 0.085937f,  0.472564f,  -0.796627f, -2.453307f, 0.569664f,
194   -0.233010f, -0.192134f, 0.064339f,  -0.111411f, -0.262469f, -0.410022f,
195   0.519993f,  -0.684620f, 0.393460f,  -0.277753f, -0.153624f, 0.528984f,
196   -0.415558f, -0.445863f, 0.588512f,  -0.142439f, -0.132127f, 0.199776f,
197   -0.579284f, 0.119488f,  -0.033590f, -0.503846f, -0.674979f, 0.335125f,
198   0.020519f,  0.233973f,  -0.297998f, -0.051511f, 0.518626f,  -0.412782f,
199   -0.074045f, 0.130523f,  0.465751f,  -0.117795f, 2.535813f,  0.352108f,
200   -0.499228f, 0.379784f,  0.056699f,  0.173142f,  -0.076519f, -0.026666f,
201   0.017834f,  0.492333f,  0.093364f,  0.037867f,  -0.165420f, -0.356429f,
202   -0.562334f, 0.057656f,  -0.307544f, 0.085857f,  -0.559851f, 0.107230f,
203   -0.398633f, 0.152618f,  -0.216835f, -0.024539f, 0.026044f,  -0.249519f,
204   -0.563594f, -0.746025f, 0.025265f,  -0.298888f, -0.185243f, 0.058794f,
205   0.233696f,  -0.115223f, 0.144617f,  -0.864390f, 0.619944f,  -0.023980f,
206   0.019481f,  0.225252f,  0.416552f,  -0.115993f, 0.935387f,  0.744386f,
207   0.053353f,  -0.052582f, -0.065650f, 0.228488f,  -0.032042f, -0.371252f,
208   -0.003638f, -0.736984f, -0.203776f, 0.030922f,  -0.065577f, -0.031643f,
209   -0.049253f, -0.054640f, 0.787134f,  0.545414f,  -0.140297f, -0.124274f,
210   -0.110011f, -0.029552f, 0.657005f,  0.214973f,  -0.374300f, 0.251642f,
211   0.276591f,  0.030566f,  -0.145470f, 0.350579f,  -0.356436f, -0.052694f,
212   -0.063966f, -0.751008f, -1.042392f, 0.328892f,  -0.425058f, -0.421571f,
213   -0.571889f, -1.141472f, -0.125216f, 0.212713f,  -0.485170f, -0.088791f,
214   0.124589f,  0.023237f,  0.077635f,  0.020901f,  -0.271402f, -0.321424f,
215   -0.513946f, -0.867872f, -0.284593f, 0.106276f,  0.220192f,  -0.143532f,
216   -0.014648f, 0.073402f,  0.327256f,  -0.139803f, 0.168763f,  0.048199f,
217   -0.122526f, 0.111713f,  -0.134257f, 0.810364f,  -0.085222f, -0.259221f,
218   -0.239349f, 0.044448f,  0.205031f,  0.413113f,  -0.107720f, -0.018816f,
219   -0.247741f, -0.004963f, 0.041170f,  -0.158019f, 0.134839f,  0.129502f,
220   0.800488f,  -1.041584f, -0.129336f, 0.170834f,  0.566586f,  -0.230443f,
221   0.437937f,  -0.149922f, -0.046665f, -0.094646f, 0.200070f,  0.072943f,
222   -0.076943f, -0.084971f, -0.515843f, -0.146720f, 0.472869f,  -0.444731f,
223   -0.100877f, 0.545196f,  -1.786626f, -0.482946f, 0.500509f,  -0.843257f,
224   0.200374f,  0.045103f,  -0.575718f, -0.164335f, -0.232522f, -0.021825f,
225   -0.139490f, 0.356058f,  -0.352075f, 0.061751f,  -0.200616f, -1.180921f,
226   -0.181355f, -0.137459f, 0.247574f,  0.181541f,  0.184314f,  -0.961482f,
227   0.493615f,  0.910261f,  -2.279238f, 0.648631f,  -0.055526f, -0.037137f,
228   0.038643f,  0.136609f,  -0.819373f, -0.040840f, -0.265989f, 0.006877f,
229   0.454651f,  -0.595323f, -0.099500f, -0.263717f, 0.150456f,  0.245077f,
230   -0.268666f, 0.162232f,  -0.516451f, -0.024501f, 0.188046f,  -0.002262f,
231   0.261319f,  0.004173f,  0.746982f,  0.174761f,  0.470447f,  -0.159558f,
232   -0.385240f, 0.023084f,  -0.133520f, -0.220607f, -0.018731f, -0.373558f,
233   -0.707763f, -1.850150f, -0.807404f, -0.168063f, -0.071435f, -0.160740f,
234   -0.478789f, -1.070674f, -0.489740f, -0.255796f, 0.100486f,  -0.153361f,
235   0.334394f,  -0.569472f, -0.198118f, 0.255922f,  0.104717f,  -0.065179f,
236   0.111879f,  -0.447237f, 1.373623f,  -0.190191f, -0.063311f, 0.337529f,
237   -0.138800f, 0.057009f,  -0.137006f, 0.641378f,  0.883147f,  -0.679655f,
238   0.267717f,  -0.351602f, -0.135225f, 0.229398f,  -0.513225f, -1.120345f,
239   0.528786f,  -0.051081f, 0.086653f,  0.140141f,  -0.563969f, 0.333402f,
240   -0.174745f, 0.321093f,  -0.438641f, -0.005131f, 0.247415f,  0.110120f,
241   -0.076308f, -0.083244f, 0.838944f,  -0.113043f, -0.013258f, -0.175028f,
242   -0.179941f, 0.272676f,  -0.047946f, -0.088076f, -0.450031f, 0.053929f,
243   -0.083549f, -0.089952f, -0.186253f, 0.257483f,  0.011019f,  0.586435f,
244   0.060580f,  -0.052078f, 0.090277f,  -0.780869f, 0.969811f,  -0.025349f,
245   -0.281917f, 0.014857f,  0.231863f,  -0.228601f, -0.003861f, 0.226550f,
246   0.141825f,  -0.102171f, -0.010387f, 0.220378f,  -2.561975f, -0.497071f,
247   -0.315117f, 0.371981f,  0.138247f,  0.625031f,  -0.308133f, -0.217876f,
248   0.005615f,  -0.860179f, 0.747491f,  0.006356f,  -0.057024f, -0.483189f,
249   0.055592f,  -0.316834f, 0.069858f,  0.218788f,  -0.200044f, 0.227588f,
250   0.215496f,  -0.055324f, -0.393147f, -0.394062f, -0.253264f, -0.075619f,
251   -0.152512f, -0.332995f, 0.129053f,  0.178668f,  -0.302694f, 0.030678f,
252   0.925896f,  0.964375f,  0.169021f,  -0.218657f, -0.627204f, 0.206437f,
253   -0.521336f, 0.176206f,  0.142733f,  0.139248f,  0.411682f,  0.181544f,
254   0.224850f,  -0.935547f, -0.558208f, 0.348096f,  0.342129f,  -0.389340f,
255   -0.236308f, -0.132099f, 0.073642f,  0.089391f,  -0.306901f, -0.397842f,
256   0.444282f,  0.074623f,  -0.051075f, -0.106617f, -0.184037f, -0.239046f,
257   -0.138761f, 0.120794f,  -0.647577f, -0.336471f, 0.527899f,  -0.164234f,
258   -0.028354f, 1.083678f,  -0.251534f, -0.145903f, -0.182783f, 0.070976f,
259   -0.199590f, -0.400306f, -0.029763f, -0.548042f, -0.266270f, -0.118084f,
260   -1.152632f, 0.383685f,  -0.105895f, -0.096829f, 0.118382f,  0.047447f,
261   -0.019051f, 0.310180f,  -0.162793f, -0.029574f, 0.058054f,  -0.636017f,
262   0.490639f,  0.158347f,  -0.385701f, -0.147057f, 1.285825f,  -1.276083f,
263   -0.021795f, -0.101600f, 0.163254f,  0.267160f,  -2.317864f, -0.098598f,
264   -0.296337f, -0.309017f, 0.164127f,  -0.270012f, -0.071187f, -0.262270f,
265   0.075415f,  -0.368328f, 0.186728f,  -0.158031f, 0.481663f,  0.515950f,
266   -0.162551f, 0.497981f,  0.262196f,  0.168479f,  0.726066f,  -0.243856f,
267   -0.058998f, 0.140168f,  0.053242f,  -0.624623f, -0.249480f, 0.055197f,
268   -1.376804f, 0.417571f,  0.203784f,  0.174370f,  -0.155531f, -0.029400f,
269   -0.491473f, 0.079811f,  -0.080123f, 1.345900f,  0.637077f,  0.434862f,
270   -1.787438f, 0.005756f,  -0.362706f, 0.179458f,  -0.288263f, 0.516788f,
271   -0.921248f, 0.043794f,  -0.137729f, -0.196171f, -0.046295f, -0.793781f,
272   -0.156532f, -0.132566f, 0.517989f,  -0.154321f, -0.054174f, -0.077900f,
273   -0.373316f, -0.117718f, 0.188986f,  -0.476188f, -0.245312f, 0.181439f,
274   -0.161024f, -0.229059f, -3.079907f, -0.225452f, -0.594355f, -0.558027f,
275   -0.135429f, 0.125766f,  -0.081314f, -0.350894f, -0.163165f, -1.936507f,
276   -0.205966f, 0.031472f,  0.744446f,  -0.006680f, -0.837551f, 0.605862f,
277   -0.854929f, -1.543750f, -0.307704f, -0.240517f, 0.178240f,  -0.183586f,
278   -0.010307f, 0.099373f,  -0.228278f, 0.175236f,  -0.000133f, 0.104491f,
279   -1.540545f, -0.570971f, -0.252885f, 0.483036f,  0.052531f,  0.260214f,
280   -0.515016f, -0.602081f, -0.485690f, -0.730710f, 0.163719f,  -1.775975f,
281   -0.298634f, 0.323626f,  -0.373579f, -0.872977f, 0.619574f,  0.026862f,
282   -0.122531f, -0.084698f, -2.436297f, 0.483996f,  -0.203640f, -0.302157f,
283   -0.150666f, -0.238320f, 0.089250f,  0.236485f,  -0.668654f, -0.122863f,
284   0.491152f,  -0.226444f, -0.181248f, 0.120158f,  0.294027f,  0.250056f,
285   0.307601f,  0.357875f,  -1.746455f, -0.175670f, 0.385447f,  -0.108808f,
286   -0.090235f, -0.642504f, -0.486004f, -0.055160f, -0.068692f, 0.009736f,
287   0.607555f,  -0.489426f, 0.150624f,  0.598114f,  -0.128816f, -0.445793f,
288   -0.066524f, -0.254380f, 0.227106f,  -0.406495f, -0.121632f, -0.275960f,
289   -0.136494f, 0.339457f,  -1.318132f, -0.417572f, -2.614077f, 0.324603f,
290   -0.001211f, 0.375192f,  -0.473448f, -0.162510f, 0.099329f,  -0.277965f,
291   0.101221f,  -0.060263f, 0.121867f,  -1.042140f, 0.440851f,  0.078898f,
292   -0.209007f, -0.243699f, 0.715197f,  -0.093997f, 0.086022f,  -0.178203f,
293   -2.275496f, -0.098413f, 0.199352f,  -0.526791f, -0.162086f, -0.197806f,
294   -0.231657f, -0.269202f, -0.794294f, -0.223461f, 0.503584f,  0.416236f,
295   0.064082f,  0.197655f,  0.340871f,  -0.186645f, -0.291498f, 0.433938f,
296   -1.110063f, 0.003751f,  0.392738f,  0.069360f,  0.102088f,  -0.302128f,
297   -1.518457f, 0.106939f,  0.404527f,  -0.306868f, -0.286928f, 0.729276f,
298   -0.531710f, 0.745048f,  -0.168837f, -1.953886f, -0.258828f, -0.190252f,
299   0.241877f,  -0.916744f, -0.030326f, -0.070541f, -0.271037f, 0.211303f,
300   -0.489957f, 0.100850f,  0.323999f,  -0.802837f, -0.462408f, -0.079350f,
301   -0.029374f, 0.131213f,  -0.825032f, 0.040202f,  0.351821f,  0.002869f,
302   -0.132516f, -0.471264f, -0.297002f, 0.263913f,  0.033478f,  0.146161f,
303   0.533229f,  -0.228608f, -0.200639f, -0.170955f, -0.915037f, 0.724491f,
304   0.005151f,  0.018584f,  -0.029771f, -0.396038f, -0.159236f, 0.038691f,
305   -1.197056f, 0.146302f,  0.226840f,  -0.852126f, 0.031214f,  0.108880f,
306   0.562000f,  -0.134633f, -0.713343f, -0.342252f, -1.764521f, -0.114653f,
307   0.515073f,  -0.080515f, -0.121155f, -0.865139f, -0.833694f, -0.368553f,
308   0.347673f,  0.623379f,  0.722067f,  -0.492458f, -0.513263f, 0.585167f,
309   0.721518f,  -0.693499f, 0.343725f,  -0.273861f, -0.040230f, -0.785664f,
310   -0.157500f, -0.308445f, 0.054062f,  0.600131f,  -0.860887f, 0.434470f,
311   -0.191382f, -0.306150f, -0.243965f, 0.705444f,  0.007789f,  -0.146154f,
312   -0.054499f, -0.073500f, -1.067364f, 0.404936f,  -2.864590f, 0.182323f,
313   0.326126f,  0.102405f,  -0.135800f, 1.128095f,  -0.012267f, -0.023996f,
314   -0.264834f, -0.108967f, -1.176746f, -0.926666f, 0.082999f,  -0.498361f,
315   0.083560f,  -0.210074f, 0.019225f,  -0.201614f, -0.904760f, 0.181421f,
316   0.586384f,  -0.177706f, 0.065471f,  0.168552f,  0.054705f,  0.045241f,
317   0.048057f,  -0.410957f, -2.188854f, -0.169812f, 0.015521f,  0.176856f,
318   -0.179331f, -0.352640f, -0.491735f, -1.743206f, 0.044227f,  0.010454f,
319   0.823643f,  -0.119781f, -0.098359f, 0.093119f,
320 };
321 
322 static const float av1_ab_partition_nn_bias_128_layer1[LABEL_SIZE] = {
323   -0.433195f, -0.120488f, -0.116721f, 0.112134f,  0.118170f, -0.259769f,
324   -0.077530f, 0.394044f,  0.279167f,  -0.317988f, 0.189538f, 0.314776f,
325   0.325655f,  -0.107123f, 0.591049f,  0.358744f,
326 };
327 
328 static const NN_CONFIG av1_ab_partition_nnconfig_128 = {
329   FEATURE_SIZE,  // num_inputs
330   LABEL_SIZE,    // num_outputs
331   1,             // num_hidden_layers
332   {
333       64,  // num_hidden_nodes
334   },
335   {
336       av1_ab_partition_nn_weights_128_layer0,
337       av1_ab_partition_nn_weights_128_layer1,
338   },
339   {
340       av1_ab_partition_nn_bias_128_layer0,
341       av1_ab_partition_nn_bias_128_layer1,
342   },
343 };
344 
345 // nn model for ab partition pruning, 64x64.
346 static const float av1_ab_partition_nn_weights_64_layer0[FEATURE_SIZE * 64] = {
347   -0.495347f, -0.049498f, -0.026804f, 0.030474f,  -0.289308f, -0.264193f,
348   -0.141121f, -0.072562f, -0.391665f, -0.051491f, -0.234761f, 0.027155f,
349   -0.038217f, 0.014872f,  -0.289728f, -0.233577f, -0.415875f, -0.343615f,
350   -0.442543f, -0.482492f, 0.073510f,  0.007503f,  2.162329f,  -0.362849f,
351   2.145915f,  -0.883135f, 0.185636f,  -0.062859f, -0.465574f, -0.486205f,
352   -0.056710f, -0.330642f, -0.321860f, 0.042321f,  -0.348965f, 0.003542f,
353   -0.291365f, -0.078164f, -0.345093f, -0.220272f, -0.471270f, -0.763853f,
354   0.246622f,  0.199651f,  -0.663420f, -0.154152f, -1.220383f, 0.047138f,
355   0.816811f,  0.083247f,  -0.218839f, 0.038143f,  -0.063436f, 0.015517f,
356   -0.307320f, -0.166956f, -0.169499f, -0.399005f, -0.234638f, -0.162266f,
357   0.050425f,  -0.221723f, -0.256942f, -0.287285f, 0.144011f,  -0.033245f,
358   0.083649f,  0.119428f,  -0.056706f, -0.117805f, 0.021866f,  -0.257300f,
359   -0.201378f, -0.217484f, -0.413780f, -0.145793f, 0.082792f,  -0.347247f,
360   0.042539f,  -0.302697f, 1.652316f,  0.000701f,  -0.482843f, -0.160332f,
361   -0.450099f, 0.212399f,  -4.715360f, -5.336774f, -5.375758f, -6.048339f,
362   0.085956f,  -0.037767f, 1.052409f,  -0.931924f, -2.221907f, 0.268946f,
363   0.015512f,  1.237094f,  -1.092185f, 0.418247f,  -0.082143f, -0.076914f,
364   -0.060749f, -0.325440f, -0.296960f, -0.066815f, -0.158477f, -0.373945f,
365   -0.122322f, -0.113495f, -0.097978f, -0.192816f, -0.270418f, 0.035840f,
366   -0.015458f, -0.121071f, -0.279582f, -0.067683f, 0.097855f,  0.019839f,
367   0.451127f,  0.004376f,  1.410392f,  3.255835f,  -0.344815f, 0.145202f,
368   0.204132f,  0.171948f,  -0.527736f, -0.110353f, 0.901448f,  0.003238f,
369   -3.822090f, 0.235462f,  1.024823f,  -0.821244f, 0.876056f,  2.553762f,
370   -3.478597f, -2.076582f, -0.265515f, -0.055923f, -0.156980f, -0.164097f,
371   -0.246040f, 0.039430f,  -0.071769f, -0.118847f, -0.304053f, -0.281541f,
372   -0.226021f, -0.263091f, -0.127359f, -0.249410f, -0.051023f, 0.083911f,
373   0.084721f,  0.168089f,  -0.272169f, -0.204998f, -0.008303f, -0.173998f,
374   0.079376f,  -0.197426f, -0.199052f, -0.118794f, -0.063753f, -0.094769f,
375   0.066176f,  -0.175832f, -0.238752f, -0.287960f, -0.134307f, -0.185953f,
376   -0.385845f, 0.119769f,  -0.006567f, -0.382126f, -0.214221f, 0.038449f,
377   -0.253484f, -0.282766f, -0.020249f, -0.193929f, 0.016281f,  -0.114423f,
378   -0.145940f, -0.281621f, -0.007588f, -0.131470f, -0.189012f, -0.185699f,
379   -0.279011f, -0.008132f, 0.208463f,  0.020569f,  -0.206803f, -0.213408f,
380   -0.206131f, -0.290245f, 0.069701f,  -0.000371f, -0.307572f, -0.451785f,
381   -0.300838f, -0.453186f, -0.301691f, 0.046327f,  -0.312668f, 0.058272f,
382   -0.303131f, -0.376252f, 0.108384f,  -0.086623f, -0.100630f, -0.027330f,
383   -0.003969f, 0.089502f,  -0.200722f, -0.107889f, 0.061843f,  -0.008478f,
384   -0.265057f, -0.271132f, -0.073562f, 0.129337f,  -0.283698f, -0.353414f,
385   0.076420f,  -0.244280f, -0.119537f, -0.105366f, -0.184692f, -0.038817f,
386   -0.478507f, -0.118808f, -0.472979f, -0.305884f, -0.462813f, -0.189581f,
387   -0.011932f, -0.585700f, 0.253212f,  -1.061900f, -0.205116f, -0.336407f,
388   -0.762199f, 0.577737f,  0.230832f,  0.434440f,  -0.096713f, 0.038552f,
389   -0.147800f, -0.213553f, 0.041740f,  -0.281907f, -0.026154f, -0.082356f,
390   -0.331871f, -0.408247f, -0.129022f, -0.037550f, -0.310233f, -0.320883f,
391   -0.391963f, -0.467392f, 0.027453f,  -0.394761f, -0.045544f, 0.076052f,
392   0.483985f,  0.067093f,  0.141361f,  0.576772f,  0.859718f,  2.566515f,
393   -0.025476f, 0.769738f,  -0.680235f, -1.683309f, -2.394131f, -0.000714f,
394   -0.615021f, -0.195856f, -0.434035f, -0.295010f, -0.668659f, -0.245959f,
395   0.551148f,  1.777227f,  -0.461630f, 0.043093f,  0.012293f,  -0.255841f,
396   -0.097070f, -0.371156f, -0.146323f, -0.015508f, -0.103873f, -0.087476f,
397   -0.297266f, -0.128699f, -0.149555f, 0.016534f,  -0.375498f, -0.346759f,
398   -0.455156f, -0.147509f, -0.427076f, -0.354431f, -0.158025f, -0.164604f,
399   -0.237038f, -0.010314f, -0.092884f, -0.397084f, -0.217980f, -0.127184f,
400   -0.048421f, -0.144133f, 0.889073f,  0.012606f,  3.007608f,  -0.602584f,
401   -1.849480f, -0.373159f, -1.890695f, -3.609938f, 0.811923f,  -1.867208f,
402   -0.244326f, -0.018012f, -0.211192f, -0.220196f, 0.169363f,  0.119141f,
403   -0.230715f, 0.083247f,  0.020367f,  -0.128629f, -0.217455f, -0.159640f,
404   1.815952f,  -0.369238f, -1.186447f, -0.658753f, -0.511026f, -0.096934f,
405   0.662971f,  0.486475f,  0.159746f,  -0.018932f, 3.692397f,  1.384353f,
406   -0.401984f, -0.248380f, -0.140861f, 0.215248f,  -0.023711f, 0.059679f,
407   -0.072260f, 0.004271f,  0.039545f,  -0.347971f, -0.081851f, -0.474896f,
408   -0.181572f, 0.066736f,  -0.157822f, -0.163760f, -0.171113f, -0.089935f,
409   -0.338281f, -0.421444f, -0.306687f, -0.085283f, -0.377953f, -0.138750f,
410   -0.102701f, -0.312336f, 0.149831f,  0.007229f,  -0.155700f, -0.173611f,
411   4.074261f,  1.342306f,  -1.272712f, 1.570899f,  -0.545093f, -0.317605f,
412   -0.189440f, -0.133910f, -0.273190f, -0.108020f, -0.166107f, 0.021413f,
413   -0.239130f, -0.067211f, 0.041957f,  -0.039234f, -1.003587f, -0.094412f,
414   0.532512f,  -0.870538f, -1.118023f, -1.160983f, -0.736307f, -0.418752f,
415   0.419466f,  0.492122f,  -0.004368f, -0.022096f, -1.115132f, 0.150886f,
416   2.396852f,  2.660000f,  -0.376537f, 0.468628f,  0.149413f,  -0.074898f,
417   -0.067154f, 0.021245f,  0.127857f,  0.294189f,  0.508056f,  0.390232f,
418   -3.899177f, -3.414681f, -3.929195f, -4.160545f, -0.274323f, -0.052583f,
419   -0.003545f, -0.433084f, -0.404891f, -0.145051f, -0.312367f, 0.004579f,
420   -0.398724f, -0.372068f, -0.234279f, 0.017799f,  -0.424760f, -0.646717f,
421   -0.047568f, 2.924664f,  -0.644165f, 0.359349f,  -0.294800f, 0.591746f,
422   -0.404710f, -0.092358f, -0.250729f, 0.030829f,  -0.147149f, -0.476023f,
423   -0.071803f, -0.482516f, -0.293117f, -0.215923f, -0.373122f, -0.085315f,
424   -0.377052f, -0.449899f, -0.056452f, 0.138081f,  -0.085350f, -0.308391f,
425   0.106661f,  0.176234f,  0.258869f,  -0.230172f, -0.233029f, -0.241208f,
426   -0.067509f, -0.223172f, -0.118353f, -0.302478f, -0.579632f, -0.561326f,
427   -0.158114f, -0.223167f, -0.026689f, 0.051863f,  0.212834f,  -0.304714f,
428   -0.169071f, -0.193695f, -0.075682f, -0.170860f, -0.241008f, -0.044648f,
429   0.280815f,  -0.002585f, -0.283552f, -0.037701f, -0.681169f, -0.274535f,
430   -0.380595f, 0.109504f,  -0.111141f, -0.437685f, -0.094459f, 0.144206f,
431   -0.106139f, -0.211832f, -0.054742f, -0.172813f, -0.295905f, -0.071907f,
432   -0.418429f, -0.183240f, 0.031319f,  -0.095785f, -0.315447f, 0.069404f,
433   -0.422910f, -0.029867f, -0.357321f, -0.199976f, -0.337707f, -0.070188f,
434   -0.178198f, 0.177208f,  0.134688f,  -0.081933f, -0.229452f, -0.208872f,
435   0.026287f,  -0.364040f, -0.063696f, -0.227443f, -0.234401f, -0.205699f,
436   -0.267238f, -0.494125f, -0.056255f, 0.053715f,  -0.487754f, 0.014818f,
437   0.087383f,  -0.077556f, -0.168085f, -0.436851f, -0.276286f, -0.137845f,
438   -0.107606f, -0.103653f, -0.233766f, -0.419083f, 0.169185f,  0.010186f,
439   -0.001587f, 0.086735f,  -2.465718f, 1.482185f,  1.621193f,  -2.081680f,
440   1.386553f,  -3.204335f, -0.267111f, -0.004508f, 0.164712f,  0.274147f,
441   1.724306f,  -2.273659f, 0.749574f,  -0.891905f, 0.105965f,  -0.030428f,
442   -0.416018f, -0.300762f, 0.122911f,  -0.316908f, -0.292504f, 0.138666f,
443   -0.161327f, -0.042143f, -0.249128f, 0.149210f,  -0.088987f, -0.654101f,
444   -1.501843f, 0.216777f,  0.955914f,  0.524158f,  -1.642561f, -1.643626f,
445   0.864797f,  -0.425451f, -2.115764f, -0.012502f, 0.065172f,  1.297270f,
446   0.018845f,  1.167276f,  -0.470970f, -0.244995f, 0.374782f,  -1.811056f,
447   -0.055430f, -0.024102f, -0.376519f, -0.339640f, -0.119177f, -0.277995f,
448   -0.290095f, -0.081362f, -0.144139f, -0.118037f, -0.180357f, -0.217559f,
449   -0.370683f, 0.172816f,  -0.265069f, 0.194321f,  -0.273478f, 0.037442f,
450   -0.235552f, -0.078625f, -0.447541f, 0.016836f,  -0.271123f, -0.171481f,
451   -0.321477f, -0.184826f, -0.442981f, -0.227273f, -0.370666f, -0.237232f,
452   -0.257493f, -0.225714f, -0.153716f, -0.283487f, -0.155399f, 0.067697f,
453   0.230343f,  -0.034318f, -0.022687f, -0.047090f,
454 };
455 
456 static const float av1_ab_partition_nn_bias_64_layer0[64] = {
457   -0.212182f, -0.233725f, -0.758846f, -0.158162f, 0.614743f,  -0.150944f,
458   -0.075727f, -0.208414f, 1.054996f,  0.713758f,  -0.300051f, -0.151482f,
459   -2.443570f, 0.430590f,  -0.129001f, -0.160733f, -0.230547f, -0.143228f,
460   -0.140577f, -0.086812f, -0.212298f, -0.159557f, -0.055647f, -0.211423f,
461   0.578161f,  -0.220318f, -0.210107f, -3.111584f, 0.604419f,  -0.232622f,
462   -0.209924f, -0.130794f, -0.084097f, -0.036005f, 0.294594f,  -2.535531f,
463   -0.209783f, -0.211189f, -2.766337f, 0.000000f,  0.450177f,  -1.754884f,
464   3.262664f,  -0.209691f, -0.614886f, -0.211257f, -0.109096f, -0.190492f,
465   -0.109007f, -0.026910f, -0.136035f, -0.212321f, -0.139320f, -0.212233f,
466   -0.305430f, 0.739171f,  0.991277f,  -0.088150f, 0.086313f,  -0.023379f,
467   -0.125366f, -0.063576f, -0.212169f, -0.047463f,
468 };
469 
470 static const float av1_ab_partition_nn_weights_64_layer1[64 * LABEL_SIZE] = {
471   -0.036800f, 0.528721f,  0.490767f,   0.144409f,  1.103640f,  0.361910f,
472   -0.180069f, 0.068033f,  -14.868382f, 0.359013f,  0.322567f,  -0.199212f,
473   0.906164f,  -0.488254f, 0.149653f,   -0.216394f, -0.099347f, 0.004936f,
474   -0.111391f, 0.074848f,  -0.041709f,  0.147627f,  -0.018905f, 0.096116f,
475   0.184817f,  -0.016241f, 0.115739f,   2.376754f,  0.637097f,  0.052954f,
476   0.136428f,  0.225267f,  -0.181873f,  -0.142876f, 0.684048f,  0.658791f,
477   0.105795f,  0.241705f,  1.381114f,   -0.209379f, 1.145949f,  0.795293f,
478   -9.361877f, 0.198302f,  0.539600f,   0.092317f,  -0.081695f, 0.200777f,
479   0.102334f,  0.081583f,  0.060948f,   -0.025110f, 0.160951f,  -0.020170f,
480   0.234006f,  -0.029369f, 0.375036f,   0.270209f,  -0.556529f, 1.402949f,
481   0.101777f,  -0.027331f, 0.004502f,   -0.153166f, -0.116651f, 0.151573f,
482   -0.022187f, 0.144044f,  -0.108719f,  -0.129942f, -0.270321f, 0.227363f,
483   1.892330f,  -0.661052f, -0.219398f,  -0.229417f, -0.856438f, -1.196988f,
484   -0.081774f, 0.078847f,  -0.207057f,  -0.048947f, 0.152073f,  -0.243056f,
485   -0.233329f, -0.288689f, -0.158333f,  -0.141177f, -0.715436f, 0.016947f,
486   -0.093752f, 0.204984f,  -1.209782f,  0.155683f,  0.092239f,  0.146495f,
487   0.813146f,  -0.027757f, 0.330982f,   2.173948f,  -0.028867f, -0.141815f,
488   0.292708f,  -0.204794f, 0.014496f,   1.032799f,  1.312155f,  0.107020f,
489   0.824752f,  -0.013945f, 0.184829f,   -0.041633f, 0.215300f,  -0.476088f,
490   -0.053213f, 0.126862f,  -0.020777f,  0.082893f,  -0.223727f, -0.923063f,
491   0.466529f,  0.082140f,  -0.845758f,  -1.140791f, -0.262033f, 0.138491f,
492   0.151717f,  -0.182479f, -0.131128f,  0.055411f,  0.106771f,  0.125552f,
493   0.297184f,  -0.257403f, -0.059884f,  -0.274903f, 2.694357f,  -0.108244f,
494   0.025377f,  0.043092f,  -0.558317f,  3.517159f,  -0.270833f, -0.240676f,
495   0.205100f,  -0.057068f, -0.140445f,  -0.193449f, -0.030061f, -0.286762f,
496   -0.467523f, -0.012647f, 0.190564f,   0.022394f,  -0.101479f, 0.339684f,
497   -0.902743f, -0.169578f, -0.178029f,  -0.041836f, -3.952108f, -0.028298f,
498   -0.221137f, -0.733895f, -0.223895f,  0.039012f,  0.687867f,  0.021423f,
499   0.113063f,  0.676087f,  -0.961000f,  -0.064847f, 0.712856f,  -0.192765f,
500   -0.001132f, 0.016689f,  -0.236020f,  -0.766186f, -0.175729f, 0.012879f,
501   -0.251064f, -0.105523f, -0.039212f,  -0.347584f, 0.304352f,  -0.034174f,
502   -0.364258f, -0.685252f, -0.266115f,  -0.247345f, -0.155905f, 0.152283f,
503   -0.156315f, 0.174082f,  -0.757654f,  0.102303f,  -2.192316f, -0.245815f,
504   0.119882f,  -0.086542f, 1.987246f,   -1.353163f, -0.374813f, -0.233504f,
505   -1.980895f, 0.692093f,  -0.168351f,  0.172700f,  -0.009052f, -0.015734f,
506   0.106679f,  -0.060472f, -0.256813f,  -0.074874f, -0.207488f, -0.329515f,
507   -0.418268f, -0.017940f, -0.036081f,  0.064719f,  -1.488016f, 0.020591f,
508   -0.176325f, -0.141074f, 0.944494f,   0.150237f,  -0.249805f, -0.277280f,
509   0.012686f,  0.132483f,  0.116123f,   0.013737f,  -0.116091f, 0.750340f,
510   3.251343f,  -0.188864f, 1.096992f,   0.058467f,  -0.041433f, -0.037937f,
511   -0.133294f, -0.137908f, -0.171132f,  0.106362f,  0.069383f,  -0.052662f,
512   -0.177883f, -0.408049f, 0.680221f,   -0.117035f, -0.904240f, -1.395228f,
513   0.154527f,  0.134427f,  0.022767f,   -0.158886f, -0.230316f, 0.161096f,
514   0.362213f,  -0.235060f, -0.941620f,  0.055912f,  -0.049458f, -0.166632f,
515   0.481418f,  0.930146f,  0.041108f,   0.033674f,  1.372066f,  -1.847709f,
516   0.003324f,  0.259534f,  0.177014f,   -0.202761f, -0.262017f, -0.190852f,
517   -0.102839f, 0.028338f,  0.187193f,   -0.041684f, 0.123973f,  -0.198576f,
518   -0.110369f, -1.431400f, 0.208369f,   -0.302370f, -0.248549f, 0.062985f,
519   0.673409f,  0.036662f,  -0.711340f,  -0.120584f, -0.189789f, 0.098812f,
520   2.947819f,  0.216567f,  -0.414472f,  -0.181742f, 1.873779f,  -0.222726f,
521   -0.782870f, 0.007889f,  0.015062f,   -0.554328f, 0.182928f,  -0.191430f,
522   0.123636f,  -0.215460f, -0.225245f,  0.251516f,  -0.013025f, -1.359595f,
523   -0.750602f, 0.342667f,  -0.141899f,  -0.687493f, -0.072639f, 0.048018f,
524   -0.242107f, -0.031917f, -0.287472f,  -0.046088f, 0.832197f,  -0.016576f,
525   -1.553349f, -0.216341f, 0.023077f,   -0.410867f, 4.243743f,  -0.514878f,
526   -0.066007f, -0.160696f, -0.262678f,  -0.648790f, -0.430586f, 0.199940f,
527   -0.202496f, -0.222241f, -0.016406f,  -0.121473f, 0.000828f,  -0.081584f,
528   -0.152641f, -0.190166f, 0.644400f,   0.040196f,  -0.302104f, -1.143654f,
529   -0.160327f, -0.320780f, -0.187006f,  0.037311f,  0.440618f,  -0.070733f,
530   -0.117785f, 1.527539f,  -0.419310f,  0.001300f,  1.389956f,  -0.036366f,
531   -0.269203f, 0.612265f,  2.721897f,   -0.086836f, -0.446999f, 0.012525f,
532   -0.078317f, -0.287052f, -0.111188f,  -0.085181f, -0.164667f, -0.010466f,
533   -0.569722f, -0.018888f, -0.101663f,  -1.147130f, -0.465204f, 0.114524f,
534   -2.192402f, -0.221325f, 0.375748f,   0.206284f,  -0.261548f, -0.246257f,
535   -0.143004f, -0.069981f, -0.057306f,  -0.116481f, -0.435903f, -0.314970f,
536   0.013210f,  -0.010175f, 4.630571f,   -0.473226f, -0.197199f, -0.028204f,
537   0.122907f,  2.475548f,  0.025011f,   -0.092603f, -0.127561f, -0.151330f,
538   -0.077295f, 0.245016f,  -0.045005f,  0.183396f,  -0.330556f, -0.384887f,
539   0.356374f,  -0.016618f, -0.463353f,  -1.291546f, -0.071986f, -0.311599f,
540   0.072385f,  -0.430786f, -2.094788f,  0.202733f,  -0.910109f, -1.336543f,
541   -0.086800f, -0.096413f, 1.544383f,   0.031860f,  -0.796211f, 0.762786f,
542   3.250022f,  -0.441798f, -0.698537f,  0.062839f,  0.033525f,  -0.362996f,
543   0.027022f,  -1.131264f, -0.228926f,  0.053885f,  -0.338628f, 0.155037f,
544   -0.046844f, -0.888172f, -0.241767f,  0.084965f,  -0.617743f, -0.049896f,
545   -0.036894f, -0.304783f, -0.002639f,  0.137957f,  0.052121f,  -0.131161f,
546   -0.117200f, -0.253380f, -0.205561f,  -0.302450f, -0.047397f, -0.330518f,
547   3.613420f,  -1.525951f, -0.026738f,  0.209150f,  -2.103534f, 2.019689f,
548   -0.366199f, -0.095260f, 0.027417f,   -0.242512f, 0.162579f,  0.052113f,
549   -0.293851f, -0.068138f, -0.005799f,  -0.344696f, -0.114824f, -0.431107f,
550   -0.120058f, -1.139926f, -1.048379f,  0.036446f,  -0.323020f, -0.432945f,
551   0.454151f,  -0.140058f, 0.050649f,   -0.094900f, -0.017278f, -0.238719f,
552   1.193153f,  0.120447f,  -0.496061f,  0.917431f,  2.936126f,  -0.115521f,
553   -0.347397f, -0.435325f, -0.004383f,  -0.211864f, 0.162383f,  -1.040726f,
554   0.089537f,  -0.128579f, -0.133505f,  0.107129f,  -0.435657f, -0.180388f,
555   0.043650f,  0.018709f,  -0.773242f,  -0.687192f, -0.120633f, -0.063626f,
556   0.029912f,  0.113972f,  -0.403502f,  -0.127640f, -0.269625f, 0.129794f,
557   -0.188539f, 0.041641f,  0.029769f,   -0.198374f, 1.401407f,  0.353887f,
558   -0.219925f, 0.260515f,  1.157034f,   -2.992044f, -0.097618f, -0.064417f,
559   -0.203626f, -0.008217f, -0.112339f,  -0.227407f, -0.155118f, 0.247705f,
560   -0.012304f, -0.248447f, -0.913463f,  -0.064788f, -0.214619f, -0.251761f,
561   -0.386861f, -0.040574f, -0.163219f,  -0.100700f, 1.488274f,  -0.071684f,
562   -0.033626f, -0.006497f, -0.246945f,  -0.145221f, -3.747390f, 0.149609f,
563   -0.263326f, -0.297385f, -1.039896f,  -0.083174f, -0.025473f, -0.235586f,
564   -0.001087f, 0.254286f,  0.265106f,   0.007325f,  0.199239f,  0.134103f,
565   -0.578211f, -0.259801f, -0.062373f,  2.368348f,  0.560556f,  -0.252260f,
566   0.889997f,  -0.447872f, -0.059218f,  -0.095315f, -0.061667f, 0.183580f,
567   -0.157479f, 0.055387f,  -0.831734f,  0.007606f,  -1.104906f, 0.301180f,
568   -0.117115f, 0.212959f,  4.727223f,   -0.243833f, -0.397495f, -0.025021f,
569   -0.367587f, -2.082058f, -0.217699f,  0.148111f,  0.252430f,  0.111088f,
570   -0.260692f, 0.095124f,  -0.407774f,  -0.322169f, 0.002927f,  0.126169f,
571   -1.272325f, -0.279772f, -0.373680f,  -0.485177f, -0.605458f, 0.021225f,
572   -0.092031f, -0.226585f, 1.895162f,   0.037866f,  -0.275475f, 1.614360f,
573   -0.014972f, -0.277679f, -3.449082f,  -0.092060f, -0.747873f, 0.020716f,
574   2.776178f,  -0.049963f, 0.183999f,   -0.295259f, -0.028868f, 0.221895f,
575   0.001265f,  0.336823f,  0.219372f,   0.112824f,  0.408132f,  -0.017940f,
576   -0.311666f, 1.489606f,  -0.058093f,  -0.305659f, -0.491933f, -0.143847f,
577   0.166115f,  0.042867f,  -0.123447f,  -0.087099f, -0.305395f, -0.365079f,
578   -0.755801f, -0.160649f, 0.736260f,   -0.008611f, 0.095836f,  -0.017345f,
579   5.697515f,  -0.498971f, -0.125280f,  0.199907f,  0.300053f,  0.605026f,
580   -0.228225f, -0.259523f, 0.016384f,   0.146973f,  0.210258f,  0.226766f,
581   -0.075178f, -0.050924f, 0.188496f,   -0.415266f, -0.484880f, -0.236384f,
582   0.071931f,  -0.331863f, -0.601243f,  -0.232479f, -0.285272f, 0.123789f,
583   -1.341333f, 0.037082f,  -0.315202f,  -1.587215f, -0.271576f, 0.003216f,
584   -4.437186f, -0.256205f, -0.576589f,  -0.114147f, 2.153916f,  -0.369618f,
585   0.271415f,  0.145036f,  -0.158731f,  -0.240938f, -0.187369f, 0.036325f,
586   0.254771f,  0.211488f,  -0.240297f,  0.098417f,  -0.415011f, 2.334793f,
587   -0.127252f, 0.020069f,  -0.168755f,  -0.448922f, -0.219207f, 0.016232f,
588   -0.221935f, -0.269500f, -0.100636f,  0.102545f,  -0.809376f, -0.054979f,
589   0.360713f,  -0.326541f, 0.112933f,   0.138073f,  4.229404f,  -0.763801f,
590   -0.305429f, 0.199955f,  -1.787713f,  0.272866f,  0.109895f,  0.138466f,
591   -0.250259f, -0.167162f, -0.212588f,  -0.217589f, -0.067125f, -0.077490f,
592   -0.208970f, -0.006863f, -0.671146f,  -0.298320f, -0.165509f, 0.044597f,
593   -1.408624f, -0.213957f, -0.220947f,  0.129718f,  1.316777f,  -0.098928f,
594   -0.008121f, -0.558293f, -0.297290f,  -0.218873f, -4.346638f, -0.228174f,
595   -0.204710f, -0.388864f, 2.697919f,   0.025260f,  0.857020f,  0.009921f,
596   0.036915f,  -0.320275f, -0.087937f,  0.022636f,  0.236667f,  0.135496f,
597   -0.059616f, -0.192955f, 0.009470f,   2.139589f,  -0.200449f, 0.129818f,
598   1.017444f,  -0.608299f, 0.257914f,   -0.134306f, -0.033327f, 0.002855f,
599   -0.338598f, 0.015559f,  0.117362f,   -0.166760f, 0.086903f,  -0.167666f,
600   0.193523f,  0.033852f,  -1.147686f,  0.489468f,  -0.006969f, 0.125630f,
601   1.557907f,  -1.604449f, -0.071114f,  0.096178f,  0.007065f,  0.200013f,
602   0.213393f,  0.168466f,  -0.100568f,  -0.117861f, -0.161542f, -0.072561f,
603   -1.069871f, -0.470138f, -0.352578f,  -1.503513f, -0.001394f, -0.380109f,
604   0.065089f,  -0.281668f, 0.988953f,   -0.002778f, -0.659026f, -0.470692f,
605   -0.407292f, 0.011710f,  -1.362085f,  0.184738f,  -0.135786f, -1.374241f,
606   4.487930f,  -0.067274f, -0.956404f,  -0.233995f, 0.224527f,  -0.454556f,
607   0.037900f,  -0.281658f, 0.208224f,   -0.254753f, 0.045740f,  0.051444f,
608   -0.388281f, 0.257112f,  -0.485030f,  -0.082659f, 0.148103f,  -1.007456f,
609   -0.022295f, 0.036984f,  -0.369401f,  -0.076943f, -0.007636f, -0.293022f,
610   0.470466f,  0.199012f,  -2.158182f,  0.036577f,  -0.014725f, -0.229516f,
611   2.236929f,  0.030945f,  -0.400045f,  0.109348f,  0.214691f,  -0.891516f,
612   -0.251379f, -0.217358f, 0.013733f,   0.205573f,  -0.151725f, -0.191782f,
613   -0.339630f, -0.163905f, -0.119191f,  -0.032516f, 0.503015f,  0.025772f,
614   0.029094f,  -1.146153f, 0.216723f,   -0.330023f, 0.064695f,  -0.262521f,
615   0.425612f,  -0.093080f, -0.489648f,  1.051293f,  -0.092332f, 0.095557f,
616   -0.874132f, 0.218483f,  -0.127648f,  -1.605802f, 2.763617f,  -0.186734f,
617   -1.243166f, -0.193514f, -0.173748f,  0.337822f,  0.183873f,  -0.251594f,
618   -0.211582f, 0.144081f,  0.029620f,   -0.024853f, -0.385140f, 0.467341f,
619   -0.928316f, -0.195442f, 0.917783f,   0.357084f,  0.174445f,  -0.073659f,
620   -0.012811f, -0.115420f, -0.181147f,  -0.364449f, -0.567395f, -0.012969f,
621   -1.680714f, 0.065323f,  0.198063f,   -0.244201f, 1.428545f,  -0.432539f,
622   -0.208931f, -0.091205f, 0.957125f,   0.813519f,  -0.262677f, 0.246852f,
623   0.015536f,  0.055026f,  0.067054f,   0.262103f,  -0.358115f, -0.095206f,
624   -0.267522f, -0.402710f, -0.680397f,  -0.123627f, -0.385590f, -1.504680f,
625   -0.169513f, -0.215338f, 0.043633f,   -0.079052f, -0.464410f, 0.122894f,
626   -0.278231f, -2.456445f, -0.159917f,  -0.015597f, -0.735449f, -0.078854f,
627   -0.400290f, -1.153870f, 3.657228f,   -0.287093f, -1.174355f, -0.102001f,
628   -0.288281f, 0.185209f,  -0.145228f,  -0.200449f, -0.099914f, -0.138354f,
629   0.254428f,  -0.161751f, -0.118206f,  0.296043f,  -0.482613f, 0.080932f,
630   1.097605f,  -0.010190f, 0.232439f,   0.447617f,  -0.133508f, 0.115763f,
631   -0.388589f, 0.174695f,  -0.236014f,  0.006284f,  -1.374129f, 0.092015f,
632   -0.241419f, -0.231667f, 2.763950f,   -0.922932f, -0.061605f, 0.208740f,
633   -1.597190f, 1.353325f,  -0.198528f,  0.250498f,  -0.013950f, -0.203861f,
634   -0.254563f, 0.081931f,  -0.413369f,  0.011844f,  0.080961f,  -0.231161f,
635   -1.234909f, -0.440843f, -0.174980f,  -0.315283f, -0.337474f, -0.123243f,
636   -0.310001f, -0.271028f, 0.364179f,   0.022845f,  -0.535517f, -0.772936f,
637   -0.188435f, 0.039667f,  -0.807463f,  0.266550f,  -0.288857f, -1.630789f,
638   1.280155f,  0.065712f,  -0.279960f,  -0.300056f, 0.258440f,  -0.073781f,
639   0.213878f,  0.042196f,  0.021360f,   0.211698f,  -0.003751f, -0.192673f,
640   -0.137008f, 0.247878f,  -0.470604f,  0.073164f,  1.523241f,  0.734755f,
641   -0.114126f, -0.193834f, -0.025759f,  0.263183f,
642 };
643 
644 static const float av1_ab_partition_nn_bias_64_layer1[LABEL_SIZE] = {
645   -0.343508f, -0.706936f, -0.160676f, -0.877101f, -0.517567f, -0.253254f,
646   -0.148074f, 0.923430f,  -0.364770f, 0.203550f,  0.401216f,  0.938246f,
647   -0.872737f, 0.718723f,  0.703398f,  2.560015f,
648 };
649 
650 static const NN_CONFIG av1_ab_partition_nnconfig_64 = {
651   FEATURE_SIZE,  // num_inputs
652   LABEL_SIZE,    // num_outputs
653   1,             // num_hidden_layers
654   {
655       64,  // num_hidden_nodes
656   },
657   {
658       av1_ab_partition_nn_weights_64_layer0,
659       av1_ab_partition_nn_weights_64_layer1,
660   },
661   {
662       av1_ab_partition_nn_bias_64_layer0,
663       av1_ab_partition_nn_bias_64_layer1,
664   },
665 };
666 
667 // nn model for ab partition pruning, 32x32.
668 static const float av1_ab_partition_nn_weights_32_layer0[FEATURE_SIZE * 64] = {
669   -0.323723f, -0.214013f, -0.007772f, -0.458851f, -0.125542f, -0.123860f,
670   -0.410973f, -0.209389f, -0.087580f, -0.272881f, -0.168500f, -1.130845f,
671   0.344916f,  -0.475017f, -0.362262f, -0.195662f, -0.566124f, 0.782163f,
672   0.411575f,  -0.013378f, -0.318650f, -0.124678f, -0.612909f, -0.315788f,
673   -0.263990f, -0.508783f, -0.048938f, -0.416407f, -0.402648f, -0.156644f,
674   0.225887f,  -0.000493f, 2.682241f,  0.871204f,  0.059014f,  0.803542f,
675   -1.407028f, -1.154669f, 1.388148f,  -0.293348f, -0.003669f, -0.009607f,
676   1.330030f,  -0.337841f, 2.118617f,  1.033059f,  -0.084788f, 0.212904f,
677   0.082405f,  -0.070579f, -0.494005f, -0.173392f, 0.039546f,  -0.463865f,
678   0.077163f,  -0.434066f, 0.030835f,  -0.427139f, -0.560520f, -0.031606f,
679   -0.368541f, -0.027458f, 0.370574f,  0.461418f,  1.087682f,  -0.572137f,
680   -1.509596f, -0.765697f, -0.499383f, -0.277998f, -0.106492f, -0.129564f,
681   -0.169133f, -0.269834f, -0.114270f, -0.275431f, 0.016339f,  -0.156744f,
682   -0.267922f, 0.171216f,  0.110556f,  0.002954f,  -0.200327f, -0.187663f,
683   3.691601f,  1.234152f,  0.186315f,  -0.125370f, -0.211235f, -0.554432f,
684   -0.131072f, -0.124982f, -0.130339f, -0.235350f, 0.018903f,  0.012896f,
685   -0.159372f, -0.269571f, -0.025709f, -0.221251f, 0.061919f,  0.016307f,
686   0.384673f,  -0.134525f, -1.599126f, -0.416459f, -0.743052f, 0.670249f,
687   -0.169709f, 0.421681f,  -0.033360f, -0.072817f, 0.003647f,  -0.110632f,
688   -0.158651f, -0.095136f, 0.223759f,  0.165767f,  -0.269129f, -0.196075f,
689   -0.023183f, -0.293420f, 0.014875f,  0.018688f,  -0.153407f, -0.172009f,
690   -0.259947f, -0.124015f, 0.173653f,  -0.089103f, -0.021001f, -0.334230f,
691   0.027177f,  0.103371f,  -0.183860f, -0.204051f, -0.023721f, -0.192297f,
692   -0.143771f, -0.247106f, 0.218116f,  -0.013240f, 2.831783f,  1.483928f,
693   -0.877025f, -0.313462f, -0.411320f, -0.447825f, 0.605977f,  0.234684f,
694   -0.119150f, -0.075182f, -0.330463f, 0.071503f,  -0.254924f, -0.360071f,
695   -0.037022f, 0.063261f,  -0.148759f, -0.238254f, -0.462018f, -0.027166f,
696   0.065318f,  -0.235743f, -0.257194f, -0.094784f, 0.022423f,  0.055925f,
697   0.086672f,  -0.021010f, 0.009965f,  -0.001648f, -0.104917f, -0.387443f,
698   -0.102673f, -0.281706f, 0.145923f,  -0.233391f, -0.378365f, -0.145584f,
699   -0.077751f, -0.121166f, 1.134565f,  -0.097500f, -0.749202f, -0.544566f,
700   -1.361374f, -0.102494f, 1.089275f,  0.375299f,  -0.105091f, 0.037641f,
701   -0.054248f, -0.282691f, -0.377797f, -0.066427f, -0.253815f, -0.329677f,
702   -0.339326f, -0.128217f, -0.282905f, 0.014937f,  1.067185f,  -0.171764f,
703   0.484458f,  0.396706f,  -0.557055f, -0.891596f, -0.257839f, -0.720879f,
704   -0.218449f, -0.004755f, 1.572857f,  0.006229f,  1.962895f,  -0.029746f,
705   -4.137691f, -2.185991f, -2.763477f, -0.520437f, -0.208708f, 0.006444f,
706   -1.263078f, -0.304560f, 1.072374f,  2.556429f,  0.312850f,  0.257488f,
707   -0.634264f, 0.156769f,  -0.188943f, 0.040295f,  -0.389915f, 0.085250f,
708   -0.248525f, 0.045667f,  -0.776115f, -0.274680f, -0.448145f, -0.566161f,
709   -1.285316f, 0.079060f,  0.389124f,  -0.510401f, -0.015299f, -0.664661f,
710   0.099901f,  -0.470694f, -0.051593f, -1.076381f, -0.442104f, -0.197867f,
711   -0.330011f, -0.448523f, -0.301018f, -0.442093f, -0.491953f, -0.582091f,
712   -0.064569f, -0.156516f, 0.543522f,  -0.005924f, 0.161432f,  0.974793f,
713   0.273712f,  1.104850f,  -0.290312f, 0.313417f,  -0.125370f, 0.136234f,
714   -0.191227f, -0.165054f, 0.011872f,  -0.298871f, 0.095740f,  0.142760f,
715   -0.215771f, -0.031437f, 0.101041f,  -0.085620f, 0.435387f,  0.002786f,
716   1.971375f,  0.018392f,  -1.771940f, -0.401433f, 0.808263f,  -3.350013f,
717   2.296952f,  -1.024403f, -0.041645f, -0.034799f, -0.024078f, -0.347301f,
718   -0.276088f, -0.455907f, 0.266021f,  0.087348f,  -0.146566f, 0.040492f,
719   -0.539866f, -0.206851f, -0.387874f, -0.125508f, -0.496676f, -0.373845f,
720   -0.472356f, -0.357082f, -0.081254f, -0.456466f, 0.554713f,  0.002185f,
721   -4.225019f, 0.344025f,  0.728796f,  -0.262936f, 1.383924f,  1.577300f,
722   -2.653320f, -2.516156f, -0.301604f, -0.204105f, -0.138252f, -0.587536f,
723   -0.097889f, -0.352414f, -0.288276f, -0.184340f, -0.122741f, -0.243376f,
724   0.031970f,  -0.373402f, -0.396079f, 0.045566f,  0.072595f,  -0.222681f,
725   -0.243802f, -0.340129f, -0.258494f, -0.192041f, -0.386112f, -0.240940f,
726   -0.047268f, -0.555802f, -0.032514f, -0.241341f, -0.167463f, -0.478308f,
727   -0.205936f, -0.316275f, 0.103729f,  -0.197893f, -0.128029f, -0.218796f,
728   -0.167362f, -0.111814f, -0.126062f, -0.394260f, -0.025357f, -0.402697f,
729   -0.587395f, -0.400385f, -0.259664f, -0.415588f, -0.338503f, -0.399166f,
730   -0.270504f, 0.234505f,  0.272144f,  0.266938f,  -0.392395f, -0.011717f,
731   -0.384221f, -0.473446f, -0.038420f, -0.241101f, -0.234402f, -0.275567f,
732   -0.410454f, -0.377599f, -0.179099f, -0.138432f, -0.248083f, -0.543026f,
733   -0.428043f, -0.239895f, -0.333193f, -0.103346f, -0.039038f, -0.171109f,
734   -0.119432f, -0.222351f, 0.000450f,  0.208724f,  -0.510526f, -0.144656f,
735   -0.316721f, -0.344846f, -0.244794f, -0.129134f, -0.045634f, -0.400183f,
736   0.043714f,  -0.235414f, 0.115594f,  -0.195616f, -0.106693f, -0.124242f,
737   0.083990f,  0.049110f,  -0.196130f, -0.059860f, -0.464235f, -0.516443f,
738   -0.101521f, -0.422379f, -0.413955f, -0.042991f, -0.345263f, -0.129264f,
739   -0.106911f, -0.140156f, -0.457841f, -0.199848f, -0.218954f, -0.329850f,
740   -0.364097f, -0.335262f, -0.312254f, -0.299331f, -0.052710f, -0.251019f,
741   -0.023459f, -0.222538f, 0.028849f,  -0.088038f, -0.301550f, -0.273566f,
742   0.067295f,  -0.174608f, -0.445784f, -0.158366f, -0.567275f, -0.557652f,
743   -0.353503f, -0.302092f, -0.302049f, -0.551793f, -0.034535f, -0.225190f,
744   -0.210733f, -0.219377f, -0.057197f, -0.430933f, -0.025185f, -0.388150f,
745   -0.086147f, -0.430088f, 0.058466f,  -0.152129f, -0.058411f, -0.236392f,
746   -0.547669f, -0.613849f, -0.893774f, -0.351715f, -0.399227f, -0.454909f,
747   -0.324501f, 0.000490f,  -0.282167f, -0.073163f, -0.281452f, 0.047932f,
748   -0.175500f, 0.165220f,  -0.276212f, 0.062153f,  -0.217054f, -0.255487f,
749   -0.146416f, -0.097718f, -0.173809f, -0.559328f, -0.055695f, -0.391193f,
750   -0.132020f, -0.561184f, -0.308666f, -0.474053f, -0.219149f, -0.246558f,
751   -0.158325f, 0.151907f,  -0.266835f, -0.144697f, -0.193960f, -0.046587f,
752   -0.220028f, -0.247355f, 0.135584f,  0.016511f,  0.367705f,  -1.855877f,
753   0.435622f,  0.444710f,  -3.372301f, -3.030489f, 1.013267f,  0.380951f,
754   -0.170011f, -0.111415f, -0.456146f, -0.107254f, -0.095220f, -0.053078f,
755   -0.135864f, -0.591949f, -0.252810f, -0.324799f, -0.094796f, -0.260969f,
756   -0.391981f, -0.063170f, -0.336130f, -0.470127f, -0.405168f, -0.433219f,
757   -0.309563f, -0.295462f, -0.552270f, -0.012300f, -0.057793f, -0.034494f,
758   -0.446843f, -0.640160f, -1.188681f, -0.791361f, 0.543271f,  1.189112f,
759   1.458468f,  -0.005876f, -0.927475f, 0.062038f,  -1.170818f, 0.338227f,
760   -3.007096f, -4.559296f, -4.045457f, -5.953635f, -0.228386f, -0.266890f,
761   -0.092595f, -0.377440f, -0.044534f, -0.053565f, -0.349268f, -0.415030f,
762   -0.310094f, 0.062721f,  0.251422f,  -0.014350f, -1.282910f, 1.619560f,
763   1.180566f,  -0.032163f, -1.322951f, -0.603601f, 1.443710f,  0.654650f,
764   -0.393227f, 0.003536f,  0.029725f,  -0.108925f, -0.053911f, 0.133977f,
765   -0.036145f, -0.168438f, 0.046989f,  -0.331463f, -0.176983f, -0.311922f,
766   -0.272389f, -0.379592f, -0.399993f, -0.297873f, -0.193425f, -0.177524f,
767   -0.258309f, -0.567312f, -0.260217f, -0.241869f, 0.024010f,  -0.032867f,
768   -0.039424f, -0.063670f, 0.193808f,  -0.303514f, -0.013376f, -0.057761f,
769   0.187922f,  0.006938f,  0.031810f,  0.180594f,  -1.198427f, 2.820662f,
770   0.154986f,  -0.375518f, 0.116925f,  -0.795782f, -0.085139f, -0.079365f,
771   -0.197936f, -0.321468f, -0.205271f, -0.558203f, -0.296235f, -0.151193f,
772   -0.158282f, -0.245402f, -0.208504f, -0.042335f, -0.087426f, -0.557129f,
773   -0.381427f, -0.441551f, -0.541011f, -0.060567f, -0.469305f, -0.032326f,
774   -2.453587f, -0.045568f, -0.296932f, 0.613061f,  -0.320284f, 0.191620f,
775   -0.827145f, -0.225277f, 0.275800f,  1.696635f,
776 };
777 
778 static const float av1_ab_partition_nn_bias_32_layer0[64] = {
779   -0.176206f, 0.660189f,  -0.186156f, -2.481963f, -1.564218f, -0.280424f,
780   0.732684f,  -0.135581f, -2.193132f, -0.172771f, 0.605001f,  -0.060392f,
781   -0.067190f, -0.132969f, -1.410812f, -0.298701f, -0.105963f, -0.086173f,
782   0.632779f,  0.005585f,  1.310169f,  1.392136f,  -0.563860f, -0.051053f,
783   0.660998f,  -0.214726f, -1.894342f, -0.128288f, -0.330721f, -0.053988f,
784   -0.177726f, 1.200859f,  -0.178902f, -0.172620f, -0.184476f, -0.175559f,
785   0.538503f,  -0.322158f, -0.219080f, -0.058208f, -0.171347f, -0.216060f,
786   -0.174950f, -0.295740f, -0.184820f, -0.213896f, 1.317728f,  -0.020116f,
787   -0.208096f, 0.000000f,  1.246166f,  -0.225421f, -0.181555f, 0.861761f,
788   1.172429f,  -0.172892f, -0.737092f, -0.189904f, -0.179385f, -0.114618f,
789   -1.384604f, -0.201713f, -0.271948f, 0.372351f,
790 };
791 
792 static const float av1_ab_partition_nn_weights_32_layer1[64 * 16] = {
793   -0.037828f,  1.529029f,  0.004927f,  1.475763f,  0.627172f,  0.325872f,
794   -0.990757f,  0.129476f,  0.889958f,  -0.082031f, 0.332133f,  0.074422f,
795   -0.176212f,  -0.074355f, 0.774378f,  0.110987f,  -0.155469f, 0.253310f,
796   0.882538f,   0.253605f,  0.332436f,  -5.389474f, 0.278470f,  0.168644f,
797   0.914611f,   0.154165f,  0.809262f,  -0.174734f, 0.923673f,  0.064716f,
798   -0.070228f,  -0.228735f, 0.002312f,  0.112222f,  -0.045502f, -0.046004f,
799   0.514101f,   0.306480f,  0.021232f,  -0.015955f, -0.288260f, 0.189177f,
800   -0.104158f,  0.103273f,  0.096910f,  -0.086328f, 1.327289f,  -0.154247f,
801   0.056676f,   -0.243327f, -0.646676f, 0.177221f,  -0.086761f, 0.729729f,
802   -14.710893f, -0.044881f, 0.339003f,  -0.134737f, 0.073621f,  -0.162913f,
803   1.215237f,   0.140723f,  0.138630f,  1.241719f,  0.204092f,  -0.463080f,
804   -0.176086f,  1.125868f,  1.034814f,  0.225455f,  -0.203421f, -0.078787f,
805   -0.527498f,  0.012491f,  -0.563307f, -0.170792f, 0.002679f,  0.116153f,
806   0.211348f,   -0.191900f, -0.212505f, 0.263445f,  -0.074679f, -0.081441f,
807   -0.815405f,  2.448215f,  0.781299f,  0.149542f,  -1.045162f, 0.043014f,
808   0.217381f,   -0.094500f, -0.090427f, 0.025784f,  -0.228906f, -2.741798f,
809   0.230475f,   -0.256112f, -0.103297f, 0.159121f,  -0.229793f, -0.014883f,
810   -0.104131f,  -0.123816f, 0.164148f,  -0.052279f, -0.071845f, -0.041197f,
811   0.208527f,   -0.234197f, -0.542336f, 0.020053f,  0.088870f,  0.014346f,
812   2.502164f,   -0.010244f, -0.267792f, 0.844394f,  2.711486f,  -0.015262f,
813   -0.868053f,  -0.295704f, 0.222289f,  -0.000286f, -0.352098f, -0.079000f,
814   0.021267f,   -0.721739f, -0.240558f, -0.384775f, 0.065974f,  -2.161058f,
815   0.195889f,   0.268966f,  -0.009329f, 0.014949f,  0.314943f,  0.235885f,
816   0.072591f,   -0.127120f, 0.150784f,  0.105697f,  -1.297403f, -0.207509f,
817   -0.217688f,  -0.076752f, 0.170952f,  -0.294235f, 0.449973f,  -1.712690f,
818   0.860989f,   0.054757f,  -0.812627f, -0.105316f, -0.736230f, -0.133192f,
819   -3.741608f,  0.495660f,  -0.288936f, 4.654852f,  -0.021305f, -0.308916f,
820   0.049205f,   -0.259996f, 0.114248f,  -0.252647f, -0.253180f, -0.449314f,
821   0.022979f,   0.063281f,  -0.196154f, 0.078295f,  -0.322317f, -0.145142f,
822   0.300573f,   0.048385f,  -0.254787f, 0.123939f,  -1.263088f, -0.228565f,
823   -0.389061f,  0.391084f,  2.322438f,  0.075009f,  0.225743f,  -0.198808f,
824   -0.280538f,  -0.173939f, -0.120543f, -0.070792f, -0.417187f, -0.781056f,
825   -0.102756f,  -1.760965f, 0.019149f,  -0.867342f, 0.347141f,  0.031588f,
826   0.302572f,   -0.203573f, -0.357320f, -0.096078f, -0.527528f, 0.046699f,
827   -0.108561f,  -0.167077f, -2.851509f, -0.307116f, 0.202720f,  -0.160280f,
828   -0.215525f,  0.064355f,  -0.427220f, 1.516230f,  0.634453f,  0.099400f,
829   -1.013887f,  -0.029740f, -0.093426f, -0.044272f, -1.297636f, -0.237614f,
830   -0.160953f,  0.399036f,  -0.030685f, -0.113619f, -0.184704f, 0.040519f,
831   -0.588252f,  -0.210235f, -0.067623f, -0.031841f, -0.107261f, -0.192582f,
832   -0.253959f,  -0.430821f, -0.103184f, -0.280185f, -0.357723f, 0.197761f,
833   -0.175087f,  -0.055171f, 1.642014f,  -0.192559f, -0.288147f, 0.610311f,
834   4.688195f,   -0.128728f, -0.914869f, -0.108286f, 0.013789f,  0.092125f,
835   0.019770f,   -0.178386f, 0.074164f,  -1.152658f, -0.216738f, -0.277286f,
836   0.012381f,   0.418259f,  -0.680727f, -0.221917f, -0.485946f, 0.101672f,
837   2.009457f,   0.054302f,  1.019838f,  -0.116170f, 0.165134f,  -0.112567f,
838   0.852632f,   -0.385796f, -0.108666f, 0.053181f,  -0.311797f, -0.372875f,
839   -0.675717f,  2.409268f,  -0.514720f, -0.214245f, -0.646596f, 0.009756f,
840   0.203993f,   0.093617f,  -0.301290f, 0.253551f,  -0.128909f, -1.448442f,
841   -0.186823f,  -0.278001f, -0.294993f, -0.176928f, -0.473605f, 0.062049f,
842   -0.212084f,  -0.137326f, 0.012505f,  0.087850f,  -0.200413f, -0.394119f,
843   -0.132224f,  0.146917f,  0.155746f,  0.198725f,  -0.322541f, 0.196391f,
844   -0.945500f,  0.036736f,  -0.155646f, -0.677341f, 1.130545f,  -0.339554f,
845   0.411628f,   -0.355813f, -0.249843f, 0.213694f,  -2.035607f, 0.055694f,
846   -0.111669f,  0.408696f,  -0.067043f, -0.048182f, 0.398110f,  -0.067542f,
847   1.459801f,   0.236833f,  -0.178806f, 0.168758f,  0.492387f,  0.099691f,
848   -0.776680f,  -0.172865f, 0.204225f,  0.193982f,  0.575685f,  -0.062248f,
849   0.011486f,   0.058571f,  -0.493391f, 0.026893f,  -0.900467f, 3.793129f,
850   -0.634613f,  -0.064660f, -0.048262f, 0.361905f,  0.033641f,  0.245171f,
851   -0.064671f,  0.034954f,  0.204358f,  -0.904023f, -0.052714f, -0.250134f,
852   0.136700f,   0.000734f,  -0.371720f, 0.226483f,  0.217958f,  0.060559f,
853   0.180111f,   0.000970f,  0.079556f,  -0.096775f, 0.093855f,  -0.026224f,
854   -0.243664f,  0.004290f,  0.123281f,  -0.239476f, 1.230374f,  -0.107826f,
855   -0.101982f,  -0.153917f, 5.464427f,  0.304375f,  -0.809957f, 0.090564f,
856   -0.278416f,  -0.245555f, -2.078421f, 0.243093f,  -0.127666f, 0.052451f,
857   -0.126662f,  -0.783505f, 0.025149f,  -1.422675f, -0.207769f, -0.362547f,
858   0.115310f,   0.133390f,  1.264754f,  -0.027055f, -0.485312f, -0.240717f,
859   -0.239722f,  0.146818f,  -1.265043f, -0.235553f, 0.267104f,  -0.021357f,
860   -0.435949f,  -0.309371f, 0.049920f,  1.302721f,  -0.233978f, -0.097551f,
861   -0.240631f,  -0.287821f, -0.378380f, -0.273131f, -3.075169f, 0.226404f,
862   -0.029361f,  2.703590f,  -0.430659f, 0.067927f,  -0.387520f, -0.370630f,
863   -0.229236f,  0.085653f,  -0.370956f, -0.065556f, -0.187859f, 0.068309f,
864   -0.109299f,  -0.259898f, -0.103644f, -0.271199f, -0.209350f, 0.140993f,
865   -0.196713f,  -0.135508f, -1.423209f, -0.406385f, -0.019956f, -0.864694f,
866   5.963707f,   -0.201157f, 0.726377f,  -0.011076f, 0.010553f,  -0.102918f,
867   -2.230088f,  -0.258098f, -0.039547f, -0.029262f, -0.082324f, -0.860222f,
868   -0.094735f,  -1.381839f, 0.587298f,  -0.173048f, 0.721360f,  0.241900f,
869   0.764302f,   -0.023609f, -1.173755f, 0.103912f,  -0.185363f, 0.078435f,
870   -2.245062f,  -0.127269f, 0.202234f,  0.158975f,  -0.260909f, 0.098608f,
871   -0.348247f,  1.732502f,  -0.412298f, -0.269602f, -0.425771f, -0.146243f,
872   -0.530730f,  0.125716f,  -1.004419f, 0.145109f,  -0.059289f, 1.096304f,
873   0.012891f,   0.045033f,  -0.306875f, 0.003514f,  -0.176110f, 0.037544f,
874   -0.441537f,  -0.518921f, -0.262149f, -0.060407f, -0.379419f, -0.141245f,
875   -0.128894f,  -0.176537f, -1.161318f, -0.249100f, -0.118330f, 0.042816f,
876   1.173404f,   0.088312f,  -0.393568f, -0.175134f, 6.529819f,  -0.326652f,
877   -0.631917f,  -0.393476f, 0.057781f,  -0.217748f, -1.781139f, -0.012614f,
878   -0.212621f,  -0.720322f, -0.218498f, -0.388556f, -0.254796f, -0.248399f,
879   -0.608744f,  -0.265146f, 0.238517f,  0.066882f,  -2.916806f, 0.054642f,
880   0.282590f,   0.075248f,  0.010188f,  -0.133486f, 0.985945f,  -0.045849f,
881   -0.347564f,  0.057320f,  -0.417920f, 0.063664f,  0.387062f,  -2.692059f,
882   -0.535549f,  0.263736f,  0.327889f,  -0.070273f, -0.775254f, 0.147250f,
883   3.309425f,   -0.212191f, -0.067204f, -2.912663f, -0.061496f, 0.084233f,
884   0.022907f,   0.138421f,  -0.112159f, -0.288447f, -0.010799f, 0.056049f,
885   -0.036527f,  0.021525f,  0.106649f,  -0.291883f, 0.088424f,  -0.057773f,
886   -0.086031f,  0.015277f,  -0.318505f, -0.269049f, -1.008913f, -0.224785f,
887   -0.025820f,  -0.649037f, 0.706381f,  0.096410f,  0.643776f,  -0.046743f,
888   -0.009654f,  -0.024246f, 1.469255f,  -0.183536f, -0.370046f, -0.048442f,
889   -0.376527f,  -0.431264f, -0.245109f, -0.093951f, 0.203683f,  -0.099872f,
890   0.087210f,   0.160692f,  -3.527694f, -0.068891f, -0.228994f, -0.231817f,
891   -0.241949f,  0.193613f,  0.979597f,  -0.091259f, 0.414424f,  -0.047341f,
892   -0.209582f,  -0.295134f, -0.016824f, 0.460327f,  -0.072671f, 0.246234f,
893   0.235896f,   0.127238f,  -1.068683f, 0.035648f,  2.254888f,  0.180105f,
894   -0.260098f,  -2.322120f, -0.184249f, -0.314801f, -0.099969f, -0.272117f,
895   -0.237916f,  0.031103f,  -0.274063f, -0.049384f, -0.044917f, 0.102477f,
896   -0.342148f,  -0.257558f, -0.346300f, 0.115333f,  -0.115456f, 0.208354f,
897   -0.359301f,  -0.167395f, 1.146514f,  -0.177861f, -0.098658f, -0.444570f,
898   6.759993f,   -0.369772f, -0.831118f, 0.001866f,  -0.073298f, -0.072095f,
899   0.811902f,   -0.431997f, -0.286587f, -0.269500f, 0.111492f,  -0.525364f,
900   -0.351785f,  -2.463474f, -1.852659f, 0.135325f,  0.138267f,  0.100643f,
901   -2.373278f,  -0.285514f, -0.395388f, -0.185016f, -0.030249f, -0.005767f,
902   -0.716424f,  -0.031674f, 0.011147f,  0.057405f,  -0.215873f, -0.094401f,
903   0.573528f,   -1.223820f, 0.414852f,  -0.059053f, -0.076488f, -0.287168f,
904   -0.842640f,  0.174084f,  -0.567186f, 0.336629f,  -0.062514f, 2.075448f,
905   -0.061680f,  -0.131529f, -0.098994f, -0.204111f, -0.347865f, 0.108516f,
906   -0.049616f,  -0.069212f, -0.273935f, -0.096545f, -0.210784f, -0.284698f,
907   0.141501f,   -0.176924f, -0.361341f, -0.251197f, -0.286694f, 0.245569f,
908   -1.521661f,  -0.122639f, -0.015760f, -0.718912f, 5.877828f,  0.146916f,
909   0.151767f,   0.220785f,  -0.032298f, 0.230902f,  0.663943f,  -0.252613f,
910   0.057718f,   -0.436038f, -0.323994f, -1.139787f, -0.042489f, -1.326298f,
911   -1.031206f,  -0.104136f, 0.389897f,  0.127602f,  -2.667789f, -0.212366f,
912   -0.506262f,  -0.009115f, -0.213202f, 0.076167f,  -1.629405f, 0.055129f,
913   0.375393f,   -0.150272f, -0.241515f, -0.326497f, 0.100069f,  0.410703f,
914   0.340622f,   0.042437f,  -0.349945f, 0.041176f,  -1.178950f, 0.030992f,
915   0.933908f,   -0.035844f, -0.098660f, 1.030584f,  -0.092043f, -0.355739f,
916   -0.305562f,  0.036161f,  -0.049558f, -0.033225f, -0.403856f, -0.088276f,
917   0.215493f,   -0.149105f, -0.013363f, 0.025886f,  -0.101306f, -0.205781f,
918   -1.072487f,  -0.076019f, 0.077555f,  0.131003f,  1.267763f,  -0.008954f,
919   -0.327617f,  -0.246539f, 6.664081f,  -0.404403f, -1.442489f, 0.191301f,
920   -0.336361f,  0.181156f,  0.833108f,  0.007879f,  -0.194464f, -1.029408f,
921   -0.036268f,  -0.927110f, -0.379190f, -0.293443f, -1.848579f, -0.242548f,
922   -0.065990f,  0.203160f,  -0.291788f, 0.000680f,  0.587011f,  -0.241289f,
923   0.037034f,   0.000552f,  1.072308f,  -0.387230f, -0.230050f, 0.292322f,
924   -0.720001f,  0.034109f,  -0.467260f, 2.211644f,  -1.839191f, -0.048797f,
925   -0.083469f,  -0.334686f, -0.269056f, 0.051295f,  1.319904f,  -0.035603f,
926   -0.018457f,  -0.824915f, -0.212285f, -0.230516f, -0.035093f, -0.400843f,
927   -0.305469f,  -0.099011f, 0.014225f,  -0.452772f, 0.170331f,  -0.389312f,
928   -0.115084f,  -0.014770f, -0.429387f, -0.155961f, -0.568200f, -0.037853f,
929   -0.125137f,  0.067228f,  -1.329271f, -0.117874f, -0.132499f, -0.218376f,
930   -0.588325f,  -0.320024f, 0.085695f,  -0.235047f, -0.217790f, 0.103015f,
931   -0.698644f,  0.017766f,  -0.058299f, 0.199411f,  -0.122485f, -0.563949f,
932   -0.349011f,  -0.557045f, -0.131165f, 0.002281f,  0.118559f,  -0.210302f,
933   -1.153815f,  0.116738f,  -0.236007f, -0.003487f, -0.006885f, -0.244816f,
934   0.953222f,   0.093748f,  0.266869f,  0.241869f,  -0.860832f, -0.387012f,
935   -0.338986f,  2.097515f,  -1.942512f, -0.298021f, 0.543911f,  -0.043214f,
936   0.082125f,   -0.120242f, 0.712231f,  0.213327f,  -0.301687f, -0.544011f,
937   -0.392131f,  0.004302f,  0.004825f,  -0.317440f, -0.107518f, -0.293407f,
938   -0.159111f,  -0.080367f, 0.132663f,  -0.017726f, -0.237521f, -0.190297f,
939   -0.361633f,  0.200518f,  -0.538296f, -0.027975f, -0.381704f, -0.016963f,
940   0.630105f,   -0.190997f, -0.287840f, -0.603488f, 3.605598f,  -0.276614f,
941   -1.346383f,  0.186912f,  -0.047575f, -0.189232f, -1.519072f, 0.097816f,
942   -0.223722f,  0.304924f,  -0.213022f, -1.052433f, -0.322283f, -1.706734f,
943   -2.458027f,  0.237976f,  0.171050f,  -0.103139f, -0.278689f, 0.329824f,
944   -0.262448f,  -0.122916f, -0.236398f, -0.013848f, -0.969160f, -0.374907f,
945   0.091018f,   -0.386471f, -0.723940f, 0.064956f,  -0.057652f, 1.321024f,
946   -1.397418f,  -0.143136f, 0.272468f,  -0.030749f, 0.037324f,  0.069316f,
947   -0.904925f,  -0.333693f, -0.117709f, 2.279598f,  -0.428065f, -0.131157f,
948   -0.014288f,  -0.402862f, -0.666090f, 0.017070f,  -0.028333f, 0.002481f,
949   0.197156f,   -0.038120f, -0.271062f, -0.188275f, -0.021370f, -0.070849f,
950   -0.905007f,  -0.095886f, -0.093055f, -0.121821f, -1.239812f, -0.411799f,
951   -0.089948f,  -0.936827f, 1.437569f,  -0.388908f, 0.126170f,  0.186162f,
952   -0.018819f,  -0.138364f, -1.066412f, -0.138222f, -0.022186f, 0.107331f,
953   -0.230436f,  -1.352605f, -0.161323f, -1.081810f, -0.933825f, -0.136675f,
954   0.378157f,   0.113377f,  -0.850610f, 0.080245f,  -0.087305f, -0.002852f,
955   0.044408f,   -0.188172f, -1.891998f, 0.092189f,  0.125325f,  -0.105090f,
956   -0.848510f,  -0.396308f, -0.384130f, 2.007509f,  -1.480787f, -0.126946f,
957   0.314767f,   0.000195f,  -0.285628f, -0.110442f, -0.293948f, 0.258559f,
958   -0.417603f,  1.570705f,  0.092459f,  -0.340974f, -0.284754f, -0.007801f,
959   -0.324610f,  -0.004734f, -0.207716f, -0.057175f, 0.055467f,  -0.210830f,
960   -0.113005f,  -0.299177f, 0.068074f,  0.017929f,  -2.897598f, -0.260074f,
961   -0.014422f,  -0.206467f, 1.246997f,  -0.372863f, -0.214160f, -0.114035f,
962   5.805862f,   0.003611f,  -1.340990f, -0.021085f, -0.260431f, -0.002720f,
963   -1.251640f,  -0.353531f, -0.304009f, -0.153376f,
964 };
965 
966 static const float av1_ab_partition_nn_bias_32_layer1[LABEL_SIZE] = {
967   -0.521497f, -1.061572f, -0.078756f, -0.660662f, -0.403741f, -0.960163f,
968   0.001427f,  0.523607f,  0.225068f,  -0.055273f, 1.019519f,  1.181880f,
969   -0.010198f, 0.130597f,  1.276752f,  2.028188f,
970 };
971 
972 static const NN_CONFIG av1_ab_partition_nnconfig_32 = {
973   FEATURE_SIZE,  // num_inputs
974   LABEL_SIZE,    // num_outputs
975   1,             // num_hidden_layers
976   {
977       64,  // num_hidden_nodes
978   },
979   {
980       av1_ab_partition_nn_weights_32_layer0,
981       av1_ab_partition_nn_weights_32_layer1,
982   },
983   {
984       av1_ab_partition_nn_bias_32_layer0,
985       av1_ab_partition_nn_bias_32_layer1,
986   },
987 };
988 
989 // nn model for ab partition pruning, 16x16.
990 static const float av1_ab_partition_nn_weights_16_layer0[FEATURE_SIZE * 64] = {
991   0.151902f,  0.007947f,  -1.788454f, 0.431869f,  -2.971387f, 0.923566f,
992   1.632542f,  -1.665136f, -0.338632f, -5.075884f, 0.398267f,  0.030467f,
993   2.263534f,  -0.045532f, -1.066128f, 0.915139f,  -0.560500f, -3.293125f,
994   2.072793f,  -1.011414f, 0.122716f,  -0.060169f, -0.388860f, 0.031019f,
995   -0.381861f, 0.001551f,  -0.328472f, 0.038296f,  -0.060398f, -0.375556f,
996   0.209226f,  0.014764f,  -1.443469f, -0.345486f, 2.409269f,  1.524846f,
997   -0.640666f, 1.322139f,  -2.074771f, -0.580944f, -0.203960f, -0.072893f,
998   0.329701f,  0.115339f,  -1.339542f, 0.249024f,  -0.421545f, -0.409151f,
999   -0.258293f, 0.836288f,  -0.073685f, -0.009624f, 0.895712f,  0.320639f,
1000   0.451002f,  -1.544558f, 0.193709f,  -1.389012f, 1.305451f,  0.089795f,
1001   0.050338f,  -0.017433f, -0.304667f, 0.500729f,  0.504346f,  0.073757f,
1002   0.582649f,  -0.993623f, 1.766766f,  -3.067265f, -0.415774f, -0.006036f,
1003   -1.245281f, 0.253205f,  -0.591245f, -0.626238f, 0.551852f,  0.593755f,
1004   0.491023f,  1.099384f,  -0.348448f, 0.054564f,  -0.451422f, -0.375781f,
1005   -0.248390f, -0.052548f, -0.380069f, -0.165391f, -0.297968f, -0.052142f,
1006   -0.316381f, -0.045246f, -0.243905f, -0.034169f, -0.247523f, -0.180773f,
1007   0.068066f,  -0.374920f, 0.057536f,  -0.189748f, 0.058375f,  -0.267749f,
1008   -0.147286f, -0.246153f, 0.006183f,  -0.202029f, -0.059128f, 0.116852f,
1009   0.134719f,  -0.126900f, -0.064646f, -0.196458f, -0.182331f, 0.108029f,
1010   -0.264499f, 0.155816f,  -0.107255f, -0.056983f, -0.209771f, -0.099070f,
1011   0.007313f,  -0.254124f, -0.231964f, -0.275972f, 0.032098f,  -0.264564f,
1012   -0.208743f, 0.155599f,  -0.121511f, -0.156145f, -0.162315f, -0.059788f,
1013   -0.257073f, -0.076654f, -0.110616f, -0.321675f, -0.051952f, 0.006301f,
1014   -0.154114f, 0.017032f,  -0.017364f, -0.233247f, 0.009918f,  -0.179289f,
1015   -0.190722f, 0.147106f,  -0.063910f, -0.396872f, -0.263123f, -0.003850f,
1016   -0.040718f, -0.324699f, 0.118660f,  -0.170727f, -0.316788f, 0.100886f,
1017   -0.202842f, 0.045371f,  0.150561f,  -0.057054f, -0.308150f, 0.028346f,
1018   -0.381473f, -0.195365f, 0.026221f,  -0.281795f, 0.087204f,  0.047689f,
1019   -0.027643f, -0.104724f, -0.089030f, -0.117661f, -0.349160f, 0.056982f,
1020   -0.340273f, 0.048086f,  0.046103f,  -0.121527f, 0.021697f,  0.054109f,
1021   -0.002768f, -0.008461f, -2.297240f, 0.124651f,  3.621661f,  -0.057120f,
1022   -1.151656f, 2.296894f,  -3.678720f, -0.290240f, 0.087683f,  -0.186389f,
1023   0.007656f,  -0.090236f, -0.245217f, 0.110389f,  -0.251719f, -0.029084f,
1024   -0.128203f, -0.100005f, -0.032779f, 0.007281f,  -0.366596f, -0.267870f,
1025   -0.215620f, 0.047687f,  0.010303f,  0.097980f,  -0.191569f, -0.341162f,
1026   0.119249f,  0.026279f,  -2.161546f, 0.459591f,  1.290566f,  1.791797f,
1027   -0.409835f, 0.127081f,  -1.156367f, 0.198286f,  0.099561f,  -0.067445f,
1028   -0.034352f, 0.017966f,  -0.277380f, -0.057220f, -0.174198f, -0.014164f,
1029   0.146090f,  -0.357530f, 0.097644f,  -0.000932f, 0.446603f,  -0.066793f,
1030   2.448620f,  0.937617f,  -1.232922f, 0.313183f,  0.816827f,  -0.275115f,
1031   -0.245205f, -0.126895f, 0.156668f,  -0.186977f, -0.273505f, 0.013315f,
1032   0.168629f,  -0.089084f, 0.006166f,  -0.116107f, -0.199316f, -0.024010f,
1033   -0.242303f, 0.011612f,  -0.218485f, -0.229661f, -0.123922f, 0.136699f,
1034   0.006732f,  -0.148718f, -0.164225f, 0.116063f,  1.587898f,  0.690519f,
1035   0.360566f,  0.009739f,  -0.678702f, -0.046003f, 0.126984f,  0.605212f,
1036   1.240663f,  -0.000228f, -1.119369f, -0.415589f, -0.721003f, 0.097936f,
1037   -1.410586f, -2.358833f, -2.773129f, -3.983361f, -0.087144f, -0.050029f,
1038   -0.242255f, 0.137424f,  -0.307490f, -0.084637f, -0.023812f, -0.196582f,
1039   -0.078695f, 0.038257f,  -0.012110f, -0.263521f, 0.009839f,  -0.109125f,
1040   -0.226036f, 0.060712f,  0.093671f,  0.153143f,  0.039116f,  -0.290891f,
1041   0.227057f,  -0.204633f, -0.207539f, -0.148242f, 0.046204f,  -0.231268f,
1042   -0.209315f, -0.307579f, -0.436556f, 0.023475f,  0.131793f,  -0.038301f,
1043   1.650584f,  0.392570f,  1.446576f,  1.254380f,  -0.516867f, -0.057116f,
1044   0.149320f,  0.414424f,  -0.246309f, 0.003877f,  -0.480238f, -1.037035f,
1045   -0.830779f, -1.122244f, -0.408267f, -0.253956f, 0.382005f,  0.940609f,
1046   -1.113370f, -0.018554f, 0.141064f,  -0.182504f, 1.270707f,  0.414904f,
1047   -0.216036f, 0.203831f,  0.450716f,  -0.452909f, 0.139358f,  -0.027143f,
1048   1.956892f,  1.643732f,  -0.867839f, -0.620520f, -0.334607f, -0.519982f,
1049   0.205023f,  0.661159f,  -0.000809f, 0.049033f,  -0.348579f, -0.200338f,
1050   -0.362144f, -0.346590f, -0.230096f, 0.180746f,  -0.149954f, -0.253429f,
1051   -0.378170f, -0.040724f, -0.041597f, 0.243659f,  -0.472181f, 0.015401f,
1052   -0.180376f, 0.153139f,  -0.247738f, -0.010485f, -0.157158f, 0.016825f,
1053   -0.238925f, -0.265798f, -0.318374f, 0.142352f,  -0.210520f, 0.051928f,
1054   -0.352190f, -0.179052f, -0.185498f, 0.025540f,  -0.111667f, -0.235187f,
1055   -0.215454f, 0.010931f,  -0.238372f, -0.126659f, 0.075691f,  -0.091167f,
1056   -2.462379f, -0.007950f, -0.637990f, 0.285554f,  -0.051275f, 0.282279f,
1057   -0.744083f, -0.570646f, 0.592198f,  1.421332f,  -0.256027f, -0.140315f,
1058   0.160247f,  -0.063185f, -0.055895f, -0.199864f, -0.287353f, -0.074561f,
1059   -0.071228f, 0.055864f,  -1.084764f, -0.263409f, 0.779266f,  0.228187f,
1060   0.375013f,  0.121204f,  -0.656948f, 0.533561f,  0.272671f,  -0.015423f,
1061   -0.124180f, -0.009127f, 2.934838f,  -0.150998f, 1.163152f,  0.081997f,
1062   -4.715939f, -3.676595f, -1.524886f, -0.167593f, 0.281186f,  0.024046f,
1063   -1.451709f, 0.332558f,  0.990504f,  0.376290f,  -1.466773f, -0.448439f,
1064   -2.929108f, -4.255188f, 0.065238f,  0.019950f,  1.372393f,  0.444052f,
1065   -2.538772f, 1.579767f,  -0.464911f, -1.866114f, 1.053958f,  0.434467f,
1066   -0.125964f, 0.034671f,  0.077116f,  -0.138466f, -0.413395f, -0.223453f,
1067   -0.172127f, -0.251265f, -0.048239f, -0.395519f, 0.023141f,  0.037459f,
1068   -0.249593f, -0.062215f, -0.047209f, -0.435189f, -0.164155f, -0.077590f,
1069   -0.241164f, -0.126128f, -0.038243f, -0.180888f, 0.198840f,  -0.328036f,
1070   -0.169790f, 0.036506f,  0.052572f,  -0.183570f, -0.073617f, -0.244959f,
1071   0.266498f,  0.032846f,  -1.902106f, 0.486078f,  2.414993f,  0.975182f,
1072   -0.382875f, 1.647810f,  -2.197017f, -0.890107f, 0.221287f,  0.010889f,
1073   3.817042f,  0.572728f,  0.092466f,  0.473337f,  -1.634659f, -1.069455f,
1074   1.486776f,  -1.023850f, 0.088184f,  0.008842f,  0.518202f,  0.270259f,
1075   1.757191f,  -0.121839f, -2.912229f, -1.250866f, -2.381808f, 0.335309f,
1076   -0.120079f, -0.061294f, -0.058725f, -0.315169f, -0.262443f, 0.072434f,
1077   -0.267836f, -0.319354f, -0.274975f, 0.068970f,  -0.406467f, 0.044074f,
1078   -0.152311f, -0.333656f, -0.228355f, -0.185613f, 0.017346f,  -0.177674f,
1079   -0.090675f, -0.102047f, -0.011768f, -0.025280f, -0.271661f, 0.098099f,
1080   -0.312272f, -0.222217f, -0.100548f, 0.106260f,  -0.034655f, 0.135109f,
1081   -0.021276f, 0.018177f,  -0.353097f, -0.011128f, 0.061136f,  -0.511662f,
1082   -0.223236f, -0.308841f, 0.118789f,  -0.154628f, -0.053178f, -0.055973f,
1083   0.013175f,  -0.368337f, -0.090863f, -0.116920f, 0.178990f,  -0.025278f,
1084   -0.190553f, -0.238092f, 0.303943f,  -0.024944f, 0.719373f,  0.384332f,
1085   -0.378480f, -0.423316f, 0.709922f,  0.758514f,  -1.559023f, -2.503173f,
1086   0.068652f,  -0.234741f, -0.182932f, 0.037878f,  0.020684f,  -0.174142f,
1087   -0.182300f, -0.052796f, -0.219145f, 0.113028f,  -1.041826f, 0.035317f,
1088   0.919904f,  -0.676011f, 0.652297f,  1.456447f,  -0.166904f, -0.861823f,
1089   0.895827f,  0.429821f,  -0.180376f, -0.076587f, -0.273945f, -0.288990f,
1090   -0.206692f, -0.080745f, -0.085444f, 0.186953f,  -0.050135f, 0.044243f,
1091   -0.391706f, -0.160498f, -0.292268f, 0.164060f,  0.412649f,  0.211611f,
1092   -0.327294f, -0.919399f, 0.320297f,  0.385284f,  -0.088848f, -0.072556f,
1093   -0.384813f, -0.176267f, -0.065918f, 0.134724f,  -0.231104f, -0.337707f,
1094   -0.195442f, -0.263569f, 0.098090f,  -0.341411f, -0.189211f, -0.439276f,
1095   -0.404046f, 0.262491f,  -0.311093f, -0.086454f, -0.013400f, -0.061447f,
1096   -0.026945f, -0.112036f, -0.322985f, 0.078500f,  -0.230205f, -0.344535f,
1097   -0.021087f, 0.110220f,  -0.128671f, 0.044219f,
1098 };
1099 
1100 static const float av1_ab_partition_nn_bias_16_layer0[64] = {
1101   2.936406f,  -0.396539f, -0.110456f, -1.254954f, 0.785350f,  0.516290f,
1102   -0.172341f, 0.254386f,  -0.192465f, -0.106751f, -0.055518f, -0.094994f,
1103   0.000000f,  -0.065018f, -0.004908f, -0.130483f, -0.119580f, -0.142072f,
1104   0.457446f,  -0.125051f, -0.107712f, 0.714607f,  -0.140809f, -1.788650f,
1105   -0.087199f, 0.000000f,  -1.290050f, 0.443930f,  -0.110634f, -0.109380f,
1106   -0.188213f, -1.414179f, 1.193579f,  0.388775f,  -0.873193f, -0.110050f,
1107   -0.072565f, -0.117050f, -0.119132f, 0.456959f,  -0.132069f, 0.131974f,
1108   1.160474f,  1.746465f,  0.442628f,  -0.188849f, -0.207794f, -0.108364f,
1109   -0.856655f, -2.141620f, 0.335476f,  -0.105508f, -0.212162f, -0.109319f,
1110   -0.237213f, -0.109980f, -0.291044f, -0.137877f, 0.470191f,  -0.023908f,
1111   0.123809f,  -0.109797f, 0.200510f,  -0.147542f,
1112 };
1113 
1114 static const float av1_ab_partition_nn_weights_16_layer1[64 * LABEL_SIZE] = {
1115   -6.823716f, 1.406568f,  -0.144009f, 2.228765f,  0.838336f,  0.738107f,
1116   -0.319014f, -0.148756f, 0.240862f,  -0.111089f, -0.004241f, 0.025758f,
1117   -0.193820f, -0.246362f, -0.181363f, -0.201556f, 0.024268f,  0.252994f,
1118   -0.289443f, 0.194932f,  0.057467f,  0.724735f,  0.014063f,  1.361352f,
1119   0.025191f,  0.024274f,  0.231462f,  -7.227959f, -0.094515f, 0.039946f,
1120   0.412719f,  0.812318f,  3.038903f,  -0.286289f, 0.647482f,  -0.115114f,
1121   0.053590f,  0.066069f,  0.153134f,  0.996250f,  -0.125700f, 0.951365f,
1122   -6.243494f, -4.827697f, 0.566320f,  0.239515f,  -0.099702f, 0.054546f,
1123   1.847330f,  3.680076f,  -3.049829f, -0.127709f, 0.068469f,  -0.017794f,
1124   0.223864f,  -0.106778f, -0.020425f, -0.040226f, -0.251890f, -0.168673f,
1125   -0.552073f, 0.043311f,  0.218668f,  0.033209f,  -3.199210f, 0.193079f,
1126   0.321406f,  0.718307f,  -0.181418f, -0.459612f, -1.981170f, 0.968496f,
1127   -0.029757f, -0.130065f, 0.043782f,  0.072394f,  -0.088686f, 0.025322f,
1128   0.129882f,  0.101324f,  0.335707f,  0.072714f,  -2.079774f, 0.203997f,
1129   0.239321f,  -0.301757f, 0.257845f,  1.288382f,  -0.031275f, -0.234194f,
1130   0.310722f,  2.045469f,  0.034716f,  0.135638f,  -0.251388f, 0.320071f,
1131   -1.065301f, -0.322731f, -0.545028f, 0.226276f,  0.090799f,  0.019289f,
1132   0.048950f,  -1.079300f, 0.231938f,  0.083683f,  4.762127f,  0.145037f,
1133   -0.145549f, 0.075592f,  0.172336f,  0.108175f,  0.333751f,  1.090501f,
1134   1.056114f,  0.047073f,  0.182052f,  -0.081587f, 0.089900f,  0.339286f,
1135   2.049988f,  0.073585f,  0.537355f,  -0.243322f, -0.010179f, -0.052601f,
1136   -0.174915f, 0.117793f,  2.222990f,  -2.520837f, -0.092699f, 1.199887f,
1137   0.138720f,  0.679918f,  -0.463155f, -0.659496f, -0.109913f, -0.003398f,
1138   0.114633f,  -0.128377f, 0.092970f,  -0.107489f, -0.191078f, 0.185182f,
1139   0.216980f,  -0.019343f, 3.443133f,  0.287953f,  0.099314f,  0.985958f,
1140   0.157268f,  -0.606516f, 0.049418f,  -0.221809f, -0.453081f, -0.344796f,
1141   -0.003735f, -0.107269f, -0.128541f, -0.259543f, -0.934806f, -0.542456f,
1142   -1.011192f, 0.022795f,  0.186363f,  -0.076356f, -0.050932f, -0.165098f,
1143   0.168177f,  -0.101596f, -5.270886f, 2.553943f,  -0.440870f, -0.017494f,
1144   0.215208f,  -0.017032f, 1.495915f,  -4.304677f, 0.762211f,  0.182937f,
1145   0.254406f,  -0.029433f, -0.088364f, -0.110160f, -0.108257f, -0.036538f,
1146   0.737697f,  -0.234989f, 0.168095f,  0.245118f,  -0.077262f, 0.195718f,
1147   0.753302f,  -1.637869f, 0.126227f,  0.982129f,  -0.121444f, -0.295570f,
1148   -1.215799f, 0.147867f,  -0.068496f, 0.132726f,  -0.005772f, -0.181774f,
1149   0.126513f,  0.204723f,  -0.366123f, 0.103906f,  -0.148053f, -0.075272f,
1150   0.243884f,  -0.104828f, 0.198988f,  0.501034f,  -0.112671f, 0.111421f,
1151   0.167508f,  -0.117803f, -0.738624f, 2.046292f,  0.124011f,  0.057983f,
1152   -0.359154f, -0.648883f, -0.259462f, -0.459041f, -2.501223f, -0.065138f,
1153   0.122417f,  0.060291f,  -0.129033f, -0.843086f, 0.268241f,  -0.399927f,
1154   1.585888f,  1.816393f,  -0.631427f, 0.127826f,  0.088105f,  0.073488f,
1155   0.717694f,  -1.497362f, 2.608528f,  0.066896f,  -0.079230f, 0.223436f,
1156   -0.010530f, 0.175310f,  1.120365f,  0.034391f,  0.835312f,  0.071652f,
1157   -0.080615f, 0.111395f,  0.162742f,  0.079927f,  -3.859582f, -0.638431f,
1158   -0.167880f, -0.992659f, -0.885355f, -1.276197f, 1.334344f,  0.931940f,
1159   -0.078244f, -0.149030f, -0.070974f, -0.133566f, 0.200034f,  0.102793f,
1160   -0.048546f, 0.063545f,  0.023864f,  -0.190863f, 1.934257f,  -0.136286f,
1161   -0.107916f, -0.637468f, 0.066449f,  1.089693f,  -0.214047f, -0.265780f,
1162   0.899660f,  -0.130333f, 0.288311f,  -0.049024f, 0.090202f,  0.487969f,
1163   0.339704f,  0.858479f,  0.841253f,  -0.184100f, -0.637070f, -0.125071f,
1164   -0.077650f, -0.087877f, 0.202268f,  -0.027300f, 2.842862f,  -0.100698f,
1165   -0.259080f, 0.260556f,  0.157912f,  -0.070364f, 0.467190f,  1.200037f,
1166   1.419317f,  -0.033588f, -0.227824f, 0.292617f,  0.228574f,  0.213839f,
1167   -1.091099f, -0.022258f, -1.294681f, 0.136118f,  0.081652f,  -0.185359f,
1168   -0.039706f, 0.191407f,  -2.053219f, -0.261934f, 0.047812f,  -0.029536f,
1169   -0.823869f, -1.090534f, -0.755890f, 0.441035f,  -0.167945f, 0.231441f,
1170   -0.135013f, -0.260762f, 0.256872f,  0.130339f,  -0.243751f, 0.189760f,
1171   -0.288454f, 0.145363f,  0.338490f,  0.403898f,  -0.022814f, -1.263598f,
1172   -0.101315f, 0.860135f,  0.136511f,  0.028942f,  0.574047f,  2.656370f,
1173   0.037587f,  -0.188690f, -0.125312f, 1.100435f,  -1.080402f, 0.380905f,
1174   0.004635f,  0.097144f,  -0.214309f, 0.085552f,  -0.285066f, -0.705134f,
1175   -0.054704f, -0.319951f, 5.486626f,  0.958158f,  -1.380585f, 0.223340f,
1176   -0.169167f, -0.170697f, -0.216748f, 0.324232f,  2.684204f,  -0.008490f,
1177   -0.211052f, -0.201190f, 0.123466f,  -0.000234f, 0.579907f,  0.096938f,
1178   -0.042745f, 0.201855f,  0.157195f,  -0.261440f, 0.029699f,  -0.046599f,
1179   1.618216f,  -2.596280f, -0.377420f, -0.526725f, -0.493592f, -0.579615f,
1180   0.579699f,  -0.100392f, 0.150694f,  0.061794f,  0.200425f,  -0.062515f,
1181   -0.179122f, 0.250112f,  -0.344675f, -0.118359f, -0.095670f, 0.152311f,
1182   3.662276f,  -0.154921f, -0.312991f, 0.972008f,  -0.308596f, -0.190426f,
1183   0.133889f,  -0.238673f, -0.094726f, 1.683835f,  -0.215629f, -0.198890f,
1184   -0.035278f, -0.367973f, -0.822435f, 0.240848f,  -0.194656f, 0.034655f,
1185   -0.079424f, 0.146670f,  0.026646f,  -0.034507f, 0.059467f,  -0.153109f,
1186   -0.431033f, 2.552991f,  -1.894091f, -0.180462f, -0.306839f, -0.025648f,
1187   1.026326f,  -3.096230f, 1.346935f,  0.033633f,  -0.181827f, 0.094376f,
1188   0.001696f,  -0.379264f, -1.069503f, -0.140972f, -0.208769f, -0.195239f,
1189   0.281795f,  -0.127251f, 0.180776f,  0.067763f,  0.697124f,  -1.040779f,
1190   0.111280f,  0.188351f,  -0.340234f, -0.207790f, -0.720075f, -0.137409f,
1191   -0.070310f, -0.032918f, -0.060787f, 0.131484f,  -0.077845f, -0.258652f,
1192   0.056911f,  -0.062034f, 0.007663f,  -0.185100f, 1.340361f,  0.014096f,
1193   -0.124602f, 0.194241f,  0.128383f,  0.360465f,  0.082979f,  -0.050475f,
1194   -0.519294f, 3.323262f,  0.067014f,  0.221203f,  -0.085082f, -0.228606f,
1195   -0.916668f, -0.022643f, -1.386737f, -0.131902f, -0.349952f, -0.032874f,
1196   -0.189190f, -0.898790f, -0.102394f, -1.017387f, 2.214050f,  1.790253f,
1197   -1.913561f, -0.043716f, -0.214924f, -0.194598f, -0.064723f, -1.671793f,
1198   2.251166f,  -0.146007f, 0.138527f,  -0.003134f, 0.103665f,  0.006928f,
1199   -0.240253f, -0.227464f, 0.578437f,  -0.214724f, 0.503085f,  0.158093f,
1200   0.033091f,  0.008061f,  4.815371f,  2.132264f,  0.281850f,  -2.288560f,
1201   -0.145012f, 1.296832f,  -0.362401f, -0.403252f, 0.109873f,  0.185746f,
1202   0.244764f,  0.172367f,  -0.185588f, 0.139801f,  -0.178254f, 0.068629f,
1203   0.358488f,  -0.153969f, -6.433524f, 0.225983f,  -0.138123f, -0.095971f,
1204   -0.036089f, -1.400083f, 0.265908f,  0.257787f,  0.181144f,  -1.647228f,
1205   -0.136289f, -0.074206f, 0.122988f,  -0.088895f, -1.266717f, 0.006010f,
1206   0.536681f,  0.263061f,  -0.032207f, -0.155136f, 0.086431f,  0.441950f,
1207   -0.060755f, -0.280683f, -0.783475f, -2.567033f, 1.093221f,  0.117667f,
1208   -0.000408f, 0.225719f,  -2.199698f, 0.141447f,  -1.459051f, 0.051315f,
1209   0.203228f,  0.354432f,  -0.005775f, -0.028073f, -0.965817f, 0.231083f,
1210   -0.666884f, 0.026283f,  -0.317486f, 0.210754f,  0.123897f,  0.223827f,
1211   4.214405f,  1.457334f,  -0.253945f, -1.306733f, -0.391235f, 0.451154f,
1212   -1.553888f, -0.353429f, 0.069533f,  0.159278f,  -0.173836f, -0.004952f,
1213   -0.137033f, 0.127012f,  0.143600f,  0.051587f,  -0.070549f, 0.066509f,
1214   -5.776547f, 0.180021f,  -0.189183f, -1.288504f, -0.233575f, -1.473873f,
1215   0.140940f,  0.144451f,  -0.104534f, 2.089873f,  -0.168168f, 0.110726f,
1216   0.132134f,  -0.215223f, -1.682754f, 0.157757f,  -0.146163f, 0.064882f,
1217   0.117313f,  -0.038780f, -0.124720f, -0.501697f, 0.092047f,  -0.233992f,
1218   3.324976f,  0.516601f,  1.294202f,  0.119989f,  0.061055f,  0.043420f,
1219   -2.750727f, -0.382812f, -0.648496f, -0.115353f, -0.334205f, 0.024354f,
1220   -0.282998f, -0.282705f, 0.073798f,  0.169851f,  0.135651f,  0.182677f,
1221   -0.040220f, 0.132462f,  -0.303120f, -0.230113f, 6.165739f,  -0.258596f,
1222   0.024127f,  -1.388283f, -0.006042f, 0.572600f,  0.348411f,  -0.387376f,
1223   -0.075845f, 0.122319f,  -0.029616f, 0.077873f,  0.154763f,  0.049073f,
1224   0.018597f,  0.102688f,  -0.204165f, 0.020734f,  -1.389133f, -0.032854f,
1225   -0.147561f, 0.853944f,  0.132100f,  -3.259659f, 0.243745f,  0.181529f,
1226   -0.738414f, 1.509994f,  0.023470f,  -0.005329f, 0.066115f,  -1.345081f,
1227   -1.455402f, -0.172023f, -0.194625f, 0.071885f,  -0.201742f, -0.262402f,
1228   0.077601f,  -0.048938f, 0.257993f,  -0.504029f, -2.032415f, 1.158880f,
1229   0.448647f,  -0.025633f, 0.117586f,  -0.072275f, -0.673744f, -3.854342f,
1230   -0.983843f, 0.047766f,  -0.017193f, -0.215775f, -0.158743f, -0.232042f,
1231   -0.509112f, 0.148812f,  0.130122f,  0.006486f,  -0.099016f, 0.022514f,
1232   -0.486850f, -0.059623f, 4.012731f,  0.025454f,  0.029059f,  -0.783546f,
1233   -0.295260f, 0.322521f,  -0.473201f, -0.172100f, -0.100087f, -0.076516f,
1234   -0.258367f, -0.112897f, 0.269364f,  -0.065912f, 0.169022f,  -0.178783f,
1235   -0.095114f, 0.122089f,  -2.790099f, -0.100431f, -0.087963f, -0.009431f,
1236   -0.087819f, -2.774399f, -0.100757f, 0.013005f,  -0.964533f, 3.236665f,
1237   -0.354903f, -0.144169f, -0.166869f, -1.396513f, -0.931271f, -0.046261f,
1238   -1.799262f, -0.365269f, 0.108611f,  0.037994f,  0.024747f,  -1.073639f,
1239   -0.203158f, -0.935006f, 1.880891f,  1.578385f,  0.726272f,  -0.024546f,
1240   -0.011626f, -0.151363f, -1.121716f, -1.787484f, 0.232806f,  0.075451f,
1241   0.182899f,  0.092215f,  -0.207347f, -0.030111f, 0.054316f,  0.192481f,
1242   0.594639f,  -0.247694f, 0.547471f,  -0.032094f, -0.065000f, 0.007198f,
1243   1.605377f,  -0.155945f, -0.066200f, -2.343716f, -1.016283f, -0.079321f,
1244   0.919365f,  0.599980f,  0.125545f,  0.265813f,  0.246884f,  0.095385f,
1245   -0.260374f, -0.202916f, -0.042770f, 0.234967f,  -0.233139f, -0.326994f,
1246   -1.375256f, 0.121766f,  0.077433f,  -1.103569f, 0.019497f,  -1.029185f,
1247   0.253905f,  0.206569f,  0.187334f,  -0.237089f, -0.294351f, 0.164137f,
1248   0.149696f,  -0.749787f, -0.413433f, 0.976587f,  1.027976f,  -0.285264f,
1249   0.209273f,  -0.124762f, 0.050884f,  0.250764f,  -0.082031f, -0.646520f,
1250   4.116680f,  0.437336f,  0.671684f,  0.129509f,  -0.078462f, 0.014072f,
1251   -0.678232f, 0.094831f,  1.125624f,  0.207070f,  -0.154750f, -0.025780f,
1252   -0.103030f, 0.118019f,  -0.908186f, -0.263546f, -1.555324f, -0.236887f,
1253   -0.217854f, -0.051790f, 0.017915f,  0.171001f,  1.355562f,  0.094603f,
1254   -0.233929f, -1.282169f, -0.773183f, -0.161682f, -0.834565f, -0.286776f,
1255   -0.298901f, 0.038162f,  0.251899f,  0.039612f,  -0.022935f, -0.232308f,
1256   -0.043855f, -0.192892f, -0.279009f, -0.182234f, -1.272808f, -0.070344f,
1257   -0.092432f, -1.915946f, -0.134373f, -1.405496f, -0.067071f, -0.131922f,
1258   0.185269f,  1.465082f,  0.040240f,  0.112665f,  0.144329f,  -0.286112f,
1259   -0.617649f, 0.916177f,  0.221044f,  -0.079867f, 0.170251f,  -0.093638f,
1260   -0.212620f, -0.305945f, -0.234356f, -0.482501f, 3.928472f,  1.241179f,
1261   0.355922f,  -0.170848f, -0.189168f, 0.080225f,  -1.357793f, 0.190890f,
1262   0.976800f,  -0.068070f, -0.016295f, -0.088623f, -0.129560f, -0.212267f,
1263   -0.071537f, -0.219501f, -0.655198f, -0.225188f, -0.116024f, 0.224174f,
1264   -0.049715f, -0.178005f, 3.029985f,  -1.141546f, 0.080066f,  -1.932316f,
1265   -0.641137f, -0.189564f, 0.935080f,  0.136119f,  0.015558f,  -0.179331f,
1266   0.204571f,  0.020350f,  0.009362f,  0.108478f,  0.037076f,  -0.049009f,
1267   0.081090f,  -0.180202f, 1.455561f,  -0.081559f, 0.059361f,  0.484971f,
1268   0.160923f,  -2.170744f, -0.013204f, 0.126561f,  -0.407122f, 1.223661f,
1269   0.044262f,  0.118044f,  0.058274f,  -1.747100f, -0.171318f, 0.971374f,
1270   0.306995f,  -0.103268f, -0.319443f, -0.333176f, -0.038608f, 0.119674f,
1271   -0.106479f, -0.907933f, 1.121231f,  1.673840f,  -0.421458f, -0.021146f,
1272   -0.254838f, 0.097632f,  0.235109f,  -2.901782f, 0.289518f,  -0.355459f,
1273   -0.068264f, -0.179121f, 0.068560f,  -0.047570f, -0.522523f, -0.228963f,
1274   -1.037158f, -0.163723f, 0.280563f,  -0.000868f, -0.197220f, -0.239329f,
1275   1.985274f,  -0.256181f, -0.064341f, -0.822417f, -0.465140f, -0.010942f,
1276   -0.792024f, -0.114290f, 0.060969f,  0.104106f,  -0.252123f, -0.150400f,
1277   -0.133277f, 0.267147f,  0.274413f,  0.223744f,  -0.180223f, -0.345415f,
1278   -0.104883f, 0.119210f,  -0.095041f, -0.301635f, 0.013175f,  -2.128121f,
1279   -0.147208f, -0.151509f, -0.692013f, 3.418555f,  -0.016541f, 0.171511f,
1280   0.107159f,  -1.516672f, 0.127408f,  0.687035f,  -0.906486f, -0.145463f,
1281   -0.169382f, -0.143906f, 0.125091f,  -0.960645f, -0.180869f, -0.716908f,
1282   2.840951f,  1.904919f,  -0.416268f, -0.425181f, -0.194697f, -0.075932f,
1283   -0.950604f, -1.599800f, 0.943671f,  -0.022744f, -0.270492f, 0.080843f,
1284   -0.372916f, 0.047838f,  -0.100300f, -0.026600f, 0.011733f,  -0.226051f,
1285   0.172790f,  -0.172982f, 0.041258f,  -0.299379f,
1286 };
1287 
1288 static const float av1_ab_partition_nn_bias_16_layer1[LABEL_SIZE] = {
1289   -0.053805f, -1.248639f, 0.520965f, -0.904962f, -0.126425f, -0.118798f,
1290   0.748430f,  0.203096f,  0.059317f, 0.418219f,  0.841294f,  0.402693f,
1291   -0.658522f, 0.723479f,  0.544264f, 1.035225f,
1292 };
1293 
1294 static const NN_CONFIG av1_ab_partition_nnconfig_16 = {
1295   FEATURE_SIZE,  // num_inputs
1296   LABEL_SIZE,    // num_outputs
1297   1,             // num_hidden_layers
1298   {
1299       64,  // num_hidden_nodes
1300   },
1301   {
1302       av1_ab_partition_nn_weights_16_layer0,
1303       av1_ab_partition_nn_weights_16_layer1,
1304   },
1305   {
1306       av1_ab_partition_nn_bias_16_layer0,
1307       av1_ab_partition_nn_bias_16_layer1,
1308   },
1309 };
1310 
1311 #undef FEATURE_SIZE
1312 #undef LABEL_SIZE
1313 
1314 #define FEATURE_SIZE 18
1315 #define LABEL_SIZE 4
1316 
1317 static const float av1_4_partition_nn_weights_16_layer0[FEATURE_SIZE * 24] = {
1318   -2.032866f, 0.056691f,  0.495960f,  0.778785f,  0.548153f,  -0.806942f,
1319   0.481155f,  0.282298f,  0.584980f,  0.504688f,  0.209648f,  0.234616f,
1320   0.213484f,  0.221969f,  0.205862f,  0.235054f,  0.317863f,  0.257139f,
1321   0.529478f,  0.098122f,  -0.657532f, 0.036296f,  0.327728f,  1.323180f,
1322   -0.813082f, 0.160216f,  -0.702030f, 0.722733f,  -0.270576f, -0.347416f,
1323   -0.264700f, -0.254248f, 0.159820f,  0.087995f,  -0.184163f, 0.117357f,
1324   0.074194f,  -0.667369f, 0.498246f,  0.420506f,  0.072409f,  -0.121581f,
1325   0.315788f,  0.000525f,  0.414986f,  0.678166f,  -0.011230f, 0.188131f,
1326   -0.227749f, 0.009564f,  0.108672f,  0.106923f,  -0.080695f, -0.279382f,
1327   -0.061339f, -0.297835f, -0.134707f, 0.145865f,  -0.009655f, -0.000842f,
1328   -0.047436f, -0.159149f, -0.320353f, -0.089646f, -0.344765f, 0.313416f,
1329   -0.143413f, 0.279668f,  0.000885f,  -0.022380f, -0.140194f, -0.310473f,
1330   0.252699f,  0.066204f,  0.477568f,  0.994609f,  -0.276000f, 1.213182f,
1331   0.277028f,  -0.411570f, -0.211559f, 0.377815f,  0.121488f,  -0.100559f,
1332   -0.317082f, -0.251039f, -0.335181f, -0.154114f, -0.052726f, -0.332558f,
1333   -0.143196f, -0.334035f, 0.162305f,  0.142279f,  -0.001210f, -0.135252f,
1334   -0.033562f, 0.204307f,  -0.039757f, -0.394174f, 0.126617f,  -0.128648f,
1335   -0.410979f, 0.107641f,  -0.117573f, -0.326512f, 0.235166f,  0.084959f,
1336   0.290063f,  -0.005838f, 0.459894f,  1.023709f,  -0.196145f, 1.100137f,
1337   -0.319815f, -0.308526f, -0.443389f, -0.272769f, -0.035259f, -0.026932f,
1338   -0.029743f, 0.125113f,  -0.131024f, -0.321458f, -0.143996f, 0.008714f,
1339   -0.101234f, 0.079706f,  -1.128615f, -0.467381f, 0.220563f,  -0.409900f,
1340   -0.435353f, 0.759499f,  -0.465799f, -0.394309f, 0.176282f,  -0.086275f,
1341   -0.161225f, -0.354814f, 0.562871f,  0.418253f,  0.414361f,  0.445480f,
1342   -0.995903f, -0.086632f, -0.230645f, 0.354656f,  -0.317576f, 0.079926f,
1343   0.424369f,  0.997232f,  -0.304388f, 1.071667f,  -0.023540f, 0.029677f,
1344   0.108564f,  0.183581f,  -0.201395f, -0.054854f, -0.193039f, -0.049899f,
1345   -0.271949f, -0.358483f, 0.304930f,  0.023823f,  -0.009319f, -0.214247f,
1346   0.100712f,  -0.050162f, 0.327103f,  -0.212999f, -0.030496f, 0.316380f,
1347   -0.439589f, -0.249959f, 0.229777f,  -0.353664f, -0.384559f, 0.114236f,
1348   0.023119f,  0.007927f,  0.618368f,  0.957759f,  -0.019780f, -1.002389f,
1349   0.564277f,  -0.839531f, 1.040445f,  0.054340f,  0.031908f,  -0.032893f,
1350   -0.019170f, -0.042011f, 0.568928f,  0.362567f,  -0.559999f, -0.605344f,
1351   -0.586146f, -0.290778f, 0.195943f,  -0.109580f, -0.088898f, -0.113054f,
1352   0.293282f,  0.429019f,  0.306136f,  0.863025f,  0.021234f,  0.125770f,
1353   -0.097108f, -0.072659f, -0.137053f, -0.191631f, 0.106281f,  0.064151f,
1354   0.029883f,  0.076287f,  0.757543f,  0.276713f,  -2.529775f, -0.351727f,
1355   -1.832316f, 0.544780f,  -0.944529f, 0.509705f,  -0.010236f, -0.016181f,
1356   0.021520f,  0.086417f,  0.041312f,  0.296853f,  -0.372378f, 0.354446f,
1357   -1.366762f, 0.048875f,  0.464918f,  -0.007450f, 0.750013f,  -0.360261f,
1358   0.518532f,  0.753776f,  0.641448f,  0.710746f,  0.250866f,  0.257063f,
1359   0.283421f,  0.253585f,  0.170303f,  0.210426f,  0.208842f,  0.158000f,
1360   -0.033144f, 0.130748f,  0.907147f,  0.409248f,  -0.854301f, -0.981307f,
1361   0.294427f,  -0.507137f, 1.079967f,  0.203203f,  0.383890f,  0.368278f,
1362   0.305122f,  0.449288f,  -0.044507f, -0.547263f, -0.298245f, -0.497834f,
1363   0.007016f,  -0.101982f, -0.073488f, -0.096111f, -0.479418f, -0.045497f,
1364   0.033502f,  -0.018578f, -0.231531f, 0.177949f,  0.099564f,  -0.010233f,
1365   -0.333055f, -0.078586f, -0.417867f, 0.171271f,  0.013662f,  -0.143599f,
1366   -0.117296f, 0.135382f,  0.048321f,  0.000924f,  -0.055024f, -0.405595f,
1367   -0.068260f, -0.271011f, -0.436425f, 0.206751f,  -0.899890f, 0.605510f,
1368   0.535649f,  -0.238919f, -0.037619f, -0.213734f, -0.391360f, -0.132344f,
1369   0.004660f,  0.176644f,  -1.008475f, -0.038895f, 0.155429f,  -0.095229f,
1370   -0.680124f, -0.258063f, -0.261901f, 0.110380f,  -0.337649f, -0.505870f,
1371   -1.428536f, 0.610629f,  0.254905f,  0.045098f,  0.044109f,  0.172329f,
1372   0.060001f,  -0.234009f, -0.184855f, -0.153028f, -0.140897f, -0.152006f,
1373   -0.312134f, 0.081261f,  0.160166f,  0.112690f,  0.266081f,  0.030175f,
1374   -0.242746f, 0.000754f,  -0.341811f, -0.149774f, -0.017484f, -0.301342f,
1375   -0.121466f, 0.067300f,  0.342176f,  0.474538f,  0.085441f,  -0.263935f,
1376   0.479235f,  -0.003713f, -0.784840f, 0.119480f,  0.456632f,  -0.640082f,
1377   -0.080575f, -0.744403f, 0.259970f,  0.034667f,  -0.274641f, -0.257594f,
1378   -1.121124f, -0.003745f, -0.420693f, 0.300441f,  -0.100976f, -1.049016f,
1379   0.201960f,  0.113054f,  0.187010f,  1.237427f,  0.054803f,  -0.028673f,
1380   0.003596f,  -0.034724f, 0.117246f,  0.190977f,  0.278915f,  0.224307f,
1381   0.017852f,  -0.336233f, -0.372311f, -0.182284f, -0.143510f, 0.331466f,
1382   0.045698f,  -0.301095f, 0.184447f,  0.348240f,  -0.017021f, -0.145064f,
1383   -0.000221f, -0.382256f, -0.302683f, -0.083927f, -0.008070f, 0.217907f,
1384   0.647597f,  -0.050490f, -0.572736f, -0.985748f, -0.289943f, 0.041391f,
1385   -0.795464f, -0.186680f, -0.354062f, -0.617400f, -0.282783f, -0.170450f,
1386   -0.197197f, -0.146496f, -0.173692f, -0.106277f, -0.071004f, -0.124405f,
1387   -0.971412f, 0.038542f,  0.705204f,  0.887113f,  0.150430f,  -0.243676f,
1388   0.638410f,  0.320953f,  0.776676f,  0.527584f,  0.070389f,  0.051554f,
1389   0.177519f,  0.140451f,  0.128892f,  0.087771f,  0.197660f,  0.194764f,
1390 };
1391 
1392 static const float av1_4_partition_nn_bias_16_layer0[24] = {
1393   0.614063f,  -0.384872f, 0.084884f,  -0.023980f, -0.378765f, -0.082312f,
1394   -0.458271f, 0.189578f,  -0.046169f, -0.073308f, -0.372322f, 0.162793f,
1395   0.148803f,  0.829214f,  -0.221162f, -0.111157f, -0.017484f, -0.280596f,
1396   -0.031905f, -0.143459f, 0.078823f,  -0.021940f, 0.026834f,  0.257472f,
1397 };
1398 
1399 static const float av1_4_partition_nn_weights_16_layer1[24 * LABEL_SIZE] = {
1400   -0.985391f, 0.587616f,  0.740683f,  0.192066f,  0.447080f,  -0.016585f,
1401   0.680449f,  0.028983f,  0.643111f,  0.234338f,  0.107148f,  0.328456f,
1402   -0.216394f, 1.106838f,  -0.179062f, -0.129108f, -0.121655f, -0.151340f,
1403   -0.306017f, -0.350989f, 0.859284f,  -0.372831f, -0.954419f, 0.250495f,
1404   1.046732f,  0.287923f,  -0.421088f, 0.326613f,  -0.314396f, -0.084757f,
1405   -0.474228f, 0.687999f,  0.052334f,  0.441708f,  -0.630698f, -0.350348f,
1406   -0.602067f, -0.434161f, -0.489824f, -0.313193f, 0.315568f,  0.603119f,
1407   0.120245f,  0.182920f,  -1.117797f, -0.239594f, -0.296296f, -0.718093f,
1408   0.489497f,  -0.527019f, 0.102453f,  0.426731f,  0.034606f,  0.311461f,
1409   -0.012723f, -0.229877f, -0.284290f, 0.383227f,  0.065696f,  -0.222400f,
1410   1.279248f,  -0.862190f, 0.629766f,  -0.250011f, -0.325060f, -0.360115f,
1411   -0.159540f, -0.291856f, -0.038348f, 0.224639f,  0.600934f,  0.030205f,
1412   1.337615f,  -0.286409f, -0.473710f, -0.418995f, -1.035249f, 0.004359f,
1413   -0.481860f, 0.563625f,  -0.154709f, -0.101198f, -0.758796f, -0.507616f,
1414   -0.095253f, -0.711135f, 0.207759f,  0.076313f,  -0.056087f, -0.162719f,
1415   -0.232918f, -0.128402f, -0.444620f, -0.447344f, 1.126012f,  -1.504446f,
1416 };
1417 
1418 static const float av1_4_partition_nn_bias_16_layer1[LABEL_SIZE] = {
1419   -0.462133f,
1420   0.465060f,
1421   0.062211f,
1422   0.401786f,
1423 };
1424 
1425 static const NN_CONFIG av1_4_partition_nnconfig_16 = {
1426   FEATURE_SIZE,  // num_inputs
1427   LABEL_SIZE,    // num_outputs
1428   1,             // num_hidden_layers
1429   {
1430       24,  // num_hidden_nodes
1431   },
1432   {
1433       av1_4_partition_nn_weights_16_layer0,
1434       av1_4_partition_nn_weights_16_layer1,
1435   },
1436   {
1437       av1_4_partition_nn_bias_16_layer0,
1438       av1_4_partition_nn_bias_16_layer1,
1439   },
1440 };
1441 
1442 static const float av1_4_partition_nn_weights_32_layer0[FEATURE_SIZE * 32] = {
1443   -0.219494f, -0.428273f, 0.471006f,  0.448210f,  -0.152935f, 0.440435f,
1444   0.922857f,  -0.074436f, 1.002195f,  0.414176f,  -0.327202f, -0.380066f,
1445   -0.212346f, 0.061868f,  -0.056620f, 0.594134f,  0.617995f,  0.308358f,
1446   0.232484f,  0.129849f,  1.483593f,  -0.071460f, 1.984515f,  1.116422f,
1447   -1.141762f, -0.306220f, 0.089075f,  -0.271845f, 0.187524f,  0.050396f,
1448   -0.061025f, 0.030809f,  0.172799f,  -0.458151f, -0.318357f, 0.122052f,
1449   -0.414329f, 0.089366f,  0.118898f,  -0.376213f, -0.206151f, -0.519946f,
1450   -0.463252f, -0.206694f, -0.254383f, -0.379487f, 0.093059f,  -0.245280f,
1451   -0.205044f, -0.280060f, -0.171229f, -0.045389f, -0.179481f, -0.306245f,
1452   -0.500856f, 0.003388f,  -0.527397f, -0.449330f, -0.174272f, 0.123769f,
1453   0.023005f,  0.157273f,  0.073400f,  0.019099f,  -0.113848f, -0.098601f,
1454   -0.290946f, -0.046770f, -0.314592f, -0.179914f, -0.391411f, -0.235631f,
1455   -1.282604f, 0.048505f,  -0.746382f, 0.093740f,  -0.706583f, -0.085729f,
1456   0.947382f,  -0.002961f, 1.175362f,  1.007309f,  0.141638f,  -0.037608f,
1457   -0.118807f, -0.021474f, -0.146763f, 0.069363f,  -0.074372f, -0.215713f,
1458   -0.004134f, -0.114110f, -0.330438f, -0.031136f, 0.111821f,  -0.534598f,
1459   -0.357759f, -0.455950f, 0.139469f,  0.036582f,  -0.384743f, -0.168828f,
1460   -0.239250f, 0.003520f,  -0.049003f, 0.075702f,  -0.025809f, -0.225972f,
1461   -0.228905f, -0.412489f, 0.060570f,  -0.328819f, -0.206446f, -0.080231f,
1462   -0.372008f, -0.218118f, -0.011954f, 0.024155f,  0.156014f,  0.020679f,
1463   0.194398f,  -0.283491f, -0.024463f, -0.275099f, 0.028031f,  0.026340f,
1464   -0.254668f, 0.103637f,  2.178693f,  0.552284f,  0.109366f,  -0.474806f,
1465   -0.379286f, -0.026315f, 2.487924f,  -0.089466f, 0.206428f,  0.114578f,
1466   0.152248f,  0.184050f,  -0.631948f, -0.014793f, -0.283782f, -0.830353f,
1467   0.009343f,  -0.021029f, -0.060534f, -0.025164f, 1.841311f,  1.842748f,
1468   -1.979708f, 0.450985f,  -1.606357f, -0.785454f, -0.212679f, -0.344342f,
1469   0.198991f,  -0.258070f, 0.055974f,  0.224069f,  0.453051f,  0.408053f,
1470   0.027873f,  -0.180538f, 0.056609f,  0.207654f,  0.104086f,  -0.194426f,
1471   -0.359789f, -0.381143f, -0.331212f, -0.203973f, -0.324313f, -0.160825f,
1472   -0.160439f, -0.044856f, -0.346647f, 0.044859f,  0.231398f,  -0.023643f,
1473   -0.140316f, -0.260177f, 0.206965f,  -0.425386f, -0.420268f, -0.409748f,
1474   0.006971f,  0.066186f,  -0.034950f, -0.345518f, 0.018633f,  -0.122489f,
1475   -0.038506f, -0.330942f, 0.161236f,  -0.314119f, -0.050202f, -0.179597f,
1476   0.731897f,  -0.184481f, 0.153598f,  -0.539501f, -0.301493f, -0.184967f,
1477   -0.883754f, -0.586959f, -0.136292f, -1.772065f, -0.196276f, -0.053272f,
1478   -0.101083f, -0.064142f, 0.161190f,  0.430826f,  0.355647f,  0.138266f,
1479   0.051114f,  -0.028893f, -0.477673f, -0.238663f, -0.354117f, -0.056747f,
1480   -0.334273f, -0.497688f, -0.486004f, -0.092033f, -0.241304f, -0.373250f,
1481   0.120193f,  0.011360f,  -0.010475f, -0.092739f, -0.159650f, -0.033129f,
1482   -0.259893f, -0.073217f, 0.200128f,  0.103407f,  -0.229233f, 0.128831f,
1483   -0.063450f, -0.241732f, -0.408428f, -0.342239f, -0.264326f, -0.105403f,
1484   -0.442879f, -0.310456f, -0.112881f, 0.263696f,  -0.205014f, -0.497936f,
1485   -0.261734f, -0.382312f, -0.426807f, -0.021995f, -0.152794f, -0.301494f,
1486   0.117232f,  -0.577809f, 0.154596f,  -0.409522f, -0.413113f, -0.359199f,
1487   0.307294f,  -0.008746f, -0.310522f, 0.347620f,  -0.384845f, -0.451398f,
1488   -0.226199f, 0.054154f,  -0.167608f, 0.046836f,  -0.013285f, -0.408119f,
1489   -0.177973f, -0.248293f, -0.465830f, 0.035827f,  -0.222208f, -0.221717f,
1490   0.066392f,  -0.349769f, -0.428029f, -0.516692f, 0.022398f,  -0.251682f,
1491   0.134746f,  0.011167f,  -2.078787f, 0.173592f,  -1.948348f, 0.330060f,
1492   1.993785f,  -0.052859f, -0.004795f, -3.703177f, 0.013450f,  -0.011687f,
1493   0.073079f,  0.034803f,  0.025515f,  0.005994f,  0.101731f,  0.074303f,
1494   -0.109962f, -0.270825f, -0.068273f, -0.163268f, -0.252826f, 0.137190f,
1495   0.007667f,  -0.358453f, 0.027412f,  0.033492f,  0.021197f,  -0.049991f,
1496   0.104468f,  -0.012157f, -0.056252f, -0.380756f, -0.338483f, 0.233235f,
1497   -0.048631f, -0.441209f, -0.158482f, -0.148108f, -0.263453f, 0.138847f,
1498   -0.304073f, -0.336312f, -0.017941f, -0.135563f, 0.075137f,  -0.246475f,
1499   -0.229144f, -0.087744f, -0.346909f, 0.172611f,  0.004377f,  -0.009386f,
1500   -0.023104f, 0.008000f,  -0.029390f, -0.317842f, 0.549674f,  -0.195337f,
1501   -0.863979f, 0.160889f,  -0.269014f, -0.442104f, -1.799191f, 1.396533f,
1502   -0.112837f, 0.881303f,  0.000764f,  -0.035415f, -0.141877f, 0.184831f,
1503   -0.363566f, -0.178569f, 0.254134f,  -0.326893f, 0.127325f,  0.310620f,
1504   -0.384621f, 0.146058f,  -0.287682f, -0.373447f, 0.026930f,  0.251650f,
1505   0.053817f,  0.227509f,  0.121396f,  0.396514f,  -0.278381f, -0.038969f,
1506   -1.538756f, -0.002856f, -0.892900f, 0.363426f,  -1.257922f, 0.743795f,
1507   0.941177f,  0.219345f,  0.684189f,  1.396858f,  0.026299f,  -0.093433f,
1508   -0.066182f, 0.057868f,  -0.089278f, -0.159680f, -0.262035f, -0.236656f,
1509   0.005349f,  -0.031314f, 0.027917f,  -0.182113f, -0.212086f, -0.160774f,
1510   0.051468f,  0.036787f,  0.183881f,  -0.288205f, -0.349691f, 0.162511f,
1511   0.117878f,  -0.294534f, -0.365037f, -0.246313f, 0.073977f,  -0.072378f,
1512   -0.173579f, -0.584560f, 0.547194f,  0.259853f,  -0.405287f, -0.421146f,
1513   0.165788f,  -0.146964f, 0.257415f,  0.772394f,  -0.475302f, -0.310906f,
1514   0.058723f,  0.276833f,  0.586842f,  0.248998f,  -0.061135f, 0.255779f,
1515   0.152158f,  -0.024781f, 2.821834f,  1.365141f,  0.914744f,  0.165752f,
1516   -1.048304f, -0.333891f, 1.804087f,  -0.437028f, -0.120211f, -0.020443f,
1517   0.040077f,  0.258600f,  -0.598893f, -0.494579f, -0.281054f, -0.517041f,
1518   0.005258f,  0.053986f,  0.322755f,  0.429495f,  -1.992364f, -0.717192f,
1519   -1.774802f, 2.047362f,  -0.016194f, 0.312606f,  0.019331f,  0.060950f,
1520   0.116428f,  0.168458f,  -0.307001f, -0.420734f, 0.475843f,  0.425346f,
1521   -0.107119f, 0.049892f,  -1.168619f, 0.010878f,  0.354872f,  0.902717f,
1522   -0.391407f, 0.332772f,  -1.335037f, -0.447100f, 0.481719f,  -0.101069f,
1523   -1.806565f, 0.925280f,  0.346999f,  0.093809f,  0.006275f,  0.270814f,
1524   -0.691123f, 0.230748f,  0.137033f,  0.068228f,  1.555975f,  -0.271637f,
1525   -0.370403f, 0.236131f,  0.367464f,  -0.136562f, 0.428838f,  0.181750f,
1526   0.338762f,  0.292449f,  -0.748204f, -0.922731f, -0.959445f, -0.806418f,
1527   -0.140501f, 0.070525f,  1.248748f,  0.637990f,  -1.307246f, -0.514055f,
1528   0.393858f,  -1.858727f, 0.713591f,  -0.141044f, 0.080723f,  0.120220f,
1529   -0.031175f, 0.224488f,  0.753818f,  -0.833351f, -1.099132f, 0.651100f,
1530   -0.135061f, -0.043820f, 0.026983f,  -0.059259f, 0.001345f,  -0.281775f,
1531   0.006958f,  0.046103f,  -0.246539f, 0.057630f,  -0.360778f, -0.160681f,
1532   -0.414870f, -0.301979f, 0.000683f,  0.132957f,  -0.477609f, 0.106110f,
1533   -0.637769f, -0.078374f, -0.229494f, 0.583108f,  -0.822973f, -0.107540f,
1534   1.063426f,  -0.268346f, 1.105787f,  2.587550f,  -0.020314f, -0.002161f,
1535   -0.063836f, -0.099990f, -0.103975f, -0.114078f, -0.094199f, -0.065181f,
1536   -0.019870f, -0.018920f, -0.219732f, 0.035608f,  -1.789450f, 0.483032f,
1537   -0.464729f, 1.563277f,  -1.054195f, 0.359991f,  0.065204f,  0.135623f,
1538   0.158380f,  -0.103815f, -1.398726f, -1.436666f, -0.356311f, 0.507752f,
1539 };
1540 
1541 static const float av1_4_partition_nn_bias_32_layer0[32] = {
1542   0.421645f,  -0.620548f, -0.187819f, -0.189414f, -0.204975f, -0.189600f,
1543   -0.174917f, -0.651928f, -0.799655f, -0.086105f, -0.163449f, -0.089212f,
1544   -0.214495f, -0.108500f, -0.065777f, -0.127704f, 1.544948f,  -0.032831f,
1545   -0.165621f, 0.145844f,  -0.032104f, -0.453246f, -0.113444f, 0.321589f,
1546   -0.862375f, -0.108826f, -0.486259f, 0.685325f,  0.072569f,  -0.187961f,
1547   0.109579f,  -0.082685f,
1548 };
1549 
1550 static const float av1_4_partition_nn_weights_32_layer1[32 * LABEL_SIZE] = {
1551   0.255012f,  0.658860f,  0.216907f,  0.165947f,  0.241182f,  0.340854f,
1552   0.409445f,  0.165220f,  0.553373f,  -0.242385f, -0.209571f, 0.255515f,
1553   0.222500f,  0.037032f,  0.238590f,  0.061624f,  -2.038693f, 0.264167f,
1554   -0.230144f, 0.129952f,  -0.027979f, 0.847761f,  0.438922f,  0.462323f,
1555   0.555345f,  0.030689f,  0.336357f,  -0.357326f, -0.113137f, 0.272631f,
1556   0.421022f,  0.367776f,  -0.197094f, 0.157117f,  -0.015008f, -0.056123f,
1557   -0.283913f, 0.186417f,  0.178561f,  -0.763041f, 0.602038f,  0.341092f,
1558   0.320453f,  -0.312776f, -0.371240f, -0.356279f, 0.220117f,  -0.131871f,
1559   1.517429f,  0.162223f,  -0.255069f, 0.451861f,  0.045071f,  -0.223257f,
1560   0.003257f,  0.015734f,  -0.630447f, -0.672588f, 0.670164f,  0.571031f,
1561   -0.657948f, 0.034506f,  -0.249076f, 0.790293f,  0.066491f,  -0.131245f,
1562   0.355173f,  0.564622f,  0.374048f,  0.033974f,  0.253970f,  0.495498f,
1563   -0.556321f, -0.104651f, 0.276947f,  0.057148f,  -0.039126f, -0.170050f,
1564   -0.141542f, 0.158541f,  0.582763f,  -0.100992f, 0.096705f,  -0.209029f,
1565   0.008449f,  0.255865f,  0.103565f,  0.317719f,  0.479499f,  0.599126f,
1566   -0.065613f, -0.268614f, 0.508736f,  0.180813f,  -0.815868f, 0.051238f,
1567   0.001223f,  -0.305423f, -0.270079f, 0.036180f,  0.304342f,  0.202634f,
1568   0.218348f,  -0.304304f, -0.438297f, 0.241123f,  0.200230f,  0.151804f,
1569   0.051944f,  0.160422f,  -0.262981f, -0.417412f, 1.845729f,  -0.086183f,
1570   0.403517f,  0.059667f,  0.564543f,  -0.081752f, 0.114907f,  -0.284489f,
1571   -0.673943f, 0.056965f,  0.362221f,  0.403224f,  -0.000233f, -0.209552f,
1572   -0.800926f, -0.134132f,
1573 };
1574 
1575 static const float av1_4_partition_nn_bias_32_layer1[LABEL_SIZE] = {
1576   -0.019518f,
1577   0.198546f,
1578   0.339015f,
1579   -0.261961f,
1580 };
1581 
1582 static const NN_CONFIG av1_4_partition_nnconfig_32 = {
1583   FEATURE_SIZE,  // num_inputs
1584   LABEL_SIZE,    // num_outputs
1585   1,             // num_hidden_layers
1586   {
1587       32,  // num_hidden_nodes
1588   },
1589   {
1590       av1_4_partition_nn_weights_32_layer0,
1591       av1_4_partition_nn_weights_32_layer1,
1592   },
1593   {
1594       av1_4_partition_nn_bias_32_layer0,
1595       av1_4_partition_nn_bias_32_layer1,
1596   },
1597 };
1598 
1599 static const float av1_4_partition_nn_weights_64_layer0[FEATURE_SIZE * 24] = {
1600   -0.152649f, 0.074509f,  1.000136f,  0.601661f,  -1.416694f, -1.932396f,
1601   -1.163850f, 0.640931f,  -0.888625f, -0.345711f, 0.161799f,  0.103165f,
1602   0.147513f,  0.089956f,  0.204329f,  0.196922f,  0.014927f,  0.283714f,
1603   -0.110422f, 0.062005f,  -0.531870f, -0.075287f, -0.448349f, -0.218881f,
1604   -0.005592f, -0.130490f, -0.015779f, 0.093521f,  -0.158487f, 0.072241f,
1605   0.066879f,  -0.418566f, -0.206281f, 0.025634f,  0.048334f,  -0.534750f,
1606   0.302081f,  0.028707f,  -1.543248f, 0.103799f,  -1.214052f, 0.395870f,
1607   0.394754f,  -0.272170f, -0.702953f, -4.057464f, -0.033497f, -0.042142f,
1608   0.014742f,  0.065263f,  0.000879f,  -0.019768f, 0.101275f,  0.163059f,
1609   -0.371392f, -0.283484f, 0.241915f,  0.012684f,  -0.210101f, -0.166534f,
1610   -0.024894f, 0.274696f,  0.098993f,  0.104086f,  0.055044f,  -0.289378f,
1611   0.146571f,  -0.147441f, 0.004056f,  0.112244f,  -0.416162f, -0.033176f,
1612   -0.214836f, -0.213787f, 0.023197f,  -0.339043f, 0.301109f,  -0.408551f,
1613   0.284922f,  -0.344418f, -0.039255f, 0.158748f,  -0.344169f, 0.078286f,
1614   -0.043957f, -0.302162f, -0.310826f, 0.063425f,  0.198166f,  -0.285324f,
1615   -0.108252f, 0.038992f,  -1.053110f, -1.663290f, -0.417185f, 1.504443f,
1616   0.643206f,  -0.850240f, 0.889641f,  -0.733214f, 0.147302f,  0.060291f,
1617   -0.052954f, 0.167453f,  0.111870f,  0.085471f,  0.035107f,  0.064361f,
1618   0.176053f,  0.184373f,  0.676576f,  0.066164f,  1.455569f,  0.925111f,
1619   -0.640845f, 0.803795f,  -0.653782f, -0.201038f, 0.060033f,  0.016964f,
1620   -0.047590f, 0.045908f,  0.354162f,  0.014812f,  0.156978f,  0.058792f,
1621   -0.238119f, 0.002450f,  -0.094388f, -0.155229f, 0.194858f,  -0.355429f,
1622   -0.187098f, -0.119264f, -0.088694f, -0.102845f, 0.184905f,  -0.425339f,
1623   -0.157808f, -0.104599f, -0.393248f, -0.379842f, 0.027741f,  -0.185816f,
1624   -0.317294f, 0.002453f,  -0.498241f, -0.204302f, -0.079093f, 0.020646f,
1625   -0.412850f, -0.426039f, -0.177050f, -0.419304f, -0.064478f, -0.191802f,
1626   -0.146812f, 0.171111f,  0.090261f,  -0.367033f, -0.299051f, -0.322132f,
1627   0.428192f,  -0.252613f, 0.488498f,  -0.559682f, 0.486720f,  -0.511084f,
1628   0.992506f,  0.346765f,  -0.118697f, -0.065127f, -0.376612f, -0.345137f,
1629   -0.426517f, -0.516836f, 0.307083f,  0.609362f,  0.369555f,  0.093775f,
1630   -0.375664f, -0.221595f, -0.025465f, 0.134374f,  -0.387031f, 0.096236f,
1631   0.337465f,  -0.124029f, -0.157340f, -0.368790f, -0.104490f, -0.279507f,
1632   -0.247705f, 0.146559f,  -0.236206f, -0.036073f, 0.064206f,  -0.330919f,
1633   0.516591f,  -0.013492f, 1.269568f,  1.182530f,  -0.455390f, -1.328091f,
1634   -0.200950f, -0.380513f, -0.195532f, -0.341479f, 0.016064f,  0.021176f,
1635   0.169119f,  0.103707f,  -0.174504f, -0.462719f, -0.079445f, -0.247128f,
1636   0.459111f,  0.036129f,  0.769570f,  -0.080405f, 1.667107f,  0.355567f,
1637   -2.433896f, 0.627572f,  -0.600090f, -0.651872f, -0.059769f, -0.041945f,
1638   -0.009933f, 0.014864f,  -0.049378f, -0.041561f, 0.075180f,  0.138307f,
1639   0.122366f,  -0.160756f, 0.215327f,  0.013572f,  0.198194f,  -0.762650f,
1640   0.054466f,  1.110332f,  1.692853f,  0.658654f,  -0.409549f, 0.506085f,
1641   0.330962f,  -0.223008f, 0.007448f,  -0.289062f, -0.476231f, -0.228359f,
1642   0.013977f,  -0.000609f, -0.673604f, 0.275996f,  0.405291f,  1.693561f,
1643   -1.079768f, 1.122516f,  -0.203227f, 0.099265f,  -0.165207f, -0.323899f,
1644   -0.269973f, -0.080122f, 0.127700f,  0.190201f,  0.219527f,  0.306194f,
1645   0.026049f,  -0.003779f, 1.107357f,  1.720315f,  1.017908f,  0.078664f,
1646   -1.599813f, -0.482636f, -0.117450f, 0.122249f,  0.030220f,  0.039794f,
1647   0.176350f,  0.129715f,  -0.305755f, -0.274044f, -0.299640f, -0.187335f,
1648   -0.073616f, -0.564507f, -0.127758f, 0.044855f,  -0.191090f, 0.039095f,
1649   0.115378f,  0.969352f,  -0.088360f, 0.301443f,  0.065726f,  -0.019740f,
1650   -0.102350f, -0.084913f, -0.194615f, 0.118582f,  0.920789f,  -0.171615f,
1651   -1.436553f, -0.026419f, -0.730864f, 0.615697f,  -0.795079f, 0.119701f,
1652   0.601782f,  0.792902f,  0.184920f,  1.635090f,  -0.085860f, -0.033187f,
1653   -0.166883f, 0.008487f,  -0.128300f, -0.089923f, -0.108781f, -0.133719f,
1654   -0.011988f, -0.239816f, -0.092563f, -0.238471f, -0.339722f, 0.177432f,
1655   -0.063101f, -0.121002f, 0.058072f,  -0.031166f, 0.086413f,  -0.016203f,
1656   -0.305075f, -0.005420f, -0.168796f, 0.148745f,  -0.116737f, -0.050222f,
1657   -0.287952f, -0.290982f, -0.090449f, 0.076098f,  -0.345632f, -0.061309f,
1658   0.142218f,  0.035692f,  0.304517f,  -0.228031f, 0.119608f,  -0.120350f,
1659   0.163404f,  -0.105605f, -0.305462f, -0.176657f, 0.210070f,  -0.227600f,
1660   -0.081965f, -0.464027f, -0.053782f, -0.018367f, 0.119159f,  0.017162f,
1661   -0.069792f, 0.305768f,  -0.421095f, 0.187740f,  -0.032059f, 0.575115f,
1662   -0.064283f, -0.091828f, 0.772648f,  -0.393189f, -0.297098f, 0.141420f,
1663   0.826389f,  -0.071586f, -0.893968f, -0.346793f, -1.151655f, 0.039393f,
1664   1.546000f,  -0.094029f, -0.005786f, -0.195764f, -0.169724f, -0.133167f,
1665   -0.129312f, -0.418860f, -0.026553f, -0.053667f, -0.091976f, -0.106275f,
1666   -0.492625f, 0.025350f,  -0.332075f, -0.475638f, -0.076667f, -0.065779f,
1667   0.108957f,  0.246298f,  -0.289007f, -0.442552f, -0.206692f, -0.257453f,
1668   0.073806f,  -0.458606f, -0.410390f, -0.312674f, -0.144813f, 0.170128f,
1669   0.018810f,  -0.098241f, 1.027369f,  0.479328f,  1.129707f,  0.484813f,
1670   -0.085207f, 0.621873f,  -0.520981f, 0.236175f,  0.273487f,  0.061426f,
1671   0.306085f,  0.161487f,  0.220991f,  0.223783f,  -0.091826f, 0.391031f,
1672 };
1673 
1674 static const float av1_4_partition_nn_bias_64_layer0[24] = {
1675   0.580225f,  -0.191304f, 1.091767f,  -0.134522f, -0.089361f, 0.398750f,
1676   -0.882708f, -0.213102f, -0.119981f, 0.378296f,  -0.075719f, 0.426598f,
1677   -2.015505f, 0.202534f,  -1.044792f, -0.841519f, 0.266421f,  -0.047115f,
1678   -0.131147f, -0.075066f, -0.009441f, 0.853007f,  -0.175606f, -0.868306f,
1679 };
1680 
1681 static const float av1_4_partition_nn_weights_64_layer1[24 * LABEL_SIZE] = {
1682   -0.851937f, -0.211148f, -2.289513f, -0.275071f, 0.251340f,  -0.340847f,
1683   0.498032f,  0.308652f,  -0.051574f, 0.323146f,  -0.097547f, -0.040269f,
1684   1.909655f,  0.098348f,  0.588136f,  0.568112f,  0.313297f,  0.920848f,
1685   -0.014486f, 0.386014f,  0.029199f,  -0.537330f, -0.021502f, 0.349073f,
1686   -0.524715f, -0.351848f, 1.565454f,  -0.297148f, 0.020177f,  0.648369f,
1687   0.027321f,  -0.096052f, -0.363163f, -0.132642f, 0.024292f,  -0.734176f,
1688   -0.782700f, 0.408299f,  0.476945f,  -0.489512f, -0.728318f, -0.632042f,
1689   0.405417f,  0.184086f,  -0.400730f, 0.359032f,  0.019710f,  -0.217409f,
1690   0.519159f,  -0.136316f, 0.993592f,  -0.147128f, 0.097495f,  0.426189f,
1691   -0.295233f, 0.278799f,  0.080667f,  -0.025052f, -0.307757f, 0.418716f,
1692   -0.853388f, -0.374878f, -0.322725f, 0.696335f,  -0.380649f, -0.160356f,
1693   -0.140060f, 0.502455f,  0.656728f,  -0.095023f, -0.184198f, -0.347069f,
1694   0.456372f,  -0.029754f, 0.907923f,  0.265710f,  -0.065505f, 0.226763f,
1695   -0.277798f, 0.413292f,  -0.593899f, -0.060740f, -0.313358f, -0.249944f,
1696   -0.627329f, -0.327151f, -0.853788f, -1.163807f, -0.388944f, -0.228788f,
1697   -0.057382f, 0.334741f,  -0.283083f, 0.368280f,  -0.407197f, -0.441849f,
1698 };
1699 
1700 static const float av1_4_partition_nn_bias_64_layer1[LABEL_SIZE] = {
1701   -0.478735f,
1702   0.292948f,
1703   0.293172f,
1704   0.040013f,
1705 };
1706 
1707 static const NN_CONFIG av1_4_partition_nnconfig_64 = {
1708   FEATURE_SIZE,  // num_inputs
1709   LABEL_SIZE,    // num_outputs
1710   1,             // num_hidden_layers
1711   {
1712       24,  // num_hidden_nodes
1713   },
1714   {
1715       av1_4_partition_nn_weights_64_layer0,
1716       av1_4_partition_nn_weights_64_layer1,
1717   },
1718   {
1719       av1_4_partition_nn_bias_64_layer0,
1720       av1_4_partition_nn_bias_64_layer1,
1721   },
1722 };
1723 
1724 #undef FEATURE_SIZE
1725 #undef LABEL_SIZE
1726 
1727 #define FEATURE_SIZE 4
1728 static const float
1729     av1_partition_breakout_nn_weights_128_layer0[FEATURE_SIZE * 32] = {
1730       -0.331785f,  0.068675f,  -0.323814f,  0.033714f,  -0.237835f, 0.166316f,
1731       -0.498766f,  -0.545634f, -0.266173f,  -0.476957f, -0.120409f, -0.021042f,
1732       0.124056f,   -0.278750f, -0.110120f,  -0.372812f, 4.547939f,  0.097618f,
1733       -0.002710f,  -0.064169f, -1.841173f,  -0.403833f, 0.005536f,  0.067188f,
1734       -0.434935f,  -0.227421f, -0.000011f,  -0.139961f, -0.174056f, -0.652384f,
1735       -0.000015f,  -0.262847f, -3.319706f,  -0.947693f, 0.002981f,  0.016717f,
1736       -10.408850f, -0.014568f, -0.000018f,  0.019084f,  1.523383f,  0.074525f,
1737       -0.002076f,  -0.020734f, 4.881495f,   0.002799f,  0.000342f,  -0.019623f,
1738       1.786154f,   0.037462f,  -0.019037f,  0.052833f,  11.408153f, -0.044602f,
1739       0.026155f,   -0.518627f, -0.474499f,  -0.427430f, -0.442733f, -0.011116f,
1740       -22.379410f, -0.000549f, -0.001418f,  0.008090f,  -0.295090f, -0.230268f,
1741       -0.337278f,  -0.001127f, -0.644282f,  -0.598783f, -0.539417f, -0.003303f,
1742       9.189824f,   0.038066f,  -0.004097f,  -0.460045f, -0.308858f, -0.242691f,
1743       -0.230835f,  -0.273057f, 0.152226f,   0.179239f,  -0.146382f, -0.004655f,
1744       -0.242940f,  -0.718862f, -0.001685f,  -0.214736f, 3.263186f,  0.079463f,
1745       -0.003854f,  -0.187461f, -0.599144f,  -0.419808f, -0.000597f, -0.136980f,
1746       0.184813f,   -0.319525f, -0.007246f,  0.079709f,  -0.883229f, -0.343748f,
1747       -0.000077f,  -0.172214f, -0.548759f,  -0.194674f, -0.144786f, 0.043896f,
1748       -0.176364f,  -0.248394f, -0.090215f,  -0.294743f, -0.280980f, -0.181436f,
1749       -0.115681f,  -0.071915f, -13.035494f, -0.075623f, 0.017052f,  -0.171152f,
1750       5.910803f,   0.128344f,  0.010256f,   -1.073301f, 2.387826f,  0.166183f,
1751       -0.007193f,  -0.257836f,
1752     };
1753 
1754 static const float av1_partition_breakout_nn_bias_128_layer0[32] = {
1755   0.115591f,  -0.100178f, -0.165523f, -0.122997f, 11.045759f,  1.034761f,
1756   -0.323672f, -0.189087f, 2.850950f,  7.010029f,  -21.447067f, 1.877031f,
1757   0.437442f,  5.929414f,  -0.117274f, 4.462253f,  -0.135198f,  -0.145927f,
1758   8.727211f,  0.000000f,  -3.532987f, -0.405898f, 11.364439f,  -0.141728f,
1759   -5.994947f, -0.362574f, 1.857687f,  -0.100400f, -0.130312f,  0.006080f,
1760   0.429660f,  -8.439470f,
1761 };
1762 
1763 static const float av1_partition_breakout_nn_weights_128_layer1[32] = {
1764   -0.013738f, 0.022052f,  -0.074437f, -0.211377f, -0.080433f, 0.015543f,
1765   0.002091f,  0.014252f,  0.134834f,  0.190263f,  0.244175f,  -0.031747f,
1766   0.020068f,  -0.068326f, 0.185471f,  0.660268f,  -0.134898f, -0.010376f,
1767   -0.276023f, -0.282921f, -0.022769f, 0.007070f,  -0.186235f, 0.024407f,
1768   -0.024837f, 0.005764f,  0.016599f,  -0.040077f, 0.020990f,  0.095054f,
1769   -0.039662f, 0.131499f,
1770 };
1771 
1772 static const float av1_partition_breakout_nn_bias_128_layer1[1] = {
1773   0.86678213f,
1774 };
1775 
1776 static const NN_CONFIG av1_partition_breakout_nnconfig_128 = {
1777   FEATURE_SIZE,  // num_inputs
1778   1,             // num_outputs
1779   1,             // num_hidden_layers
1780   {
1781       32,  // num_hidden_nodes
1782   },
1783   {
1784       av1_partition_breakout_nn_weights_128_layer0,
1785       av1_partition_breakout_nn_weights_128_layer1,
1786   },
1787   {
1788       av1_partition_breakout_nn_bias_128_layer0,
1789       av1_partition_breakout_nn_bias_128_layer1,
1790   },
1791 };
1792 
1793 static const float
1794     av1_partition_breakout_nn_weights_64_layer0[FEATURE_SIZE * 16] = {
1795       0.872892f,  -0.235539f, -0.412159f, -0.142533f, -2.251479f, -0.057073f,
1796       -0.001373f, 0.112147f,  5.281734f,  0.060704f,  0.000838f,  -0.961554f,
1797       0.244995f,  0.154515f,  -0.292654f, -0.167177f, -3.759112f, -0.486347f,
1798       0.003208f,  -0.418226f, 2.618152f,  0.026832f,  0.003988f,  -0.404406f,
1799       -0.405434f, 0.102791f,  -0.033406f, -0.029820f, -4.492342f, -0.154291f,
1800       0.012947f,  -0.195075f, 0.009311f,  -0.411410f, -0.010986f, -0.554822f,
1801       0.160576f,  0.020796f,  -0.457230f, -0.191111f, -7.759542f, -0.065039f,
1802       -0.001322f, 0.055691f,  0.291924f,  -0.053076f, -0.148379f, -0.298383f,
1803       1.022023f,  -0.033668f, -0.000804f, -0.825778f, -3.902254f, -0.085812f,
1804       -0.052520f, -0.035012f, -0.465468f, -0.319231f, -0.497529f, -0.183068f,
1805       -2.407131f, -0.062304f, 0.000874f,  0.108786f,
1806     };
1807 
1808 static const float av1_partition_breakout_nn_bias_64_layer0[16] = {
1809   0.081425f,  -14.404084f, 11.511393f, -0.930053f, 1.841889f,  15.020920f,
1810   -1.872288f, 5.392535f,   -0.329335f, -0.005358f, 12.600776f, 0.000000f,
1811   -0.337413f, 4.492778f,   0.000000f,  17.043072f,
1812 };
1813 
1814 static const float av1_partition_breakout_nn_weights_64_layer1[16] = {
1815   -0.465338f, -0.103023f, -0.174808f, -0.005156f, -0.016366f, -0.172494f,
1816   0.014185f,  0.067030f,  -0.001939f, -0.175049f, 0.245992f,  -0.181660f,
1817   -0.038572f, 0.307899f,  -0.294283f, 0.118323f,
1818 };
1819 
1820 static const float av1_partition_breakout_nn_bias_64_layer1[1] = {
1821   -1.33438122f,
1822 };
1823 
1824 static const NN_CONFIG av1_partition_breakout_nnconfig_64 = {
1825   FEATURE_SIZE,  // num_inputs
1826   1,             // num_outputs
1827   1,             // num_hidden_layers
1828   {
1829       16,  // num_hidden_nodes
1830   },
1831   {
1832       av1_partition_breakout_nn_weights_64_layer0,
1833       av1_partition_breakout_nn_weights_64_layer1,
1834   },
1835   {
1836       av1_partition_breakout_nn_bias_64_layer0,
1837       av1_partition_breakout_nn_bias_64_layer1,
1838   },
1839 };
1840 
1841 static const float
1842     av1_partition_breakout_nn_weights_32_layer0[FEATURE_SIZE * 16] = {
1843       -4.825528f, -0.145737f, 0.001907f,  0.145415f,  -1.858153f, -0.080744f,
1844       0.000601f,  0.211991f,  0.384265f,  -0.043945f, -0.521332f, -0.170622f,
1845       -0.046866f, -0.600506f, -0.001216f, -0.332760f, -0.447677f, -0.605844f,
1846       -0.121008f, -0.119936f, -0.215739f, -0.269665f, -0.668587f, 0.071318f,
1847       -1.202551f, -0.729727f, -0.370084f, 0.088215f,  -1.926800f, -0.086519f,
1848       0.000359f,  0.215120f,  0.718749f,  0.022942f,  0.003840f,  -0.176518f,
1849       1.213451f,  0.080786f,  0.001557f,  -1.053430f, 0.202698f,  -0.583919f,
1850       -0.535512f, -0.239927f, -0.110151f, -0.128832f, -0.441087f, -0.145575f,
1851       -0.178518f, -0.585784f, 0.000029f,  -0.833014f, -0.331358f, -0.520297f,
1852       -0.088676f, -0.178487f, -1.430755f, 0.022981f,  -0.106931f, 0.015573f,
1853       -0.520814f, -0.045386f, -0.443123f, -0.484209f,
1854     };
1855 
1856 static const float av1_partition_breakout_nn_bias_32_layer0[16] = {
1857   11.747026f, -9.337718f, 0.341648f, -0.155847f, -0.104005f, 4.666283f,
1858   6.669584f,  16.625504f, 9.885626f, 15.439183f, -0.346080f, 0.000000f,
1859   -0.423808f, 0.000000f,  6.352258f, -0.155787f,
1860 };
1861 
1862 static const float av1_partition_breakout_nn_weights_32_layer1[16] = {
1863   0.168561f,  -0.122519f, 0.524667f,  0.032474f,  0.059097f,  0.011900f,
1864   0.166445f,  0.127256f,  -0.034838f, -0.212586f, -0.317973f, 0.348419f,
1865   -0.004171f, 0.157694f,  0.117845f,  0.272115f,
1866 };
1867 
1868 static const float av1_partition_breakout_nn_bias_32_layer1[1] = {
1869   0.09049262f,
1870 };
1871 
1872 static const NN_CONFIG av1_partition_breakout_nnconfig_32 = {
1873   FEATURE_SIZE,  // num_inputs
1874   1,             // num_outputs
1875   1,             // num_hidden_layers
1876   {
1877       16,  // num_hidden_nodes
1878   },
1879   {
1880       av1_partition_breakout_nn_weights_32_layer0,
1881       av1_partition_breakout_nn_weights_32_layer1,
1882   },
1883   {
1884       av1_partition_breakout_nn_bias_32_layer0,
1885       av1_partition_breakout_nn_bias_32_layer1,
1886   },
1887 };
1888 
1889 static const float
1890     av1_partition_breakout_nn_weights_16_layer0[FEATURE_SIZE * 16] = {
1891       0.209371f,  0.028758f,  0.005764f,  -0.384401f, -0.625777f, -0.005647f,
1892       -0.316867f, 0.042985f,  0.127344f,  0.025461f,  0.011465f,  -0.071043f,
1893       -0.295977f, -0.076093f, -0.209681f, -0.311653f, -0.147538f, 0.009910f,
1894       -0.130997f, -0.012326f, 0.024124f,  -0.323578f, -0.005790f, -0.085664f,
1895       -1.575066f, -0.119221f, 0.015018f,  0.187204f,  0.238117f,  0.084924f,
1896       -0.004444f, -1.271538f, -0.709860f, -0.006226f, -0.903111f, 0.090573f,
1897       -0.278642f, -0.011114f, 0.021162f,  0.081290f,  -0.467486f, -0.040771f,
1898       -0.224069f, -0.714390f, -0.281905f, -0.001336f, -0.761212f, -0.060385f,
1899       -0.814479f, -0.050450f, -0.003666f, 0.085668f,  -0.272589f, 0.057330f,
1900       -0.206540f, -0.303418f, 0.075335f,  -0.180468f, -0.064872f, -0.755948f,
1901       -0.509287f, -0.048877f, -0.001512f, 0.077086f,
1902     };
1903 
1904 static const float av1_partition_breakout_nn_bias_16_layer0[16] = {
1905   16.421495f, 4.012273f,  -1.828571f, 0.000000f,  -0.263564f, -0.201972f,
1906   6.564987f,  14.651000f, -3.227779f, 2.241833f,  -0.137116f, 0.762876f,
1907   5.625762f,  0.615822f,  0.040057f,  16.668884f,
1908 };
1909 
1910 static const float av1_partition_breakout_nn_weights_16_layer1[16] = {
1911   -0.096440f, 0.184316f,  -0.021148f, 0.424974f, 0.003743f,  0.006310f,
1912   0.046266f,  -0.219224f, -0.087004f, 0.024623f, -0.275798f, 0.120164f,
1913   0.269773f,  -0.021105f, -0.146698f, 0.188764f,
1914 };
1915 
1916 static const float av1_partition_breakout_nn_bias_16_layer1[1] = {
1917   1.60751927f,
1918 };
1919 
1920 static const NN_CONFIG av1_partition_breakout_nnconfig_16 = {
1921   FEATURE_SIZE,  // num_inputs
1922   1,             // num_outputs
1923   1,             // num_hidden_layers
1924   {
1925       16,  // num_hidden_nodes
1926   },
1927   {
1928       av1_partition_breakout_nn_weights_16_layer0,
1929       av1_partition_breakout_nn_weights_16_layer1,
1930   },
1931   {
1932       av1_partition_breakout_nn_bias_16_layer0,
1933       av1_partition_breakout_nn_bias_16_layer1,
1934   },
1935 };
1936 
1937 static const float
1938     av1_partition_breakout_nn_weights_8_layer0[FEATURE_SIZE * 16] = {
1939       -0.255885f, 0.109548f,  -0.111054f, -0.476119f, -1.083031f, -0.342003f,
1940       0.048241f,  -0.356013f, -0.085054f, 0.124908f,  0.000084f,  -0.149906f,
1941       -0.729829f, 0.133535f,  -0.002125f, 0.207516f,  -0.210163f, -0.567365f,
1942       -0.590103f, 0.045308f,  -0.539406f, 0.130550f,  -0.663879f, -0.170549f,
1943       0.017587f,  -0.054187f, 0.000550f,  0.038297f,  -0.112891f, -0.012751f,
1944       -0.048067f, 0.095564f,  0.079892f,  0.077285f,  -0.749708f, -0.286312f,
1945       -0.054334f, 0.132242f,  -0.004152f, -0.209758f, -0.073407f, 0.082306f,
1946       -0.001034f, -0.090990f, 0.122823f,  -0.109794f, -0.230066f, -0.391155f,
1947       -0.262245f, -0.004744f, -0.232246f, 0.099290f,  -0.637484f, 0.111937f,
1948       -0.548556f, -0.598344f, 0.123265f,  -0.281395f, -0.399711f, -0.525671f,
1949       -0.596269f, 0.098494f,  -0.005765f, 0.173652f,
1950     };
1951 
1952 static const float av1_partition_breakout_nn_bias_8_layer0[16] = {
1953   0.194141f, -0.111223f, 2.503733f, -7.155602f, -0.695068f, 0.114874f,
1954   2.056990f, 5.284306f,  0.639643f, -2.792049f, -2.232339f, -0.232209f,
1955   2.336705f, -0.278834f, 0.231905f, 7.954366f,
1956 };
1957 
1958 static const float av1_partition_breakout_nn_weights_8_layer1[16] = {
1959   -0.014439f, 0.010171f, 0.048116f,  -0.090659f, -0.081235f, -0.021840f,
1960   -0.017360f, 0.031063f, -0.031737f, -0.023439f, -0.037725f, 0.021954f,
1961   0.055858f,  0.230970f, -0.056466f, 0.119780f,
1962 };
1963 
1964 static const float av1_partition_breakout_nn_bias_8_layer1[1] = {
1965   1.27784479f,
1966 };
1967 
1968 static const NN_CONFIG av1_partition_breakout_nnconfig_8 = {
1969   FEATURE_SIZE,  // num_inputs
1970   1,             // num_outputs
1971   1,             // num_hidden_layers
1972   {
1973       16,  // num_hidden_nodes
1974   },
1975   {
1976       av1_partition_breakout_nn_weights_8_layer0,
1977       av1_partition_breakout_nn_weights_8_layer1,
1978   },
1979   {
1980       av1_partition_breakout_nn_bias_8_layer0,
1981       av1_partition_breakout_nn_bias_8_layer1,
1982   },
1983 };
1984 #undef FEATURE_SIZE
1985 
1986 #define FEATURE_SIZE 9  // Input layer size
1987 #define NUM_NODES 32    // Hidden layer size
1988 #define LABEL_SIZE 3    // Output layer size
1989 
1990 static const float av1_rect_partition_nn_weights_8_layer0[FEATURE_SIZE *
1991                                                           NUM_NODES] = {
1992   0.22151f,  0.99424f,  0.23415f,  -1.13841f, -0.11277f, 0.09530f,  0.14769f,
1993   -1.18895f, -0.96640f, -0.21421f, -0.13974f, 0.03236f,  0.15777f,  -0.03176f,
1994   0.02729f,  -0.37344f, -0.01727f, -0.05469f, 0.19402f,  -3.45508f, 0.90106f,
1995   -2.91557f, 0.19379f,  0.14356f,  -0.13291f, 0.05734f,  -0.03032f, -0.13060f,
1996   0.35744f,  1.31630f,  -1.54493f, -0.20749f, -0.24413f, -0.04524f, -0.12400f,
1997   1.08305f,  -0.21596f, 0.76244f,  1.10616f,  -1.71706f, 0.05768f,  0.10966f,
1998   0.00949f,  -0.12680f, 0.00699f,  -0.11522f, -0.38566f, 0.34283f,  -0.35266f,
1999   -0.40643f, -0.22462f, 0.32300f,  -0.39737f, -0.20587f, -0.16096f, 1.07543f,
2000   0.30314f,  -1.35659f, -0.38212f, 0.45857f,  0.76615f,  0.16819f,  -1.24459f,
2001   0.39677f,  0.87436f,  -2.33757f, 1.27471f,  0.27488f,  0.01019f,  -0.01221f,
2002   -0.07461f, -0.14577f, -0.01231f, -0.64426f, -1.02733f, -1.96242f, 0.95143f,
2003   -0.06777f, -1.13868f, 0.01354f,  -0.75590f, -0.78222f, -0.07453f, 0.61788f,
2004   0.56899f,  1.17144f,  0.70899f,  0.48568f,  0.11266f,  0.81579f,  -0.03929f,
2005   0.01088f,  0.33599f,  -0.22401f, -0.49654f, -0.02598f, 0.04509f,  -0.08217f,
2006   -0.30687f, 0.19851f,  -2.96860f, -2.30698f, 0.01848f,  0.11801f,  0.06614f,
2007   0.01673f,  -0.11002f, -0.08168f, 0.09204f,  -0.06379f, 0.27972f,  -0.31716f,
2008   -0.00566f, -0.13651f, -0.37276f, 0.01511f,  -0.23697f, 0.21696f,  -0.19480f,
2009   0.60758f,  -0.43506f, -0.02247f, -1.45073f, 0.84442f,  -0.94018f, 0.32550f,
2010   0.03985f,  -0.06581f, 0.21665f,  0.79472f,  -2.41080f, 0.04788f,  -0.09492f,
2011   -0.10677f, 0.07250f,  0.14329f,  -0.37319f, 0.53043f,  -0.49108f, 0.25792f,
2012   -0.36569f, -0.28669f, -0.18416f, -0.52385f, -1.17081f, -1.32153f, -1.13403f,
2013   -0.26196f, 0.93379f,  0.72115f,  0.54464f,  0.27642f,  0.04757f,  2.01629f,
2014   1.55787f,  -0.11665f, 1.00722f,  -0.24352f, 0.53308f,  0.57719f,  0.39344f,
2015   0.19174f,  0.06339f,  -0.02530f, 0.07724f,  -0.32416f, -0.26992f, -0.35887f,
2016   -0.35285f, -0.33379f, -0.37475f, -0.77335f, 1.70027f,  -1.52153f, -0.26503f,
2017   0.97552f,  -2.96705f, -0.91220f, -0.11827f, 0.00406f,  -0.14514f, 0.18417f,
2018   -0.20874f, 0.27293f,  -0.34072f, -0.34838f, -0.19054f, -0.29806f, -0.27960f,
2019   -0.19293f, -0.18275f, -0.05902f, 0.58625f,  -0.05470f, -0.48814f, -0.45382f,
2020   -0.05959f, 2.01250f,  -0.30014f, 0.69546f,  -1.24180f, 1.34923f,  0.20337f,
2021   0.16850f,  0.07187f,  0.72630f,  -0.15380f, -2.40973f, -2.73561f, -1.71375f,
2022   -1.61695f, 0.50052f,  0.09730f,  0.00579f,  0.06133f,  -0.06512f, -0.61439f,
2023   -1.16173f, -0.58716f, 1.60438f,  0.23242f,  0.91847f,  0.49041f,  -0.16277f,
2024   -0.02574f, -0.64593f, 1.17028f,  0.46852f,  0.14926f,  0.73853f,  -0.78521f,
2025   0.05959f,  -0.35590f, 0.02039f,  0.10812f,  -0.28650f, 1.34038f,  -0.72188f,
2026   0.62385f,  -0.35271f, -0.39599f, 0.41543f,  0.53124f,  -0.23510f, -0.15480f,
2027   -0.05066f, -0.33529f, 0.05238f,  -0.35311f, -0.26983f, -0.39764f, 0.01085f,
2028   0.26593f,  -0.18411f, -0.29945f, 0.50090f,  -0.03397f, 0.78562f,  -0.33068f,
2029   1.21308f,  -2.23273f, -0.33366f, -0.15164f, -1.13270f, 0.17394f,  0.65567f,
2030   0.76496f,  0.44325f,  0.01368f,  -0.33619f, -0.64256f, 0.64478f,  0.84553f,
2031   1.74183f,  0.22563f,  -0.14550f, -0.16258f, 0.03010f,  0.49922f,  0.64575f,
2032   -0.29187f, -0.10348f, -1.43619f, -0.56540f, -0.14779f, 0.04616f,  0.87411f,
2033   -1.08228f,
2034 };
2035 
2036 static const float av1_rect_partition_nn_bias_8_layer0[NUM_NODES] = {
2037   0.33919f,  -0.03003f, 0.79073f,  -0.18508f, 0.00668f,  -0.12017f, 0.35362f,
2038   -0.51642f, 0.06536f,  0.41668f,  -0.06509f, 0.94606f,  -0.15385f, 0.14936f,
2039   1.46274f,  -0.06961f, 2.82537f,  -1.95576f, -0.09457f, 0.02042f,  -0.07480f,
2040   -0.55083f, 0.26170f,  4.39883f,  0.33999f,  -0.10502f, 0.70884f,  -0.06992f,
2041   -0.22638f, 1.40940f,  -0.09309f, 0.05828f,
2042 };
2043 
2044 static const float av1_rect_partition_nn_weights_8_layer1[NUM_NODES *
2045                                                           LABEL_SIZE] = {
2046   0.09209f,  0.26236f,  0.62136f,  0.76324f,  -1.14678f, 0.42289f,  -0.08895f,
2047   -0.97267f, 2.05958f,  0.00843f,  0.35335f,  1.12096f,  -0.11679f, 0.07350f,
2048   -1.23231f, -0.61990f, 1.51379f,  -1.99450f, 0.22441f,  2.41974f,  -0.30488f,
2049   -0.37869f, 0.47168f,  -3.70132f, 0.00061f,  0.19432f,  0.11512f,  0.26200f,
2050   -0.35285f, 0.37985f,  0.90571f,  0.27344f,  0.74840f,  -0.17965f, -2.51433f,
2051   0.59235f,  1.16670f,  -0.53446f, 0.67897f,  0.04505f,  -0.86874f, 0.45361f,
2052   -0.35033f, 1.21283f,  0.31426f,  -0.20841f, 0.56757f,  0.45909f,  -1.23683f,
2053   0.09835f,  -0.17214f, -0.96323f, 0.01138f,  -0.50233f, 0.30104f,  2.01814f,
2054   1.15821f,  -0.11947f, 0.74574f,  -0.30714f, -0.39646f, -1.30086f, -0.88541f,
2055   -0.12259f, -0.54977f, 0.30069f,  1.84299f,  -0.95141f, -0.65887f, -0.25888f,
2056   -0.63265f, 1.29531f,  -0.56672f, 0.10837f,  -0.21297f, -2.19131f, 0.01156f,
2057   0.51912f,  0.46704f,  0.42810f,  -0.59271f, 0.98469f,  -0.17914f, -1.91163f,
2058   -0.32807f, 0.48199f,  -0.99525f, 1.67108f,  -0.87631f, -0.60258f, -0.78731f,
2059   -0.32877f, 0.44237f,  0.01087f,  0.07489f,  -0.28224f,
2060 };
2061 
2062 static const float av1_rect_partition_nn_bias_8_layer1[LABEL_SIZE] = {
2063   1.70665f,
2064   -0.77954f,
2065   -0.92709f,
2066 };
2067 
2068 static const NN_CONFIG av1_rect_partition_nnconfig_8 = {
2069   FEATURE_SIZE,  // num_inputs
2070   LABEL_SIZE,    // num_outputs
2071   1,             // num_hidden_layers
2072   {
2073       NUM_NODES,
2074   },  // num_hidden_nodes
2075   { av1_rect_partition_nn_weights_8_layer0,
2076     av1_rect_partition_nn_weights_8_layer1 },
2077   { av1_rect_partition_nn_bias_8_layer0, av1_rect_partition_nn_bias_8_layer1 }
2078 };
2079 
2080 static const float av1_rect_partition_nn_weights_16_layer0[FEATURE_SIZE *
2081                                                            NUM_NODES] = {
2082   -0.18480f, -0.05410f, -0.18957f, 0.15451f,  -0.38649f, -0.26162f, -0.22727f,
2083   -0.38555f, -0.36738f, 0.74384f,  -1.85999f, 0.98491f,  -0.72119f, 1.77321f,
2084   0.39983f,  0.96314f,  0.23695f,  0.30200f,  0.30629f,  -0.47617f, -1.43320f,
2085   -1.81730f, 0.36554f,  -0.07142f, -1.27242f, -1.27697f, 0.00110f,  -0.32179f,
2086   0.27460f,  0.45428f,  0.15308f,  -0.73906f, -0.28577f, -0.01238f, -0.16958f,
2087   -0.85390f, 1.05484f,  -1.62812f, 0.77632f,  -0.27327f, -0.32527f, 0.32726f,
2088   1.73255f,  0.53763f,  0.59121f,  -0.39068f, -0.32451f, -0.31869f, 0.17777f,
2089   0.07519f,  -0.18066f, -0.11250f, -0.14616f, -0.16882f, -0.04099f, -0.67959f,
2090   0.39674f,  -0.08596f, 0.18587f,  -2.04097f, -1.73993f, 1.57212f,  1.42410f,
2091   -1.36762f, -0.41485f, -1.12103f, 0.56959f,  0.11500f,  0.48945f,  -0.13585f,
2092   1.22125f,  0.67071f,  -1.11812f, -0.20660f, -0.52856f, 0.70663f,  0.74382f,
2093   0.61114f,  -0.11454f, 1.14687f,  0.80322f,  -0.45965f, -0.44466f, -0.05830f,
2094   0.13206f,  -0.53750f, -0.11324f, -0.37971f, -0.13491f, -0.21268f, 1.93407f,
2095   1.34433f,  2.49427f,  2.91955f,  1.71730f,  0.03295f,  0.03587f,  -0.14550f,
2096   0.08189f,  -0.38655f, -0.35432f, -0.62706f, -0.01849f, -0.57882f, -0.60438f,
2097   -1.01334f, -0.57302f, 0.22592f,  0.05916f,  -0.05305f, -0.89824f, -0.52969f,
2098   -0.24542f, 0.27029f,  -0.40924f, -0.82452f, -0.60665f, -5.03025f, 0.83302f,
2099   1.83695f,  2.19716f,  2.31001f,  0.03657f,  0.00063f,  -0.04379f, 0.05835f,
2100   -0.08623f, 0.20557f,  -0.17791f, 0.07874f,  -0.25456f, -0.19513f, -0.27753f,
2101   -0.31982f, 0.00245f,  -0.33183f, 0.26059f,  -0.22165f, 0.37582f,  -0.30411f,
2102   -0.22639f, -0.14739f, -0.20201f, -0.37507f, -1.30653f, 0.49570f,  1.03673f,
2103   0.66139f,  0.44941f,  -0.44461f, -0.50376f, -0.49664f, 0.18608f,  -0.26175f,
2104   0.14844f,  0.78715f,  -0.70344f, -0.87624f, -0.98535f, -0.35346f, 0.37094f,
2105   -0.43135f, -0.22571f, 3.46263f,  3.13580f,  -1.33203f, -0.15247f, -0.15866f,
2106   -0.11214f, 0.12211f,  0.03964f,  -1.87597f, -4.81597f, -4.80195f, -4.98096f,
2107   -5.62336f, -0.05337f, -0.00943f, 0.00792f,  0.02742f,  1.05679f,  2.41455f,
2108   0.85382f,  1.42504f,  0.58096f,  0.21443f,  1.02694f,  1.06746f,  1.20242f,
2109   0.60767f,  1.98667f,  -0.80879f, -0.63495f, 1.95508f,  0.23952f,  -0.15019f,
2110   -0.16097f, 0.30155f,  -3.42407f, -1.34998f, 9.07689f,  -2.22559f, 2.22562f,
2111   -0.03348f, -0.05229f, 0.05931f,  0.03042f,  -0.18068f, -0.05732f, -0.33010f,
2112   -0.32279f, -0.26607f, -0.02723f, -0.04067f, 0.08700f,  -0.16366f, -0.24935f,
2113   -0.69124f, 0.58508f,  0.50654f,  0.04492f,  1.38340f,  -1.51487f, 1.72889f,
2114   -1.95618f, -3.65013f, -1.38525f, -3.05516f, -2.40448f, 2.47467f,  0.03784f,
2115   0.08052f,  -0.01971f, -0.08918f, -0.84997f, -0.55302f, -1.07861f, -0.62626f,
2116   0.61751f,  -0.11012f, -0.24185f, -0.39201f, -1.85390f, -0.31261f, -0.11927f,
2117   0.15671f,  -0.23450f, -0.14916f, -0.31715f, -0.19350f, 0.01795f,  -0.11533f,
2118   -0.05799f, -0.03142f, 0.20218f,  -0.39499f, -0.33859f, -0.13201f, -0.19527f,
2119   -0.28459f, -0.20346f, 0.89457f,  -2.22103f, -2.37455f, -2.00221f, 2.44553f,
2120   0.33915f,  0.50047f,  -0.34625f, -0.19667f, -0.56333f, -0.84328f, 1.25767f,
2121   -1.70297f, 1.00482f,  -0.00103f, -1.40813f, 0.21311f,  0.39230f,  -0.07302f,
2122   -3.49100f, 1.60675f,  -2.90692f, 0.11022f,  0.13507f,  -0.13308f, 0.15201f,
2123   -0.05573f,
2124 };
2125 
2126 static const float av1_rect_partition_nn_bias_16_layer0[NUM_NODES] = {
2127   -0.16783f, -0.16023f, 0.52215f,  -0.04109f, 2.00122f,  -0.11633f, 0.25535f,
2128   1.80638f,  1.69273f,  -0.25998f, -6.83550f, -0.79682f, -1.03466f, 1.42721f,
2129   0.00000f,  -0.00000f, -0.11665f, -0.12047f, -1.01497f, 7.27181f,  -0.78548f,
2130   -1.39335f, -5.42248f, -0.10388f, 0.07634f,  2.81012f,  -0.57429f, -0.15629f,
2131   -0.12044f, 1.65478f,  -0.75153f, 1.18441f,
2132 };
2133 
2134 static const float av1_rect_partition_nn_weights_16_layer1[NUM_NODES *
2135                                                            LABEL_SIZE] = {
2136   -0.26407f, 0.06322f,  0.87932f,  0.17772f,  0.71686f,  -0.12283f, 0.08454f,
2137   0.20098f,  -0.31763f, -0.33178f, -4.59535f, -0.04367f, 0.17099f,  3.80486f,
2138   0.16750f,  0.29218f,  0.57234f,  -0.96550f, -0.10599f, -4.91130f, -0.14658f,
2139   0.95803f,  -4.13925f, 0.24567f,  0.25708f,  1.60547f,  -1.03251f, -0.31053f,
2140   -0.05659f, -0.94121f, -0.68926f, -0.24738f, -0.38019f, 0.98950f,  0.13689f,
2141   0.24504f,  0.49623f,  0.19980f,  0.38349f,  0.37481f,  0.54540f,  -0.02198f,
2142   3.43385f,  1.02543f,  -0.40921f, -3.07235f, 0.02996f,  0.00323f,  -0.35414f,
2143   0.71099f,  1.39334f,  2.43741f,  -1.11007f, -0.22739f, -4.21757f, 0.11905f,
2144   0.00353f,  -1.69637f, 0.45944f,  -0.19884f, 0.03624f,  0.25729f,  0.23659f,
2145   -2.08405f, 0.08573f,  -0.53393f, -1.28103f, -0.53970f, -0.65465f, 0.31821f,
2146   -0.09884f, -0.69026f, -0.37284f, 0.04622f,  1.32973f,  -0.15414f, 0.19138f,
2147   -0.67927f, -0.17658f, 0.36008f,  -0.51832f, 0.09887f,  -1.94414f, 2.95227f,
2148   1.76937f,  -0.26687f, 8.50976f,  0.26247f,  0.60262f,  -0.27910f, 0.30061f,
2149   -0.05117f, 0.16018f,  0.71195f,  0.57871f,  1.57794f,
2150 };
2151 
2152 static const float av1_rect_partition_nn_bias_16_layer1[3] = {
2153   2.68750f,
2154   -1.31894f,
2155   -1.36768f,
2156 };
2157 
2158 static const NN_CONFIG av1_rect_partition_nnconfig_16 = {
2159   FEATURE_SIZE,  // num_inputs
2160   LABEL_SIZE,    // num_outputs
2161   1,             // num_hidden_layers
2162   {
2163       NUM_NODES,
2164   },  // num_hidden_nodes
2165   { av1_rect_partition_nn_weights_16_layer0,
2166     av1_rect_partition_nn_weights_16_layer1 },
2167   { av1_rect_partition_nn_bias_16_layer0, av1_rect_partition_nn_bias_16_layer1 }
2168 };
2169 
2170 static const float av1_rect_partition_nn_weights_32_layer0[FEATURE_SIZE *
2171                                                            NUM_NODES] = {
2172   -0.54654f, -0.43537f, -0.10620f, -0.48051f, -0.43543f, -0.22737f, -0.15429f,
2173   -0.09858f, -0.09438f, 0.37306f,  0.23934f,  -1.86375f, -1.18307f, -0.32995f,
2174   -0.09745f, 0.05431f,  -0.13799f, 0.14734f,  -0.33219f, 0.18057f,  -0.23792f,
2175   -0.28126f, 0.02977f,  -0.07431f, 0.07860f,  0.00067f,  -0.01927f, 1.01841f,
2176   -0.57739f, 0.08412f,  -1.33843f, -1.05563f, -0.28693f, -0.39425f, -0.69572f,
2177   -0.16703f, 0.02808f,  0.11994f,  -0.26267f, 0.19706f,  -0.29707f, -0.25305f,
2178   -0.07050f, -0.02704f, -0.31528f, -0.42301f, 0.22496f,  -0.37001f, -0.23319f,
2179   -0.11139f, -0.30513f, 0.04213f,  -0.12550f, 0.02504f,  0.33245f,  0.01102f,
2180   -0.35950f, -0.05949f, -0.19590f, -0.27457f, -0.28339f, -0.15676f, -0.21538f,
2181   0.65066f,  0.28443f,  -1.24943f, -3.00246f, -1.01897f, 0.09304f,  0.70052f,
2182   -0.12877f, 0.21120f,  -0.37476f, 0.23261f,  -0.28401f, 0.09837f,  0.00020f,
2183   -0.12106f, -0.32354f, -0.02472f, -0.19772f, 1.01886f,  0.16596f,  -0.06532f,
2184   1.72938f,  1.57754f,  0.55963f,  0.33246f,  -0.20023f, 0.30715f,  0.08629f,
2185   0.18945f,  -0.45988f, -1.22610f, -0.05152f, -0.48859f, -1.02104f, -0.27315f,
2186   -0.57698f, 0.04157f,  -0.92428f, -1.31268f, 1.78210f,  0.10291f,  1.55042f,
2187   -1.26793f, 1.39042f,  -1.43729f, 0.25600f,  5.21263f,  5.31955f,  5.19316f,
2188   5.43430f,  0.00294f,  -0.00970f, -0.02333f, 0.00250f,  1.17672f,  6.27544f,
2189   4.95973f,  3.54009f,  4.51269f,  0.30750f,  0.78780f,  -0.44741f, -0.76442f,
2190   0.75050f,  0.58799f,  0.03400f,  -2.09859f, 1.67313f,  0.12503f,  0.28609f,
2191   1.15809f,  2.46530f,  -0.04898f, 0.23072f,  -0.12635f, -0.82097f, -0.63827f,
2192   2.16779f,  1.77132f,  0.15434f,  -1.06427f, 0.06206f,  -0.87732f, -0.61897f,
2193   -0.44593f, -0.77131f, -0.15979f, -0.02282f, -0.74381f, 0.66052f,  -0.22992f,
2194   1.74638f,  1.29199f,  -0.55464f, 0.98316f,  0.06665f,  0.50254f,  -0.66292f,
2195   0.17113f,  -0.32633f, -1.85803f, -0.92759f, 4.44965f,  1.33057f,  0.02135f,
2196   -0.27446f, -0.26018f, -0.12613f, -0.14470f, -0.23355f, -0.09717f, -0.24123f,
2197   -0.05535f, -0.19146f, -0.36222f, -0.30458f, -0.40323f, 0.21779f,  0.14248f,
2198   -0.48630f, 0.18840f,  0.11040f,  0.17287f,  -0.51880f, 1.12466f,  -0.38888f,
2199   -0.16421f, -0.31784f, -0.36112f, -0.25386f, -0.01636f, 0.10029f,  -0.26881f,
2200   -0.17051f, -0.30903f, -0.08573f, -0.28774f, -0.01173f, -0.09706f, -0.23089f,
2201   -0.12922f, -0.17463f, -0.12433f, -0.23074f, 0.15220f,  1.29826f,  0.23788f,
2202   0.04189f,  2.66416f,  0.48815f,  -0.06803f, 0.96742f,  1.27165f,  -0.70348f,
2203   -0.09941f, -0.42948f, -0.20243f, -0.02364f, -0.26689f, -0.40629f, -0.68217f,
2204   -0.48073f, 2.43657f,  -2.60191f, -1.82837f, 0.50440f,  0.71829f,  0.76491f,
2205   0.28293f,  0.20568f,  0.92642f,  -0.02496f, 1.43637f,  -0.24474f, -1.21030f,
2206   0.54084f,  1.05130f,  1.29572f,  0.03750f,  -0.36894f, 0.74548f,  -1.33857f,
2207   -0.84858f, 1.35230f,  0.80175f,  0.66136f,  1.06473f,  0.18701f,  1.42413f,
2208   0.04661f,  -0.07820f, 0.64990f,  -0.43595f, 1.18304f,  -0.11437f, -0.06365f,
2209   0.03558f,  0.78260f,  -1.74890f, 1.56217f,  -1.23424f, 4.59193f,  -3.35072f,
2210   0.01180f,  -0.18296f, -0.20870f, 0.04510f,  1.52595f,  -1.37402f, -0.33123f,
2211   -0.85957f, 0.80598f,  0.03743f,  0.02354f,  0.37707f,  1.62095f,  -0.29627f,
2212   -0.31778f, -0.45789f, -0.14906f, 0.25315f,  -0.10817f, -0.32610f, -0.40890f,
2213   0.33984f,
2214 };
2215 
2216 static const float av1_rect_partition_nn_bias_32_layer0[NUM_NODES] = {
2217   -0.17482f, 0.39042f,  0.00000f,  1.69677f,  0.08792f,  -0.09301f, 0.13809f,
2218   4.84061f,  0.00000f,  0.40515f,  0.46246f,  0.20644f,  -5.77478f, -1.54510f,
2219   0.05660f,  -0.32013f, 0.23649f,  0.03778f,  -2.53710f, -0.27869f, 0.45623f,
2220   -0.04155f, -0.18445f, -0.73405f, -0.50243f, 2.23191f,  1.93272f,  -1.07032f,
2221   -0.27602f, -1.98063f, 0.20816f,  -0.01315f,
2222 };
2223 
2224 static const float av1_rect_partition_nn_weights_32_layer1[NUM_NODES *
2225                                                            LABEL_SIZE] = {
2226   0.02827f,  1.02560f,  -0.07137f, -0.31911f, 0.11365f,  0.13684f,  -0.07816f,
2227   -5.23036f, -0.34340f, 0.84526f,  -1.51845f, 0.07017f,  -8.12570f, 6.24061f,
2228   0.35739f,  -0.09937f, -0.30978f, 0.22032f,  0.74968f,  -0.34557f, 0.45547f,
2229   -0.16512f, 0.07118f,  1.66415f,  0.41320f,  -1.81533f, -1.96004f, 1.04666f,
2230   0.84049f,  4.31009f,  0.68850f,  0.26322f,  -0.24634f, -1.25889f, 0.31952f,
2231   0.63632f,  0.05801f,  -0.10664f, -0.21992f, 2.44386f,  0.19526f,  -0.09838f,
2232   1.53049f,  -0.26630f, 3.54126f,  -3.40574f, 0.72730f,  0.04557f,  0.92652f,
2233   0.15522f,  2.35895f,  -0.13347f, 0.56907f,  0.15352f,  0.01823f,  -0.73939f,
2234   0.43104f,  1.90321f,  0.31267f,  -0.51972f, 0.50094f,  -3.98372f, -3.41518f,
2235   -0.48183f, 0.26661f,  0.64146f,  0.14500f,  -0.01695f, 0.16653f,  -0.37846f,
2236   0.08412f,  2.69714f,  -0.20258f, -0.75786f, 0.11201f,  0.61878f,  4.22231f,
2237   -3.55330f, -1.14137f, -0.37722f, -0.28000f, -0.72581f, -2.62827f, -0.19448f,
2238   -0.59398f, -0.30136f, -0.17725f, -0.69630f, -0.41132f, 0.12208f,  2.11441f,
2239   -1.08794f, -1.41694f, 0.02620f,  2.18792f,  0.04271f,
2240 };
2241 
2242 static const float av1_rect_partition_nn_bias_32_layer1[3] = {
2243   2.47332f,
2244   -1.65756f,
2245   -0.81573f,
2246 };
2247 
2248 static const NN_CONFIG av1_rect_partition_nnconfig_32 = {
2249   FEATURE_SIZE,  // num_inputs
2250   LABEL_SIZE,    // num_outputs
2251   1,             // num_hidden_layers
2252   {
2253       NUM_NODES,
2254   },  // num_hidden_nodes
2255   { av1_rect_partition_nn_weights_32_layer0,
2256     av1_rect_partition_nn_weights_32_layer1 },
2257   { av1_rect_partition_nn_bias_32_layer0, av1_rect_partition_nn_bias_32_layer1 }
2258 };
2259 
2260 static const float av1_rect_partition_nn_weights_64_layer0[FEATURE_SIZE *
2261                                                            NUM_NODES] = {
2262   0.08972f,  4.09095f,  -0.31398f, -2.43631f, -0.74767f, 1.42471f,  1.60926f,
2263   1.44721f,  1.88259f,  2.35375f,  1.88299f,  2.01109f,  0.98679f,  2.24131f,
2264   0.06279f,  -0.08315f, 0.32107f,  0.91334f,  -0.36569f, 5.55049f,  5.44943f,
2265   5.20471f,  5.39099f,  -0.01943f, -0.00284f, 0.02203f,  -0.01309f, 1.41917f,
2266   6.68460f,  -6.15986f, 6.41341f,  -3.20630f, -0.00567f, -0.00038f, 0.05960f,
2267   0.04308f,  0.95366f,  3.48535f,  2.98266f,  4.11784f,  3.44255f,  0.61630f,
2268   0.71405f,  0.63945f,  -0.00713f, 0.39193f,  1.91621f,  3.32755f,  0.71674f,
2269   -0.11647f, 2.07090f,  2.64191f,  0.07949f,  -0.05023f, 0.99935f,  0.83145f,
2270   0.75898f,  -0.98764f, -0.58731f, 1.21734f,  -0.08076f, -3.26780f, 1.66278f,
2271   0.04189f,  -0.33177f, -1.58648f, 1.00883f,  -0.56132f, -2.34877f, 0.67056f,
2272   -2.32297f, -0.91641f, -1.02909f, 4.19781f,  3.87484f,  4.32778f,  -1.97171f,
2273   -0.24734f, 0.00822f,  0.05892f,  0.12697f,  -3.62915f, -2.93127f, 7.94856f,
2274   -3.29311f, 3.26001f,  -0.02231f, 0.02741f,  0.05919f,  0.08190f,  -1.49344f,
2275   -0.64475f, -0.24627f, 4.03324f,  -1.14799f, -0.18465f, -0.17829f, 0.10394f,
2276   0.08580f,  -5.74721f, 4.42467f,  3.63964f,  3.00258f,  -1.22744f, -0.29408f,
2277   0.00767f,  0.12305f,  0.05249f,  -0.17166f, -0.20120f, -0.32941f, -0.31901f,
2278   0.04628f,  -0.35249f, -0.18272f, 0.03956f,  -0.19329f, -0.33564f, 0.09856f,
2279   -0.00173f, -0.31751f, -0.05702f, -0.20558f, -0.31464f, -0.02488f, -0.00729f,
2280   -0.35854f, -0.14762f, -0.34897f, -0.12746f, 0.04011f,  -0.24918f, -0.53516f,
2281   -0.28440f, -0.36789f, -1.34889f, -9.10044f, -9.19238f, 4.48042f,  6.54429f,
2282   -0.00226f, 0.00430f,  0.00321f,  0.00442f,  0.87551f,  -0.16224f, -0.22832f,
2283   -0.60640f, -0.28738f, 0.18062f,  0.22008f,  -0.47406f, 0.80302f,  0.12149f,
2284   1.49530f,  1.05069f,  -2.02985f, -0.92833f, 0.25616f,  0.12852f,  3.51840f,
2285   0.25226f,  -2.63283f, -4.04386f, 8.46300f,  -2.93408f, 0.44069f,  0.08276f,
2286   0.34482f,  -0.22615f, 0.28666f,  3.02962f,  -1.20055f, -1.04832f, -0.97632f,
2287   -0.99530f, 1.44196f,  1.68550f,  0.49360f,  1.08155f,  -0.26059f, -0.02876f,
2288   -0.27492f, -0.06205f, -0.09496f, -0.12314f, -0.30228f, -0.07453f, -0.38857f,
2289   1.17443f,  2.41497f,  1.90537f,  2.37716f,  2.91495f,  -0.44455f, -0.51176f,
2290   0.48195f,  0.53032f,  0.23696f,  -1.06211f, 1.47459f,  -0.89029f, 0.29521f,
2291   0.66291f,  -0.42653f, 1.82308f,  -1.30372f, -0.36192f, -3.40388f, -1.61476f,
2292   -2.29745f, -0.66886f, -2.08252f, -0.54552f, -4.06849f, 0.02948f,  0.27297f,
2293   -4.81472f, 4.60404f,  -0.11053f, 0.14765f,  0.02826f,  -0.14688f, -0.07066f,
2294   -0.01224f, 1.20377f,  7.02725f,  -6.02627f, 6.87255f,  -3.14257f, 0.01074f,
2295   0.02397f,  -0.02359f, 0.01901f,  0.14956f,  -1.67671f, 2.26714f,  2.57043f,
2296   -0.45888f, -1.60265f, -2.11475f, -2.74029f, -2.74658f, -0.35630f, -2.63013f,
2297   -2.14814f, -0.67266f, -1.56850f, 0.57137f,  -1.14428f, -0.34265f, -0.12521f,
2298   0.01220f,  -0.74906f, -0.19270f, 0.68110f,  -0.24737f, -0.70568f, -1.64826f,
2299   -0.35847f, -0.15984f, -1.17932f, -8.72306f, -8.72834f, 3.93701f,  6.17812f,
2300   -0.03191f, -0.00104f, 0.01402f,  -0.00046f, -0.94517f, 1.51266f,  -0.56318f,
2301   0.72260f,  -0.09253f, -0.09069f, -2.16695f, -0.23653f, 0.24418f,  2.21148f,
2302   -1.47954f, -1.01439f, 0.31536f,  0.77238f,  -0.85083f, -0.15758f, -0.50886f,
2303   0.09101f,
2304 };
2305 
2306 static const float av1_rect_partition_nn_bias_64_layer0[NUM_NODES] = {
2307   0.91706f,  -1.31328f, -5.16196f, 1.13191f,  -0.98044f, -1.61122f, 1.03039f,
2308   -0.98537f, -4.45568f, -4.34802f, -0.92116f, 0.66836f,  -0.10752f, -0.13065f,
2309   -0.35567f, -0.35693f, 1.74941f,  1.17379f,  -3.45555f, 5.66321f,  -0.24917f,
2310   -1.11940f, -0.73656f, -0.19299f, -0.04181f, 1.11010f,  -2.97859f, -0.16774f,
2311   0.59835f,  -0.31269f, -0.30585f, -1.66212f,
2312 };
2313 
2314 static const float av1_rect_partition_nn_weights_64_layer1[NUM_NODES *
2315                                                            LABEL_SIZE] = {
2316   0.58963f,  4.20320f,  -8.62465f, -6.54014f, 5.41108f,  2.33581f,   -0.10354f,
2317   -1.17753f, -3.45909f, -2.24722f, 2.20881f,  3.21971f,  -0.09087f,  -0.21624f,
2318   0.16529f,  -8.40985f, -1.60205f, -1.41538f, 4.41826f,  -4.63069f,  -0.27742f,
2319   4.08710f,  0.26439f,  -1.46028f, 0.51234f,  6.25212f,  -3.35650f,  -1.21348f,
2320   1.37201f,  8.89151f,  0.28859f,  -0.97328f, -0.36196f, -2.71701f,  4.54196f,
2321   -0.62476f, -2.43814f, -1.34209f, 0.12850f,  1.73859f,  3.09809f,   -4.42434f,
2322   -1.82552f, -3.66420f, -0.31535f, 0.00968f,  -0.02019f, 9.66824f,   0.58835f,
2323   1.50425f,  2.84487f,  2.55522f,  0.01409f,  -2.27594f, -0.31800f,  0.91076f,
2324   -0.66808f, 0.33120f,  -0.12460f, 0.64457f,  -0.36416f, -10.30843f, 1.51013f,
2325   2.06861f,  -0.20989f, -0.87119f, 3.68642f,  7.33662f,  -2.88037f,  -0.52414f,
2326   -0.35036f, -0.45947f, -0.07406f, 6.46346f,  -0.16031f, 0.27071f,   0.38845f,
2327   -0.21940f, 0.08583f,  -1.39526f, 0.50554f,  0.45279f,  -6.61856f,  1.84069f,
2328   -0.19149f, -1.77235f, 0.75136f,  1.11797f,  0.32677f,  -7.10427f,  3.82908f,
2329   1.04238f,  -0.91435f, 1.93317f,  -1.84946f, -0.48909f,
2330 };
2331 
2332 static const float av1_rect_partition_nn_bias_64_layer1[3] = {
2333   0.32215f,
2334   -0.57522f,
2335   0.25314f,
2336 };
2337 
2338 static const NN_CONFIG av1_rect_partition_nnconfig_64 = {
2339   FEATURE_SIZE,  // num_inputs
2340   LABEL_SIZE,    // num_outputs
2341   1,             // num_hidden_layers
2342   {
2343       NUM_NODES,
2344   },  // num_hidden_nodes
2345   { av1_rect_partition_nn_weights_64_layer0,
2346     av1_rect_partition_nn_weights_64_layer1 },
2347   { av1_rect_partition_nn_bias_64_layer0, av1_rect_partition_nn_bias_64_layer1 }
2348 };
2349 
2350 static const float av1_rect_partition_nn_weights_128_layer0[FEATURE_SIZE *
2351                                                             NUM_NODES] = {
2352   -0.70901f, -3.03481f, 3.30604f,  -1.28803f,  -0.08610f, -0.33320f, -0.30716f,
2353   0.25100f,  0.14323f,  -0.98422f, -0.89084f,  -0.24508f, -1.10785f, -0.82524f,
2354   0.11766f,  -0.42777f, 1.08965f,  4.35125f,   -1.19388f, 4.22042f,  4.96306f,
2355   6.32406f,  3.29899f,  -0.90768f, 0.05203f,   0.38467f,  1.74257f,  -0.19918f,
2356   -0.11335f, 0.00140f,  -0.42303f, -0.04419f,  0.03583f,  -0.05441f, -0.19586f,
2357   0.01484f,  -1.19964f, 0.25497f,  3.04502f,   0.05446f,  -0.23253f, 0.00266f,
2358   0.07117f,  -2.78986f, -4.62953f, 1.45331f,   0.43923f,  0.92298f,  -0.47736f,
2359   1.49165f,  0.45942f,  -1.99787f, 3.33510f,   0.17234f,  0.04024f,  -1.42780f,
2360   0.23566f,  -0.90970f, 1.18041f,  -1.45865f,  2.30878f,  -1.28507f, 1.87290f,
2361   1.91186f,  4.74826f,  -3.70735f, 4.49808f,   -4.72275f, -0.02696f, -0.02642f,
2362   -0.06093f, -0.01121f, -0.70683f, 2.69737f,   -1.88563f, 2.48637f,  1.10922f,
2363   0.74624f,  0.40308f,  2.06396f,  1.39289f,   0.00909f,  -2.05271f, -1.53539f,
2364   -1.38323f, 0.83303f,  -0.32250f, 0.51172f,   3.91249f,  1.66373f,  1.13184f,
2365   -2.22874f, -1.13448f, -0.11185f, 0.19387f,   0.36770f,  -0.58933f, 0.22789f,
2366   1.17307f,  0.77461f,  0.20817f,  0.33417f,   0.54037f,  0.32961f,  -0.18456f,
2367   -9.78171f, -0.17216f, -3.44703f, -2.42158f,  0.51946f,  4.35949f,  -0.73335f,
2368   -1.61515f, -0.29622f, -0.37617f, -0.42316f,  0.74922f,  1.44386f,  3.92704f,
2369   -3.76274f, 4.19775f,  -3.86958f, 0.00074f,   -0.02418f, -0.12944f, 0.05857f,
2370   -0.85507f, 5.42546f,  5.40338f,  5.54347f,   5.59791f,  -0.01611f, 0.01618f,
2371   -0.01654f, -0.00270f, -0.39608f, -0.40410f,  -0.24551f, 0.09124f,  -0.34413f,
2372   -0.11504f, 0.12793f,  -0.31523f, 0.09148f,   -0.08567f, -0.05140f, -0.13310f,
2373   -0.81200f, 0.06882f,  -0.52537f, -12.74048f, -0.45395f, -4.04775f, -1.84887f,
2374   -1.02573f, 0.32788f,  1.06828f,  -1.25503f,  -0.42693f, 2.01413f,  -2.29103f,
2375   0.62271f,  1.11764f,  -1.83113f, -1.32325f,  -1.65651f, -2.87826f, 1.46910f,
2376   0.60885f,  0.16079f,  0.00171f,  -0.25658f,  -0.25465f, -0.14149f, 0.19497f,
2377   -0.07866f, -0.37080f, -0.05778f, -0.08870f,  -0.20491f, 0.84521f,  -0.18214f,
2378   -1.38441f, -1.08932f, -1.76627f, 0.73172f,   0.05967f,  1.28057f,  3.42722f,
2379   1.69287f,  0.77169f,  0.44528f,  1.85513f,   0.07840f,  1.31252f,  2.89948f,
2380   1.49489f,  0.15281f,  0.54708f,  -1.14185f,  -2.51063f, 0.36618f,  -0.55322f,
2381   0.96671f,  1.59470f,  1.38252f,  1.99697f,   0.03266f,  -0.23200f, -0.01127f,
2382   -0.18918f, -0.37598f, -0.03119f, -0.36039f,  -0.21192f, -0.11565f, -4.22635f,
2383   1.41252f,  0.56608f,  -0.08867f, 3.11924f,   -0.54597f, -0.12504f, -0.05289f,
2384   -0.28665f, -0.58297f, -1.18362f, -0.76201f,  -1.22011f, -0.58756f, 0.14740f,
2385   1.43971f,  0.98381f,  -0.02998f, -0.40678f,  -0.23047f, -0.12979f, 0.04003f,
2386   -0.22081f, -0.09294f, -0.15955f, -0.10379f,  -0.10192f, -1.51316f, 2.39482f,
2387   -1.69975f, 3.58976f,  -0.91032f, -0.03498f,  0.48982f,  -0.13418f, 0.76256f,
2388   1.61003f,  -2.01676f, -1.24430f, -3.25763f,  1.12314f,  2.00740f,  0.04613f,
2389   -0.14746f, -0.57374f, 3.44511f,  -0.56767f,  -4.08432f, -2.04894f, 2.35951f,
2390   -0.00458f, 0.18512f,  0.09916f,  -0.04084f,  -1.56207f, 1.38034f,  4.17302f,
2391   -1.47326f, -2.03530f, -0.00210f, 0.27469f,   -0.17423f, 0.86860f,  2.76195f,
2392   2.43269f,  -3.57331f, 2.08715f,  -1.44171f,  -0.17389f, 2.26157f,  -0.07852f,
2393   2.02519f,
2394 };
2395 
2396 static const float av1_rect_partition_nn_bias_128_layer0[NUM_NODES] = {
2397   2.53427f,  1.66678f,  -0.84914f, -0.15070f, -1.74769f, 0.45218f,  -0.26067f,
2398   2.05916f,  0.08978f,  5.30984f,  2.66243f,  -1.62740f, 0.70018f,  1.96403f,
2399   -4.97152f, -0.05425f, -3.84474f, -1.28006f, 3.47490f,  -0.08373f, 0.00225f,
2400   -1.40692f, -0.27569f, -0.30253f, 0.77377f,  -0.67636f, -0.26379f, 1.82348f,
2401   0.66120f,  0.61119f,  -1.42293f, 0.32676f,
2402 };
2403 
2404 static const float av1_rect_partition_nn_weights_128_layer1[NUM_NODES *
2405                                                             LABEL_SIZE] = {
2406   1.53453f,  -0.23707f, 7.88368f,  0.33340f,  0.97523f,  1.38538f,  -0.16746f,
2407   4.42070f,  3.18678f,  -5.03545f, -2.27029f, -3.75719f, -0.26850f, -4.93432f,
2408   -8.75673f, 0.27398f,  -5.77882f, -0.91616f, -2.62725f, -0.23961f, 0.31249f,
2409   3.32134f,  0.25375f,  -0.00394f, 2.30213f,  -0.14183f, 0.14544f,  -1.42830f,
2410   1.31101f,  3.99389f,  -0.00017f, -2.90184f, -2.11444f, 2.16734f,  -3.05133f,
2411   0.39206f,  4.61489f,  -2.88181f, -0.47745f, 2.86649f,  -1.20621f, 3.70550f,
2412   1.58029f,  -4.58731f, -2.29350f, -0.76930f, 5.19135f,  -0.22521f, -5.08782f,
2413   2.17316f,  1.30563f,  0.16777f,  -2.17767f, -2.09904f, 1.37001f,  0.25091f,
2414   -1.76743f, 1.57940f,  0.30544f,  -2.39895f, -0.08532f, -1.77122f, 1.84010f,
2415   -0.88449f, 0.79299f,  -1.35368f, -4.54110f, 0.02244f,  -5.11580f, 1.60883f,
2416   0.29352f,  -6.47042f, -1.81426f, 1.24013f,  0.90980f,  7.93977f,  2.12555f,
2417   5.24720f,  4.19508f,  0.21499f,  11.06045f, -0.74752f, 0.89396f,  0.26422f,
2418   1.72332f,  -1.25113f, -1.71136f, 0.13676f,  -0.07867f, -0.96929f, 0.19911f,
2419   3.58233f,  -0.76470f, -2.24162f, -2.87465f, 3.18736f,
2420 };
2421 
2422 static const float av1_rect_partition_nn_bias_128_layer1[3] = {
2423   1.09014f,
2424   -0.53317f,
2425   -0.55668f,
2426 };
2427 
2428 static const NN_CONFIG av1_rect_partition_nnconfig_128 = {
2429   FEATURE_SIZE,  // num_inputs
2430   LABEL_SIZE,    // num_outputs
2431   1,             // num_hidden_layers
2432   {
2433       NUM_NODES,
2434   },  // num_hidden_nodes
2435   { av1_rect_partition_nn_weights_128_layer0,
2436     av1_rect_partition_nn_weights_128_layer1 },
2437   { av1_rect_partition_nn_bias_128_layer0,
2438     av1_rect_partition_nn_bias_128_layer1 }
2439 };
2440 #undef FEATURE_SIZE
2441 #undef NUM_NODES
2442 #undef LABEL_SIZE
2443 
2444 // Below are the models used for simple_motion_search_based_split
2445 static const float av1_simple_motion_search_based_split_thresh_128 = 2.0f;
2446 static const float av1_simple_motion_search_based_split_thresh_64 = 2.0f;
2447 static const float av1_simple_motion_search_based_split_thresh_32 = 2.0f;
2448 static const float av1_simple_motion_search_based_split_thresh_16 = 2.0f;
2449 static const float av1_simple_motion_search_based_split_thresh_8 = 2.0f;
2450 
2451 // BLOCK_128X128
2452 #define NUM_HIDDEN_LAYERS_128 1
2453 #define NUM_FEATURES_128 6
2454 #define NUM_LAYER_0_UNITS_128 16
2455 #define NUM_LOGITS_128 1
2456 
2457 static const float av1_simple_motion_search_based_split_layer_0_kernel_128[] = {
2458   -0.807346f,  0.242298f,   12.9862f,   -1.19161f,  5.21734f,    -1.1363f,
2459   -2.39127f,   0.930915f,   -2.44285f,  -2.42966f,  5.73476f,    0.0506879f,
2460   -0.234878f,  -0.317875f,  0.361322f,  0.431648f,  -0.39105f,   -0.110225f,
2461   -2.46236f,   0.979713f,   -10.5596f,  -7.76653f,  -3.06518f,   2.42554f,
2462   0.0492961f,  -0.467176f,  0.130746f,  0.494527f,  -0.0336645f, 0.501755f,
2463   0.176486f,   -0.869541f,  7.77757f,   6.81303f,   6.00771f,    7.35696f,
2464   0.150731f,   -0.307017f,  -0.437639f, -0.082924f, 0.379107f,   0.452278f,
2465   -0.0143836f, -0.183691f,  -0.604698f, -9.2681f,   -2.06087f,   11.0256f,
2466   0.0487599f,  -0.249168f,  -0.180407f, 0.304772f,  0.218642f,   -0.406073f,
2467   -0.0289919f, -0.794381f,  5.45092f,   5.38374f,   3.25745f,    5.32903f,
2468   1.12718f,    -0.0215478f, 2.78552f,   4.8951f,    -0.959671f,  0.694264f,
2469   -0.0611219f, -0.331937f,  0.258252f,  -0.495331f, -0.285923f,  0.294713f,
2470   -0.119947f,  0.0753204f,  10.2021f,   -5.82147f,  -12.0137f,   3.0365f,
2471   0.366697f,   0.142683f,   -3.29731f,  -5.76651f,  -5.62578f,   10.9462f,
2472   -0.325459f,  0.092602f,   -0.868027f, -0.691768f, -0.292017f,  -0.00841203f,
2473   0.702545f,   -0.612227f,  -7.68881f,  9.52225f,   -1.18581f,   -2.56762f
2474 };
2475 
2476 static const float av1_simple_motion_search_based_split_logits_kernel_128[] = {
2477   0.364895f,    0.577553f,  0.115758f,  -0.999496f, 0.124885f, 3.23193f,
2478   -0.00386642f, 0.970794f,  0.136637f,  -4.28052f,  -1.49234f, 0.370436f,
2479   0.576981f,    -0.469656f, -0.124071f, 1.07669f
2480 };
2481 
2482 static const float av1_simple_motion_search_based_split_layer_0_bias_128[] = {
2483   1.32916f,    0.817212f,  0.0f,       -0.921066f, 0.0f,      3.57649f,
2484   -0.0204517f, 2.97286f,   0.0f,       5.49957f,   -8.14518f, 0.0f,
2485   1.30826f,    -0.349536f, -0.638933f, 5.4496f
2486 };
2487 
2488 static const float av1_simple_motion_search_based_split_logits_bias_128[] = {
2489   0.683442f
2490 };
2491 
2492 static const NN_CONFIG av1_simple_motion_search_based_split_nn_config_128 = {
2493   NUM_FEATURES_128,
2494   NUM_LOGITS_128,
2495   NUM_HIDDEN_LAYERS_128,
2496   {
2497       NUM_LAYER_0_UNITS_128,
2498   },
2499   {
2500       av1_simple_motion_search_based_split_layer_0_kernel_128,
2501       av1_simple_motion_search_based_split_logits_kernel_128,
2502   },
2503   {
2504       av1_simple_motion_search_based_split_layer_0_bias_128,
2505       av1_simple_motion_search_based_split_logits_bias_128,
2506   },
2507 };
2508 
2509 #undef NUM_HIDDEN_LAYERS_128
2510 #undef NUM_FEATURES_128
2511 #undef NUM_LAYER_0_UNITS_128
2512 #undef NUM_LOGITS_128
2513 
2514 // BLOCK_64X64
2515 #define NUM_HIDDEN_LAYERS_64 1
2516 #define NUM_FEATURES_64 6
2517 #define NUM_LAYER_0_UNITS_64 16
2518 #define NUM_LOGITS_64 1
2519 
2520 static const float av1_simple_motion_search_based_split_layer_0_kernel_64[] = {
2521   0.0345945f,  -0.394064f,  0.0919978f, 0.270358f,  -0.384502f, -0.504608f,
2522   -0.25759f,   0.155981f,   2.62567f,   -10.7204f,  -0.709802f, 8.15948f,
2523   0.589866f,   -0.445645f,  -1.68232f,  10.0061f,   -3.17671f,  4.87259f,
2524   -0.448886f,  -0.205568f,  -0.462388f, 0.385001f,  -0.451687f, 0.49602f,
2525   -0.256708f,  0.803322f,   3.25594f,   0.38541f,   -1.83867f,  -2.15132f,
2526   0.936059f,   -0.203056f,  -5.92959f,  -6.24554f,  -6.68631f,  -6.85977f,
2527   -0.0407565f, -0.258902f,  0.195053f,  -0.366515f, 0.339543f,  -0.433017f,
2528   -2.67026f,   0.385457f,   1.86683f,   1.9501f,    0.0381398f, 1.086f,
2529   -0.153729f,  0.173772f,   -42.9029f,  -36.8934f,  -2.892f,    -0.0540691f,
2530   0.77469f,    -0.380145f,  2.2689f,    -9.53332f,  1.15712f,   2.86601f,
2531   -0.437036f,  0.247132f,   -8.51058f,  -3.62972f,  -8.99449f,  -0.638738f,
2532   0.0609263f,  -0.0614603f, 5.42307f,   5.35926f,   5.27437f,   5.26599f,
2533   -0.0729677f, 0.0306104f,  -7.77867f,  5.03598f,   -8.17832f,  5.85461f,
2534   -0.253269f,  0.164582f,   -4.49713f,  3.83265f,   9.04851f,   -2.85668f,
2535   1.22618f,    0.166904f,   -1.51975f,  -4.01576f,  -1.44374f,  -2.22147f,
2536   -0.217072f,  -0.0984913f, -0.265515f, 0.360021f,  0.0779512f, 0.361516f
2537 };
2538 
2539 static const float av1_simple_motion_search_based_split_logits_kernel_64[] = {
2540   0.470821f, 0.474747f, -0.571292f, 0.403221f,  0.628966f,  -0.617029f,
2541   0.501105f, 0.499962f, -1.5451f,   -0.473518f, -0.730568f, -5.55817f,
2542   0.776761f, 0.42569f,  0.311925f,  0.469968f
2543 };
2544 
2545 static const float av1_simple_motion_search_based_split_layer_0_bias_64[] = {
2546   -0.134085f, 0.0758715f, 1.10419f,  0.0f,       -5.75737f, 1.65494f,
2547   0.0f,       3.44047f,   0.394852f, 3.43858f,   3.65871f,  -4.84987f,
2548   1.21207f,   -1.7705f,   -5.46469f, -0.0889634f
2549 };
2550 
2551 static const float av1_simple_motion_search_based_split_logits_bias_64[] = {
2552   -0.479491f
2553 };
2554 
2555 static const NN_CONFIG av1_simple_motion_search_based_split_nn_config_64 = {
2556   NUM_FEATURES_64,
2557   NUM_LOGITS_64,
2558   NUM_HIDDEN_LAYERS_64,
2559   {
2560       NUM_LAYER_0_UNITS_64,
2561   },
2562   {
2563       av1_simple_motion_search_based_split_layer_0_kernel_64,
2564       av1_simple_motion_search_based_split_logits_kernel_64,
2565   },
2566   {
2567       av1_simple_motion_search_based_split_layer_0_bias_64,
2568       av1_simple_motion_search_based_split_logits_bias_64,
2569   },
2570 };
2571 
2572 #undef NUM_HIDDEN_LAYERS_64
2573 #undef NUM_FEATURES_64
2574 #undef NUM_LAYER_0_UNITS_64
2575 #undef NUM_LOGITS_64
2576 
2577 // BLOCK_32X32
2578 #define NUM_HIDDEN_LAYERS_32 1
2579 #define NUM_FEATURES_32 6
2580 #define NUM_LAYER_0_UNITS_32 16
2581 #define NUM_LOGITS_32 1
2582 
2583 static const float av1_simple_motion_search_based_split_layer_0_kernel_32[] = {
2584   -1.61796f,   0.0585128f,  1.57904f,   1.52703f,   0.367779f, 0.220434f,
2585   1.66652f,    -1.77782f,   6.41118f,   4.16976f,   4.97299f,  4.84111f,
2586   -0.0956536f, -0.163284f,  -0.143662f, 0.129329f,  0.449659f, -0.528844f,
2587   -1.00067f,   1.17203f,    -4.26777f,  -4.78521f,  8.45658f,  -3.49498f,
2588   -1.78386f,   0.111488f,   4.176f,     6.31911f,   -10.5369f, 6.26983f,
2589   -1.32233f,   1.22999f,    -4.1666f,   -10.0359f,  -4.14779f, -10.4695f,
2590   1.83011f,    -0.333152f,  -9.87986f,  -8.11992f,  -8.2775f,  -7.79918f,
2591   -0.101404f,  0.00401393f, 8.89046f,   -7.32186f,  -6.59597f, 9.66257f,
2592   -1.1492f,    1.23067f,    -3.6341f,   6.59275f,   -3.2373f,  -3.42564f,
2593   0.371736f,   -0.140902f,  -2.75715f,  5.92487f,   -7.9185f,  9.13743f,
2594   -3.52698f,   -0.191044f,  5.96691f,   6.26327f,   4.36378f,  5.69354f,
2595   -0.608845f,  -0.191236f,  -0.482191f, -0.180474f, -3.8838f,  -3.92934f,
2596   -1.03191f,   0.994568f,   7.95516f,   -4.0035f,   -2.86266f, -4.96105f,
2597   1.75022f,    0.125058f,   -1.52159f,  -3.59304f,  -2.82634f, -2.49556f,
2598   -2.05557f,   -0.222577f,  3.7608f,    5.50475f,   2.7046f,   5.25952f,
2599   -1.91327f,   -0.0356497f, 1.47611f,   1.27499f,   -1.76108f, -0.578954f
2600 };
2601 
2602 static const float av1_simple_motion_search_based_split_logits_kernel_32[] = {
2603   -0.220382f, -0.693902f, 0.424827f, 0.379952f, -0.413791f, -0.326785f,
2604   -0.455086f, 0.242402f,  0.307986f, 0.175746f, 0.498901f,  -0.628053f,
2605   0.285447f,  0.230052f,  0.415151f, -0.842946f
2606 };
2607 
2608 static const float av1_simple_motion_search_based_split_layer_0_bias_32[] = {
2609   -1.80751f, 6.40356f,   -0.0512058f, -4.59163f, -0.369933f, -0.195755f,
2610   -0.16648f, -0.599755f, -5.35975f,   -1.21349f, 2.48414f,   1.07096f,
2611   -3.66684f, -6.17761f,  4.2159f,     -1.05286f
2612 };
2613 
2614 static const float av1_simple_motion_search_based_split_logits_bias_32[] = {
2615   -2.58676f
2616 };
2617 
2618 static const NN_CONFIG av1_simple_motion_search_based_split_nn_config_32 = {
2619   NUM_FEATURES_32,
2620   NUM_LOGITS_32,
2621   NUM_HIDDEN_LAYERS_32,
2622   {
2623       NUM_LAYER_0_UNITS_32,
2624   },
2625   {
2626       av1_simple_motion_search_based_split_layer_0_kernel_32,
2627       av1_simple_motion_search_based_split_logits_kernel_32,
2628   },
2629   {
2630       av1_simple_motion_search_based_split_layer_0_bias_32,
2631       av1_simple_motion_search_based_split_logits_bias_32,
2632   },
2633 };
2634 
2635 #undef NUM_HIDDEN_LAYERS_32
2636 #undef NUM_FEATURES_32
2637 #undef NUM_LAYER_0_UNITS_32
2638 #undef NUM_LOGITS_32
2639 
2640 // BLOCK_16X16
2641 #define NUM_HIDDEN_LAYERS_16 1
2642 #define NUM_FEATURES_16 6
2643 #define NUM_LAYER_0_UNITS_16 16
2644 #define NUM_LOGITS_16 1
2645 
2646 static const float av1_simple_motion_search_based_split_layer_0_kernel_16[] = {
2647   -0.611497f,  -0.0422086f, -0.555957f,   -0.632451f, -0.144179f, -0.152722f,
2648   -0.330265f,  -0.419866f,  0.287343f,    0.385295f,  -0.424486f, 0.424281f,
2649   2.27442f,    -2.47933f,   5.24731f,     4.33827f,   4.73215f,   3.41909f,
2650   1.16058f,    -0.364505f,  0.12207f,     -0.287749f, 0.0509783f, -0.0200119f,
2651   1.52907f,    -1.1905f,    -2.56978f,    -3.00186f,  -3.56084f,  -3.89276f,
2652   0.00365657f, 1.57125f,    -4.421f,      -2.48803f,  -2.51531f,  -4.28646f,
2653   2.52248f,    -1.03377f,   -1.09607f,    -1.44633f,  -1.58736f,  -1.25927f,
2654   -1.45841f,   -0.566619f,  -0.246166f,   -0.182289f, -0.238156f, 0.177991f,
2655   0.0112509f,  -0.17677f,   -0.485877f,   0.0812852f, 0.104975f,  0.222793f,
2656   -0.372858f,  -0.48624f,   -0.00870389f, -0.385019f, 0.405842f,  0.288523f,
2657   0.167374f,   -0.204208f,  -8.74148f,    -8.59267f,  -8.42492f,  -8.3778f,
2658   -5.57063f,   -0.406818f,  -0.873199f,   -0.896224f, -0.701479f, -0.985736f,
2659   -0.625956f,  -0.0446202f, -0.509987f,   -0.321804f, -0.470759f, -0.248556f,
2660   -0.369436f,  -0.160828f,  0.0591148f,   0.405218f,  0.142584f,  -0.130106f,
2661   0.125321f,   0.0888179f,  7.34822f,     -6.71488f,  -7.06592f,  6.33224f,
2662   0.0333619f,  -0.377782f,  0.160767f,    -0.128169f, -0.484818f, -0.311973f
2663 };
2664 
2665 static const float av1_simple_motion_search_based_split_logits_kernel_16[] = {
2666   -0.132207f,   0.15176f,   -0.680086f, 0.605921f, -0.43294f,  0.485811f,
2667   -0.306286f,   0.551368f,  0.413904f,  0.548748f, -0.437391f, 0.560778f,
2668   -0.00685266f, -0.558657f, 0.122127f,  0.260165f
2669 };
2670 
2671 static const float av1_simple_motion_search_based_split_layer_0_bias_16[] = {
2672   -0.200928f, -0.074132f, 8.69963f,    -9.00807f,  9.08983f, -6.83586f,
2673   -3.89329f,  10.4881f,   -0.0670618f, 0.0f,       9.21614f, 8.41773f,
2674   -0.145851f, 0.0f,       -1.43038f,   -0.0460311f
2675 };
2676 
2677 static const float av1_simple_motion_search_based_split_logits_bias_16[] = {
2678   -4.19885f
2679 };
2680 
2681 static const NN_CONFIG av1_simple_motion_search_based_split_nn_config_16 = {
2682   NUM_FEATURES_16,
2683   NUM_LOGITS_16,
2684   NUM_HIDDEN_LAYERS_16,
2685   {
2686       NUM_LAYER_0_UNITS_16,
2687   },
2688   {
2689       av1_simple_motion_search_based_split_layer_0_kernel_16,
2690       av1_simple_motion_search_based_split_logits_kernel_16,
2691   },
2692   {
2693       av1_simple_motion_search_based_split_layer_0_bias_16,
2694       av1_simple_motion_search_based_split_logits_bias_16,
2695   },
2696 };
2697 
2698 #undef NUM_HIDDEN_LAYERS_16
2699 #undef NUM_FEATURES_16
2700 #undef NUM_LAYER_0_UNITS_16
2701 #undef NUM_LOGITS_16
2702 
2703 #if !CONFIG_DISABLE_FULL_PIXEL_SPLIT_8X8
2704 // BLOCK_8X8
2705 #define NUM_HIDDEN_LAYERS_8 1
2706 #define NUM_FEATURES_8 6
2707 #define NUM_LAYER_0_UNITS_8 16
2708 #define NUM_LOGITS_8 1
2709 
2710 static const float av1_simple_motion_search_based_split_layer_0_kernel_8[] = {
2711   0.0370236f,   -0.580211f,  2.0134f,    1.69637f,    2.43181f,   -0.521648f,
2712   -0.00375187f, 0.122712f,   -4.74411f,  7.36187f,    5.42574f,   -5.53557f,
2713   0.0993344f,   -0.358843f,  0.0765453f, -0.615987f,  -0.754633f, -0.175846f,
2714   0.714976f,    0.492862f,   0.346604f,  -1.23922f,   -2.67031f,  2.12749f,
2715   1.71511f,     -1.4239f,    2.09396f,   2.42478f,    2.40151f,   2.90487f,
2716   0.540813f,    -0.0954257f, -4.57571f,  -4.88078f,   -4.62386f,  -5.75167f,
2717   1.35351f,     -1.08114f,   1.43744f,   1.44333f,    0.608153f,  0.193742f,
2718   -0.405512f,   -0.155164f,  0.0771456f, -0.473182f,  -0.057984f, 0.140435f,
2719   0.743021f,    -0.418589f,  -0.377622f, -0.531411f,  -0.668025f, -0.826607f,
2720   1.37834f,     -1.07753f,   0.870466f,  0.516756f,   0.708689f,  0.286795f,
2721   -3.97895f,    -0.338629f,  2.79427f,   1.80561f,    1.46275f,   1.50438f,
2722   0.0232533f,   -0.43174f,   -0.348251f, 0.0863006f,  0.0321103f, 0.129674f,
2723   -1.12024f,    -0.0990596f, -0.283472f, -0.238713f,  -0.239175f, -0.40816f,
2724   -0.00106566f, 0.0972736f,  5.19284f,   -3.70862f,   6.39657f,   -5.27588f,
2725   -2.08003f,    0.38825f,    2.38771f,   -1.27501f,   -2.45619f,  3.07324f,
2726   0.616966f,    -0.451472f,  -0.319365f, 0.00807278f, -0.303261f, -0.351679f
2727 };
2728 
2729 static const float av1_simple_motion_search_based_split_logits_kernel_8[] = {
2730   -0.625847f, 0.381323f, 0.342475f, 0.526161f,  -0.665965f, -0.515317f,
2731   -0.406218f, 0.568007f, 0.479397f, -0.426116f, 0.615638f,  0.338572f,
2732   0.185583f,  0.308031f, 0.260748f, 0.531619f
2733 };
2734 
2735 static const float av1_simple_motion_search_based_split_layer_0_bias_8[] = {
2736   4.73775f,  -1.12658f, -0.258038f, -6.06696f, 1.79131f, 2.49609f,
2737   4.28388f,  0.0f,      -4.63598f,  3.06034f,  5.31994f, -0.152142f,
2738   0.514738f, -1.30098f, 3.00296f,   -3.83481f
2739 };
2740 
2741 static const float av1_simple_motion_search_based_split_logits_bias_8[] = {
2742   -3.44508f
2743 };
2744 
2745 static const NN_CONFIG av1_simple_motion_search_based_split_nn_config_8 = {
2746   NUM_FEATURES_8,
2747   NUM_LOGITS_8,
2748   NUM_HIDDEN_LAYERS_8,
2749   {
2750       NUM_LAYER_0_UNITS_8,
2751   },
2752   {
2753       av1_simple_motion_search_based_split_layer_0_kernel_8,
2754       av1_simple_motion_search_based_split_logits_kernel_8,
2755   },
2756   {
2757       av1_simple_motion_search_based_split_layer_0_bias_8,
2758       av1_simple_motion_search_based_split_logits_bias_8,
2759   },
2760 };
2761 
2762 #endif
2763 
2764 // Model based on simple_motion_search
2765 
2766 // Thresholds for doing a single type of partition
2767 // TODO(chiyotsai@google.com): Set the thresholds for PARTITION_SPLIT.
2768 static const float av1_simple_motion_search_prune_part_only_thresh_128[10] = {
2769   1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
2770 };
2771 
2772 static const float av1_simple_motion_search_prune_part_only_thresh_64[10] = {
2773   1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
2774 };
2775 
2776 static const float av1_simple_motion_search_prune_part_only_thresh_32[10] = {
2777   1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
2778 };
2779 
2780 static const float av1_simple_motion_search_prune_part_only_thresh_16[10] = {
2781   1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
2782 };
2783 
2784 static const float av1_simple_motion_search_prune_part_only_thresh_8[10] = {
2785   1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f
2786 };
2787 
2788 // Thresholds for pruning a partition type
2789 static const float av1_simple_motion_search_prune_part_prune_thresh_128[10] = {
2790   0.0f, 0.0288721601835f, 0.0288721601835f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
2791   0.0f
2792 };
2793 
2794 static const float av1_simple_motion_search_prune_part_prune_thresh_64[10] = {
2795   0.0f, 0.0281573780991f, 0.0281573780991f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
2796   0.0f
2797 };
2798 
2799 static const float av1_simple_motion_search_prune_part_prune_thresh_32[10] = {
2800   0.0f, 0.0225501403434f, 0.0225501403434f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f,
2801   0.0f
2802 };
2803 
2804 static const float av1_simple_motion_search_prune_part_prune_thresh_16[10] = {
2805   0.0f,
2806   0.000961189195907f,
2807   0.000961189195907f,
2808   0.0f,
2809   0.0f,
2810   0.0f,
2811   0.0f,
2812   0.0f,
2813   0.0f,
2814   0.0f
2815 };
2816 
2817 static const float av1_simple_motion_search_prune_part_prune_thresh_8[10] = {
2818   0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f
2819 };
2820 
2821 // Mean and std
2822 static const float av1_simple_motion_search_prune_part_mean_128[25] = {
2823   13.292176f, 13.231236f, 11.098058f, 11.049944f, 10.481336f,
2824   10.431587f, 10.789337f, 10.732787f, 10.233817f, 10.173738f,
2825   12.214045f, 12.157505f, 11.863353f, 11.802220f, 12.204053f,
2826   12.152315f, 11.517566f, 11.465651f, 5.383040f,  0.757934f,
2827   4.012611f,  4.052191f,  0.853365f,  3.954503f,  3.944135f,
2828 };
2829 
2830 static const float av1_simple_motion_search_prune_part_std_128[25] = {
2831   2.589217f, 2.559396f, 2.268402f, 2.282274f, 3.341234f, 3.341994f, 3.033007f,
2832   3.041550f, 3.786247f, 3.784053f, 2.523459f, 2.511275f, 3.349364f, 3.340481f,
2833   2.390149f, 2.384226f, 3.599467f, 3.587460f, 2.319911f, 0.428335f, 1.241087f,
2834   1.208679f, 0.353742f, 1.228122f, 1.211777f,
2835 };
2836 
2837 static const float av1_simple_motion_search_prune_part_mean_64[25] = {
2838   11.439831f, 11.382639f, 9.647134f, 9.578121f, 9.146770f,
2839   9.084122f,  8.559063f,  8.499496f, 8.095865f, 8.041795f,
2840   10.547537f, 10.486240f, 9.362147f, 9.308391f, 10.548071f,
2841   10.484358f, 10.002225f, 9.944480f, 4.964504f, 0.897164f,
2842   3.306144f,  3.351039f,  0.928582f, 3.319739f, 3.287726f,
2843 };
2844 
2845 static const float av1_simple_motion_search_prune_part_std_64[25] = {
2846   2.033404f, 2.050657f, 2.064671f, 2.081519f, 2.916312f, 2.914649f, 3.628949f,
2847   3.618760f, 4.011421f, 3.996068f, 2.087127f, 2.103106f, 3.885277f, 3.876166f,
2848   2.035599f, 2.052976f, 3.052501f, 3.050985f, 2.232998f, 0.303745f, 1.111161f,
2849   1.081292f, 0.257521f, 1.112510f, 1.089404f,
2850 };
2851 
2852 static const float av1_simple_motion_search_prune_part_mean_32[25] = {
2853   9.862349f, 9.793658f, 8.043962f, 7.954083f, 8.058867f, 7.966165f, 8.046844f,
2854   7.956817f, 8.061414f, 7.967906f, 8.966450f, 8.890165f, 8.968315f, 8.891513f,
2855   8.953573f, 8.877070f, 8.974275f, 8.895363f, 4.387239f, 0.954143f, 2.701000f,
2856   2.751266f, 0.963302f, 2.716584f, 2.709725f,
2857 };
2858 
2859 static const float av1_simple_motion_search_prune_part_std_32[25] = {
2860   1.971555f, 1.985517f, 1.935986f, 1.944743f, 1.924122f, 1.932169f, 1.943151f,
2861   1.950612f, 1.931156f, 1.938242f, 1.987803f, 1.997670f, 2.000859f, 2.009913f,
2862   1.938270f, 1.949277f, 1.922999f, 1.933145f, 1.991504f, 0.209175f, 0.973824f,
2863   0.952221f, 0.188018f, 0.985295f, 0.946228f,
2864 };
2865 
2866 static const float av1_simple_motion_search_prune_part_mean_16[25] = {
2867   8.391692f, 8.303431f, 6.590342f, 6.459725f, 6.460719f, 6.333274f, 6.592615f,
2868   6.461661f, 6.464787f, 6.337191f, 7.499753f, 7.395166f, 7.503220f, 7.398344f,
2869   7.498312f, 7.395039f, 7.353743f, 7.253139f, 3.874267f, 0.979701f, 2.087404f,
2870   2.131698f, 0.981005f, 2.110868f, 2.106539f,
2871 };
2872 
2873 static const float av1_simple_motion_search_prune_part_std_16[25] = {
2874   1.865867f, 1.870012f, 1.773885f, 1.770447f, 1.972922f, 1.961361f, 1.777224f,
2875   1.772864f, 1.974519f, 1.962281f, 1.831632f, 1.831837f, 1.837595f, 1.837008f,
2876   1.822791f, 1.822053f, 2.074991f, 2.067200f, 1.676261f, 0.141022f, 0.840297f,
2877   0.829935f, 0.136507f, 0.828972f, 0.808563f,
2878 };
2879 
2880 static const float av1_simple_motion_search_prune_part_mean_8[25] = {
2881   6.997798f, 6.867032f, 5.134819f, 4.883330f, 5.134804f, 4.879707f, 5.140518f,
2882   4.886751f, 5.142186f, 4.885262f, 6.069946f, 5.896944f, 6.080442f, 5.906130f,
2883   6.077539f, 5.905929f, 6.083087f, 5.909298f, 3.552709f, 0.990654f, 1.497349f,
2884   1.531762f, 0.989606f, 1.496581f, 1.484139f,
2885 };
2886 
2887 static const float av1_simple_motion_search_prune_part_std_8[25] = {
2888   1.727562f, 1.725050f, 1.633396f, 1.618773f, 1.633586f, 1.620657f, 1.620798f,
2889   1.604892f, 1.621570f, 1.607439f, 1.691024f, 1.684225f, 1.676065f, 1.668442f,
2890   1.680016f, 1.672452f, 1.677775f, 1.671586f, 1.451902f, 0.096223f, 0.751190f,
2891   0.754040f, 0.101419f, 0.738239f, 0.729455f,
2892 };
2893 
2894 #define NUM_HIDDEN_LAYERS_128 1
2895 #define NUM_FEATURES_128 25
2896 #define NUM_LAYER_0_UNITS_128 8
2897 #define NUM_LOGITS_128 4
2898 
2899 static const float av1_simple_motion_search_prune_part_logits_kernel_128[] = {
2900   -0.129103f, 0.457758f,  -0.489986f, 0.65462f,   -0.184312f, 3.81202f,
2901   -0.444407f, -0.64198f,  -0.575008f, 0.0311711f, 0.525243f,  -20.892f,
2902   1.08811f,   -65.0976f,  -12.3973f,  -1.38278f,  -0.264233f, 0.241636f,
2903   -10.6925f,  -0.725414f, -18.8987f,  -40.2284f,  -16.08f,    0.995331f,
2904   1.47614f,   -0.964864f, 0.405506f,  0.140449f,  0.459534f,  -1.9093f,
2905   0.398452f,  0.696949f
2906 };
2907 
2908 static const float av1_simple_motion_search_prune_part_layer_0_bias_128[] = {
2909   1.22789f, -1.34527f, 0.759048f,  0.315086f,
2910   1.0834f,  -1.58019f, -0.465158f, 1.20716f
2911 };
2912 
2913 static const float av1_simple_motion_search_prune_part_layer_0_kernel_128[] = {
2914   -0.668677f,  0.58694f,    -0.417094f,   0.754735f,   -0.7859f,
2915   0.377479f,   -0.0415929f, -0.0140585f,  -0.730001f,  0.747528f,
2916   -0.135247f,  0.406505f,   -0.234184f,   0.956362f,   -0.637555f,
2917   0.791884f,   0.0303722f,  1.04424f,     -0.727859f,  -0.274321f,
2918   -0.122986f,  0.066312f,   -0.00559175f, -0.239643f,  -0.0188767f,
2919   -0.102787f,  -0.262967f,  0.071882f,    -0.283398f,  0.111607f,
2920   -0.425826f,  0.02699f,    0.108873f,    -0.180558f,  -0.0794057f,
2921   0.29665f,    -0.0252969f, -0.0266213f,  -0.277462f,  -0.361973f,
2922   0.512552f,   0.395011f,   -0.225876f,   0.301924f,   0.136954f,
2923   0.507259f,   1.23425f,    0.0137135f,   0.662572f,   0.591583f,
2924   0.101564f,   0.416805f,   -0.645081f,   -0.179086f,  -0.36747f,
2925   -0.332213f,  0.095177f,   0.220739f,    -0.153256f,  0.706155f,
2926   0.161701f,   0.696815f,   -1.21531f,    -0.115059f,  0.486764f,
2927   -0.396093f,  0.784883f,   0.535357f,    -0.278021f,  0.143496f,
2928   -0.44931f,   -0.144543f,  0.319326f,    0.0190167f,  -0.206295f,
2929   0.373995f,   -0.247897f,  -0.608095f,   -0.41796f,   -0.137129f,
2930   -0.709562f,  0.678273f,   0.537607f,    0.557474f,   0.453308f,
2931   0.21405f,    -0.0466495f, 0.519139f,    -0.168832f,  0.902911f,
2932   0.681131f,   -0.139876f,  -0.2052f,     -0.393271f,  0.262222f,
2933   -0.246246f,  -0.213993f,  0.646619f,    0.0496181f,  -0.00354157f,
2934   0.822927f,   0.0939522f,  0.180738f,    0.118355f,   0.120456f,
2935   -0.0472214f, -0.144958f,  0.173405f,    -0.886644f,  -0.0949769f,
2936   -0.813518f,  -0.3947f,    -0.128021f,   0.356196f,   0.469169f,
2937   -0.413702f,  1.04242f,    0.428853f,    -0.387293f,  0.0850877f,
2938   0.279409f,   -0.142276f,  0.0579376f,   0.211112f,   0.0703013f,
2939   -1.9274f,    -0.729147f,  0.534193f,    0.773586f,   0.922864f,
2940   0.642881f,   1.15127f,    0.621032f,    0.933942f,   1.01837f,
2941   -0.660282f,  -0.40059f,   -1.11279f,    -0.77088f,   -0.43349f,
2942   0.202361f,   -0.0840912f, 0.0935707f,   0.056333f,   -0.0779369f,
2943   0.0173447f,  -0.0104756f, 0.0115005f,   -0.0195593f, 0.03592f,
2944   -0.343454f,  -0.618048f,  0.258172f,    -0.412322f,  -0.0463746f,
2945   -0.0413654f, -0.0400194f, 0.615981f,    -0.452094f,  0.644555f,
2946   0.0822476f,  -0.359791f,  -0.0904274f,  0.209427f,   0.0116338f,
2947   -0.190978f,  0.890233f,   0.737769f,    -1.66663f,   -0.392605f,
2948   0.0785728f,  -0.224553f,  -0.128258f,   -0.227227f,  -0.0777773f,
2949   0.685976f,   0.347042f,   -0.555325f,   -0.249221f,  0.0919837f,
2950   -0.0660016f, -0.272316f,  0.0390632f,   -0.619624f,  -0.0565801f,
2951   0.585026f,   0.597375f,   0.54114f,     0.593389f,   0.604391f,
2952   0.0820294f,  -0.85339f,   -1.40741f,    -0.391675f,  0.0579205f,
2953   -0.197626f,  0.130044f,   -0.234488f,   -0.0373991f, -0.0717973f
2954 };
2955 
2956 static const float av1_simple_motion_search_prune_part_logits_bias_128[] = {
2957   1.58571f, -4.6314f, -2.00273f, 0.543699f
2958 };
2959 
2960 static const NN_CONFIG av1_simple_motion_search_prune_part_nn_config_128 = {
2961   NUM_FEATURES_128,
2962   NUM_LOGITS_128,
2963   NUM_HIDDEN_LAYERS_128,
2964   {
2965       NUM_LAYER_0_UNITS_128,
2966   },
2967   {
2968       av1_simple_motion_search_prune_part_layer_0_kernel_128,
2969       av1_simple_motion_search_prune_part_logits_kernel_128,
2970   },
2971   {
2972       av1_simple_motion_search_prune_part_layer_0_bias_128,
2973       av1_simple_motion_search_prune_part_logits_bias_128,
2974   },
2975 };
2976 
2977 #undef NUM_HIDDEN_LAYERS_128
2978 #undef NUM_FEATURES_128
2979 #undef NUM_LAYER_0_UNITS_128
2980 #undef NUM_LOGITS_128
2981 
2982 #define NUM_HIDDEN_LAYERS_64 1
2983 #define NUM_FEATURES_64 25
2984 #define NUM_LAYER_0_UNITS_64 32
2985 #define NUM_LOGITS_64 10
2986 
2987 static const float av1_simple_motion_search_prune_part_logits_kernel_64[] = {
2988   0.10424f,    -0.346025f,  0.534547f,   -0.385925f,  2.58341f,    -0.256414f,
2989   -0.232498f,  0.329823f,   -0.0777376f, -0.590939f,  0.062657f,   -0.628252f,
2990   0.0934588f,  2.04029f,    -0.224448f,  0.371168f,   -0.385348f,  -0.589883f,
2991   -3.73627f,   -0.943144f,  0.346409f,   -0.211215f,  -0.351008f,  0.418807f,
2992   0.943663f,   0.173267f,   1.16585f,    -0.0840888f, 0.227464f,   0.374412f,
2993   0.0422597f,  -0.338868f,  0.222576f,   0.431713f,   1.12366f,    0.00753411f,
2994   0.248412f,   -0.0902425f, 0.542455f,   -0.665629f,  -0.311245f,  -0.205639f,
2995   -0.447149f,  -0.0502733f, -0.290186f,  -0.794384f,  0.0940881f,  -0.0686117f,
2996   -0.0199961f, -0.587965f,  0.777096f,   -0.083381f,  -1.21282f,   0.652959f,
2997   -1.18238f,   0.539991f,   0.352497f,   -0.540076f,  -0.26222f,   -0.568556f,
2998   0.409102f,   -0.131146f,  -0.407161f,  -0.188287f,  -0.478657f,  0.000401932f,
2999   -0.689324f,  0.351064f,   -1.43704f,   -0.315185f,  -0.868726f,  0.376341f,
3000   -0.0566277f, 0.364831f,   0.611298f,   -0.495253f,  -0.0193132f, 0.617978f,
3001   0.189586f,   -0.236758f,  -0.608246f,  -0.149017f,  -1.78303f,   0.143023f,
3002   0.698386f,   -0.994086f,  -0.673327f,  0.233868f,   0.360425f,   0.0294123f,
3003   -0.248683f,  -0.148392f,  0.0861829f,  -0.190843f,  -0.414906f,  0.607378f,
3004   -0.756715f,  -0.511713f,  -0.321556f,  1.0078f,     -1.18141f,   0.519751f,
3005   0.834629f,   -0.359343f,  0.612262f,   -0.0730553f, 0.262935f,   0.488276f,
3006   0.387071f,   -1.44123f,   1.08269f,    0.554402f,   -0.069f,     0.14113f,
3007   0.323817f,   0.824314f,   -0.431417f,  -0.349448f,  0.950728f,   -0.587836f,
3008   -0.83914f,   -0.10844f,   0.26602f,    0.831933f,   -0.271315f,  0.231563f,
3009   0.417049f,   0.190627f,   -0.0940667f, 0.255363f,   -0.0741022f, -0.0987662f,
3010   -0.847522f,  0.00287554f, 0.0615741f,  -0.0832218f, 0.0847148f,  -0.392843f,
3011   -0.938068f,  -0.10621f,   -0.260859f,  -0.825175f,  -0.401039f,  0.315213f,
3012   -0.108269f,  0.288036f,   -8.66166f,   -0.970752f,  -0.66678f,   -0.593405f,
3013   -0.518294f,  -0.138722f,  -0.454698f,  -0.22969f,   -0.553006f,  -0.440111f,
3014   0.462661f,   -0.536854f,  0.0108295f,  -0.522888f,  0.00111157f, 0.229999f,
3015   0.0267768f,  0.176266f,   -1.57043f,   0.0318106f,  0.257534f,   -0.198583f,
3016   0.175564f,   -0.251465f,  -0.262441f,  -1.65283f,   -0.319603f,  -0.875282f,
3017   -0.301303f,  0.0170948f,  -0.227075f,  0.0299545f,  -4.98346f,   0.470046f,
3018   -1.28051f,   -0.213809f,  -0.486585f,  -0.906463f,  -0.169984f,  -0.333153f,
3019   -0.376733f,  0.108016f,   0.486744f,   -0.186936f,  -0.429259f,  0.056501f,
3020   -0.266545f,  0.265447f,   -0.137718f,  -0.490687f,  -0.935668f,  -0.16229f,
3021   -0.696932f,  0.173157f,   0.434959f,   -0.140595f,  0.345845f,   -1.08013f,
3022   -0.0205929f, -0.815874f,  -0.179812f,  0.02767f,    -0.141727f,  0.471936f,
3023   -7.29453f,   -1.04362f,   -0.745482f,  -0.28725f,   -0.214997f,  -0.0850651f,
3024   -0.748471f,  0.161325f,   -1.04387f,   -0.705305f,  0.489427f,   -0.765373f,
3025   -0.301576f,  0.0742467f,  -0.331282f,  0.0372328f,  -0.90298f,   -0.0608646f,
3026   -2.18756f,   0.170384f,   -0.258357f,  0.106287f,   -0.161684f,  -0.103799f,
3027   -0.127774f,  -0.156313f,  0.0705286f,  -0.977908f,  -0.281191f,  -0.056757f,
3028   -0.309474f,  0.050476f,   -9.78198f,   -2.42795f,   -0.289626f,  -1.07579f,
3029   -0.439256f,  -1.09948f,   -0.564671f,  0.0913182f,  -0.417216f,  -1.19909f,
3030   0.287063f,   0.402315f,   -0.17646f,   0.540488f,   0.00840239f, 0.397492f,
3031   0.702393f,   -0.10566f,   0.655296f,   -0.0443876f, 0.154918f,   -0.760479f,
3032   -0.0523153f, -0.366199f,  -1.08212f,   -0.398556f,  -0.415203f,  -1.10488f,
3033   0.208349f,   0.27079f,    0.101546f,   -0.205752f,  -13.7923f,   -0.218637f,
3034   -1.10077f,   0.355735f,   -0.306196f,  0.627434f,   -0.473101f,  -0.308027f,
3035   -1.12724f,   0.301597f,   0.660785f,   0.0576217f,  -0.155925f,  -0.56107f,
3036   -0.223537f,  0.114299f,   -0.53803f,   -0.252674f,  -2.66103f,   -0.185245f,
3037   -0.314673f,  0.403337f,   0.679821f,   -0.69231f,   0.506264f,   -0.999705f,
3038   -0.549097f,  0.353745f,   0.188249f,   0.414484f,   -0.615853f,  0.525681f,
3039   -5.23065f,   -3.05174f,   1.02074f,    -0.965499f,  -0.158947f,  0.0436088f,
3040   -0.485824f,  0.0375094f,  -1.39985f,   -0.481392f,  0.485785f,   -0.24874f,
3041   -0.359633f,  0.668108f
3042 };
3043 
3044 static const float av1_simple_motion_search_prune_part_layer_0_bias_64[] = {
3045   0.0735592f, -0.045064f, -0.0114103f, 1.39246f,    -0.683467f,  0.155765f,
3046   -0.667652f, -0.202425f, -0.585433f,  -0.146752f,  -0.0812931f, 0.580642f,
3047   0.578542f,  -0.831916f, 0.610063f,   0.0101856f,  -0.235863f,  0.538141f,
3048   -2.91334f,  -1.71887f,  0.126616f,   0.582497f,   -0.438879f,  0.221833f,
3049   0.850773f,  -0.280886f, 0.443233f,   -0.0964873f, -0.216161f,  0.34413f,
3050   0.656818f,  0.0169274f
3051 };
3052 
3053 static const float av1_simple_motion_search_prune_part_layer_0_kernel_64[] = {
3054   -0.310947f,   -0.232675f,    0.0171092f,    0.0834474f,   0.373977f,
3055   0.300429f,    0.215072f,     -0.454074f,    0.187565f,    0.282742f,
3056   0.562562f,    -0.0419322f,   0.000978486f,  -0.298267f,   0.216934f,
3057   -0.388722f,   -0.146866f,    -0.275946f,    0.202361f,    0.225847f,
3058   1.42868f,     0.473127f,     -0.145747f,    -0.104986f,   0.153459f,
3059   0.69382f,     0.162266f,     0.0207715f,    -0.45095f,    -0.412071f,
3060   -0.235109f,   -0.130199f,    0.231741f,     0.460193f,    0.0378202f,
3061   0.429516f,    0.387691f,     -0.272479f,    0.0723884f,   -0.453914f,
3062   -0.150618f,   -0.10745f,     -0.258615f,    0.0838312f,   -0.00554958f,
3063   0.105377f,    -0.0415479f,   0.13228f,      1.09044f,     -0.73053f,
3064   -0.422553f,   -0.435842f,    0.211416f,     0.420332f,    0.0181353f,
3065   -0.030891f,   0.522788f,     0.613526f,     0.374032f,    0.287986f,
3066   -0.403118f,   -0.287362f,    -1.11523f,     -0.577713f,   -0.020228f,
3067   0.86465f,     -0.0590579f,   0.341274f,     -0.0115644f,  -0.260236f,
3068   0.192123f,    -0.0849825f,   0.0501709f,    0.444382f,    0.0762727f,
3069   0.0926596f,   -0.101157f,    -0.142787f,    0.40861f,     0.555805f,
3070   -0.00614654f, -0.122846f,    0.203163f,     0.234266f,    0.409795f,
3071   -0.0206245f,  -0.224679f,    0.025081f,     0.518044f,    -0.287186f,
3072   0.016494f,    -0.0886331f,   0.236438f,     -1.01032f,    0.118332f,
3073   0.364217f,    0.061438f,     0.0381303f,    0.128418f,    0.0257077f,
3074   -0.975751f,   -0.694894f,    0.00351914f,   0.278179f,    0.29363f,
3075   0.525576f,    0.0604849f,    0.531734f,     0.406643f,    0.812497f,
3076   -0.403196f,   -0.16664f,     -0.620887f,    -0.428194f,   0.275401f,
3077   0.432063f,    -0.00378342f,  0.295758f,     0.105615f,    -0.00683626f,
3078   0.00396146f,  0.00598654f,   -0.0131701f,   -0.0115787f,  0.00386643f,
3079   -0.69686f,    -0.139623f,    -0.440817f,    0.0542873f,   0.217962f,
3080   0.527035f,    -0.0201046f,   0.0471354f,    0.0271858f,   -0.0775197f,
3081   -0.309797f,   0.184879f,     -0.232854f,    -0.407081f,   0.706227f,
3082   -0.0877534f,  0.306843f,     0.455075f,     -0.333961f,   0.0759148f,
3083   0.0444791f,   -0.0693626f,   -0.0850289f,   -0.513063f,   -0.643971f,
3084   -0.630279f,   -0.153889f,    0.123315f,     0.00548238f,  0.170707f,
3085   0.734339f,    -0.176988f,    0.322519f,     0.178365f,    0.183519f,
3086   -0.698683f,   -0.12043f,     -0.349914f,    -0.0696762f,  -0.53986f,
3087   -0.104738f,   1.05264f,      0.983568f,     -0.109035f,   0.0113748f,
3088   0.0815189f,   -0.0628812f,   0.0769389f,    0.010261f,    0.146573f,
3089   -0.433194f,   -0.211572f,    -0.000397392f, 0.445325f,    0.145091f,
3090   -0.0625902f,  0.29394f,      0.302315f,     0.0892226f,   -0.209504f,
3091   -0.0150374f,  0.242608f,     0.216223f,     0.366857f,    0.209829f,
3092   -0.540035f,   0.117599f,     -0.329315f,    0.0471133f,   -0.0115449f,
3093   -0.0638235f,  0.0527461f,    0.348149f,     0.360802f,    1.06624f,
3094   -0.615991f,   -0.341396f,    0.18972f,      0.0709888f,   -0.0414466f,
3095   -0.0193809f,  0.0938933f,    0.209058f,     0.575042f,    0.483608f,
3096   -0.285875f,   -0.115905f,    -0.363637f,    0.375425f,    0.336217f,
3097   0.0336358f,   -0.00265618f,  -0.406854f,    -0.792959f,   -0.219354f,
3098   0.0331615f,   0.0298859f,    -0.211446f,    -0.00280773f, -0.194011f,
3099   0.262109f,    0.548076f,     0.120183f,     -0.661603f,   0.241855f,
3100   -0.501428f,   0.00102718f,   -0.347331f,    -0.58306f,    0.0977254f,
3101   0.117491f,    0.0840667f,    0.00693675f,   0.000600294f, 0.649569f,
3102   -0.0553811f,  -0.197198f,    0.397236f,     -0.523737f,   -0.564192f,
3103   -0.374679f,   -0.249344f,    0.00861428f,   0.00393439f,  -0.0834608f,
3104   0.124389f,    -0.0393049f,   0.0425391f,    -0.153383f,   -0.182346f,
3105   0.420953f,    0.464221f,     0.288984f,     0.570921f,    -0.239965f,
3106   0.247239f,    -0.083434f,    0.714418f,     0.986323f,    -0.460244f,
3107   -0.260993f,   -0.947743f,    -1.0789f,      -0.0391231f,  0.612407f,
3108   -0.0306767f,  0.281419f,     0.0072426f,    -0.37623f,    0.188744f,
3109   0.221666f,    -0.424914f,    0.29703f,      0.261715f,    0.277809f,
3110   -0.0617616f,  -0.000611999f, -0.0547053f,   -0.0901018f,  -0.347669f,
3111   0.856072f,    0.596675f,     -0.467639f,    -1.09324f,    -0.184224f,
3112   -0.56051f,    -0.0144704f,   0.102894f,     -0.122982f,   -0.0020749f,
3113   -0.0423487f,  0.0328702f,    -0.0154263f,   0.0349021f,   -0.00315595f,
3114   0.0254802f,   -0.729191f,    0.207296f,     -0.0212349f,  -0.207078f,
3115   0.20636f,     -0.156883f,    0.429765f,     -0.42672f,    0.138775f,
3116   -0.0267343f,  0.631528f,     0.300646f,     -0.4793f,     -0.273833f,
3117   -0.0135367f,  -0.530819f,    -0.534881f,    0.830896f,    0.0266992f,
3118   0.473744f,    0.210334f,     0.0234739f,    0.255394f,    0.123531f,
3119   -0.489341f,   -0.796627f,    0.372617f,     0.190136f,    0.275342f,
3120   0.739505f,    0.402354f,     0.782806f,     0.437374f,    1.04948f,
3121   -0.55963f,    0.382704f,     -0.698321f,    0.0817868f,   -0.440108f,
3122   -0.0635004f,  -0.277851f,    -0.524194f,    0.286157f,    -0.01097f,
3123   -0.0293145f,  -0.0405071f,   -0.035662f,    -0.012871f,   -0.0516409f,
3124   -0.406671f,   0.709259f,     -0.525177f,    0.521123f,    -0.44813f,
3125   0.48412f,     -0.0546513f,   0.305253f,     -0.468328f,   0.316453f,
3126   -0.36307f,    0.497515f,     -0.0606276f,   0.315764f,    -0.422066f,
3127   0.554025f,    -0.679183f,    0.616914f,     0.00283324f,  -0.000643824f,
3128   0.0639999f,   0.0488285f,    -0.141031f,    0.068003f,    -0.0792678f,
3129   -0.425307f,   -0.152235f,    0.269917f,     -0.352327f,   0.44792f,
3130   -0.116514f,   -0.465868f,    0.154287f,     0.0161028f,   -0.16848f,
3131   -0.255487f,   0.189832f,     0.254883f,     0.0240822f,   0.432638f,
3132   -0.136564f,   0.137036f,     0.0375734f,    0.989246f,    -0.126287f,
3133   0.111416f,    -0.0271002f,   0.718755f,     -0.0412969f,  0.00645681f,
3134   0.253811f,    -0.0186998f,   0.691971f,     -0.282042f,   -0.0783915f,
3135   0.274592f,    -0.358449f,    0.34155f,      -0.186374f,   -0.136907f,
3136   -0.192334f,   -0.251168f,    -0.100874f,    -0.166578f,   -0.336507f,
3137   0.402373f,    0.173695f,     0.108788f,     0.00885581f,  -0.310063f,
3138   1.05545f,     0.0295867f,    0.180785f,     -0.173469f,   -0.469924f,
3139   -0.224155f,   0.665862f,     -0.126546f,    0.240691f,    -0.0415301f,
3140   -0.598534f,   0.0012723f,    -0.122297f,    -0.558947f,   0.268844f,
3141   0.241193f,    0.0524422f,    -0.1683f,      0.575588f,    -0.139012f,
3142   0.0636691f,   -0.446709f,    -0.094532f,    0.883809f,    -0.112981f,
3143   -0.224047f,   0.0811193f,    -0.140571f,    -0.09683f,    -0.0796143f,
3144   -0.102246f,   -0.863392f,    -0.0755124f,   0.23125f,     -0.0301361f,
3145   -0.153029f,   -0.172238f,    -0.0286382f,   -0.338495f,   -0.317216f,
3146   -0.146629f,   -0.242264f,    -0.702306f,    -0.285052f,   0.0623479f,
3147   0.265735f,    0.00674475f,   0.666196f,     0.883586f,    0.278416f,
3148   -0.341692f,   -0.509931f,    -0.156263f,    0.635885f,    -0.544143f,
3149   -0.572632f,   -0.213285f,    0.443396f,     -0.268329f,   0.0638439f,
3150   -0.185397f,   0.071126f,     0.386503f,     -0.402212f,   -0.140784f,
3151   -0.411661f,   0.049398f,     -0.0672907f,   -0.267034f,   -0.0560875f,
3152   0.0607937f,   0.0445484f,    -0.547651f,    0.574718f,    0.417189f,
3153   -0.0610166f,  0.0632293f,    0.391619f,     -0.00671215f, -0.136883f,
3154   -0.339346f,   0.0356183f,    0.511993f,     0.178676f,    0.286998f,
3155   0.136511f,    -0.00796929f,  0.203985f,     0.0423532f,   -0.175196f,
3156   0.378534f,    0.770417f,     0.593778f,     0.0256067f,   -0.82394f,
3157   -0.500691f,   -0.425725f,    -0.623708f,    -0.0406241f,  -0.00226464f,
3158   0.0207836f,   0.30732f,      -0.00784268f,  0.0065445f,   -0.0991039f,
3159   -0.20871f,    -0.206835f,    0.281219f,     0.119361f,    0.259346f,
3160   -0.102713f,   0.186488f,     -0.034455f,    -0.00198392f, -0.279107f,
3161   -0.638993f,   -0.374404f,    -0.48601f,     -0.262345f,   0.624532f,
3162   0.620632f,    -0.227014f,    0.433579f,     -0.0455096f,  1.22123f,
3163   -0.429156f,   0.12396f,      0.0815152f,    -0.0837355f,  0.0282623f,
3164   -0.407475f,   0.787321f,     -0.434974f,    0.312904f,    -0.230805f,
3165   0.213042f,    -0.250929f,    0.302997f,     -0.354709f,   0.0504905f,
3166   -0.561706f,   0.595558f,     0.374951f,     0.802969f,    -0.674902f,
3167   0.33136f,     0.156606f,     0.0218968f,    -0.694188f,   -0.0221949f,
3168   -0.00639123f, 0.0146536f,    0.0104145f,    0.021635f,    -0.0499428f,
3169   -0.575116f,   -0.239035f,    -0.0588276f,   0.599722f,    0.541932f,
3170   0.437433f,    0.716268f,     0.193207f,     0.548351f,    0.326951f,
3171   -0.197124f,   0.0355353f,    -0.0952009f,   -0.217265f,   -0.389789f,
3172   0.0528124f,   -0.21334f,     -0.190296f,    -1.17367f,    0.108905f,
3173   0.109397f,    -0.0192577f,   0.0343813f,    0.085004f,    -0.0556737f,
3174   -0.0411158f,  -0.534989f,    0.0361896f,    0.124415f,    0.291603f,
3175   -0.0311974f,  -0.326726f,    0.343131f,     0.0276456f,   -0.231827f,
3176   -0.373894f,   -0.208898f,    -0.273011f,    0.061323f,    -0.0910538f,
3177   -0.30746f,    -0.108644f,    -0.190736f,    1.58048f,     -0.0739711f,
3178   -0.0623489f,  -0.137967f,    -0.0601359f,   -0.133004f,   -0.0857153f,
3179   0.00955987f,  -0.365561f,    -0.0329051f,   0.463463f,    0.14758f,
3180   -0.512256f,   -0.227463f,    -0.26008f,     -0.567777f,   0.0646234f,
3181   1.02161f,     0.66157f,      -0.16733f,     0.264921f,    -0.242036f,
3182   0.214622f,    0.0712054f,    -0.260377f,    0.0849665f,   0.735094f,
3183   0.11001f,     0.297301f,     -0.333342f,    0.066978f,    -0.123625f,
3184   1.07596f,     0.401263f,     0.0800875f,    -0.340862f,   -0.115587f,
3185   -0.32692f,    -0.300842f,    0.0277397f,    0.0630788f,   -0.261198f,
3186   0.428695f,    -0.0544757f,   -0.124511f,    0.036992f,    0.126322f,
3187   0.0317603f,   0.0820762f,    0.117277f,     -1.14594f,    -0.108076f,
3188   -0.0258198f,  -0.00337525f,  -0.00512531f,  0.1274f,      -0.0660535f,
3189   -0.640733f,   0.197142f,     0.147278f,     0.489271f,    0.226507f,
3190   -0.0668414f,  0.0946318f,    0.0994164f,    -0.820516f,   0.512939f,
3191   -0.305172f,   -0.715187f,    -0.195125f,    0.279346f,    0.462144f,
3192   0.913882f,    -0.453879f,    0.0582033f,    -0.462866f,   0.0538736f,
3193   0.0115737f,   0.00626993f,   -0.0185185f,   0.0114601f,   -0.0181164f,
3194   0.41588f,     -0.0447331f,   0.611756f,     0.43385f,     0.834465f,
3195   0.122019f,    -0.352983f,    0.340429f,     -0.245425f,   -0.365328f,
3196   -0.521825f,   0.0371057f,    0.172188f,     -0.387949f,   0.221054f,
3197   0.0126359f,   0.422958f,     0.584198f,     -0.581498f,   -0.019466f,
3198   -0.0271737f,  -0.0740885f,   0.00540879f,   0.186086f,    -0.0324402f,
3199   -0.563462f,   -0.458759f,    -0.425296f,    -0.0118862f,  -0.641508f,
3200   0.0132084f,   0.0581128f,    0.0231444f,    0.468587f,    0.258838f,
3201   0.0296665f,   0.0562801f,    0.630014f,     0.381816f,    -0.269761f,
3202   -0.135515f,   0.046186f,     1.07632f,      -0.050616f,   0.104987f,
3203   0.29991f,     0.119316f,     0.117248f,     0.0795009f,   0.242573f,
3204   0.0416634f,   -0.0577639f,   -0.0974078f,   0.106255f,    -0.13098f,
3205   0.0141486f,   -0.00418257f,  0.144848f,     -0.463934f,   0.0452591f,
3206   0.252617f,    0.205222f,     -0.189843f,    0.0652245f,   -0.135386f,
3207   0.0500646f,   -0.200368f,    -0.0142312f,   -0.0286832f,  -0.254355f,
3208   -1.02752f,    -0.73549f,     0.0364518f,    0.0416227f,   -0.13185f,
3209   -0.0886515f,  -0.502314f,    -0.102916f,    0.410911f,    -0.355655f,
3210   0.400416f,    -0.340217f,    0.208829f,     0.245972f,    0.149739f,
3211   -0.49458f,    0.589482f,     0.550827f,     0.912709f,    -0.351275f,
3212   -0.128076f,   -0.285172f,    -0.672752f,    0.090583f,    -0.245286f,
3213   -0.737297f,   -0.201515f,    -0.025122f,    -0.109854f,   0.36738f
3214 };
3215 
3216 static const float av1_simple_motion_search_prune_part_logits_bias_64[] = {
3217   0.346819f,  0.442965f,  -0.0216032f,  0.0229235f, -0.402797f,
3218   -0.666074f, -0.455388f, -0.00353411f, -0.595511f, -0.845667f
3219 };
3220 
3221 static const NN_CONFIG av1_simple_motion_search_prune_part_nn_config_64 = {
3222   NUM_FEATURES_64,
3223   NUM_LOGITS_64,
3224   NUM_HIDDEN_LAYERS_64,
3225   {
3226       NUM_LAYER_0_UNITS_64,
3227   },
3228   {
3229       av1_simple_motion_search_prune_part_layer_0_kernel_64,
3230       av1_simple_motion_search_prune_part_logits_kernel_64,
3231   },
3232   {
3233       av1_simple_motion_search_prune_part_layer_0_bias_64,
3234       av1_simple_motion_search_prune_part_logits_bias_64,
3235   },
3236 };
3237 
3238 #undef NUM_HIDDEN_LAYERS_64
3239 #undef NUM_FEATURES_64
3240 #undef NUM_LAYER_0_UNITS_64
3241 #undef NUM_LOGITS_64
3242 
3243 #define NUM_HIDDEN_LAYERS_32 1
3244 #define NUM_FEATURES_32 25
3245 #define NUM_LAYER_0_UNITS_32 28
3246 #define NUM_LOGITS_32 10
3247 
3248 static const float av1_simple_motion_search_prune_part_logits_kernel_32[] = {
3249   0.486581f,    0.340847f,   -0.109226f,   0.467224f,   -0.541561f,
3250   0.0943619f,   -0.429442f,  -0.207442f,   0.959963f,   0.618666f,
3251   -0.0636751f,  0.144508f,   -0.0278289f,  0.332293f,   -0.751493f,
3252   0.245438f,    -0.917758f,  0.612128f,    -0.32648f,   0.534618f,
3253   -0.615239f,   2.71641f,    0.233759f,    0.820558f,   -0.249758f,
3254   -0.427783f,   -0.359361f,  0.0375732f,   0.806973f,   0.352512f,
3255   -0.0532192f,  0.0576861f,  -0.464178f,   -0.334877f,  -0.697042f,
3256   0.0538218f,   0.0919659f,  -0.00765812f, 0.0603847f,  -0.460315f,
3257   0.37979f,     -0.0867612f, -0.670683f,   -0.188619f,  -0.570586f,
3258   0.233418f,    0.153581f,   0.290905f,    -0.624885f,  -0.557842f,
3259   -0.555567f,   0.463773f,   -0.123909f,   -0.277731f,  0.0374468f,
3260   0.409903f,    0.287638f,   -0.593066f,   -0.223434f,  0.154263f,
3261   -0.250464f,   -0.077696f,  0.229652f,    -0.304174f,  0.308053f,
3262   0.33155f,     -0.502825f,  0.361216f,    -0.499294f,  0.00595444f,
3263   -0.307201f,   0.5766f,     -0.438384f,   -0.093701f,  -0.118586f,
3264   0.202337f,    -0.486623f,  0.261552f,    0.139756f,   -0.655642f,
3265   -0.0627001f,  -0.213053f,  -0.243037f,   0.205918f,   0.0718368f,
3266   0.188041f,    0.141529f,   -0.132239f,   0.425827f,   -0.218353f,
3267   0.153114f,    0.33268f,    0.0226116f,   0.167394f,   0.269854f,
3268   -0.457001f,   0.1973f,     -0.526087f,   0.467528f,   0.290934f,
3269   1.16267f,     0.0823663f,  -0.754389f,   -0.83716f,   0.270157f,
3270   -1.41229f,    0.148511f,   -0.286832f,   0.664796f,   0.492254f,
3271   0.360567f,    -0.533993f,  0.0435672f,   -0.103001f,  0.220668f,
3272   0.594621f,    -0.0213356f, -0.347638f,   -0.694457f,  0.0759505f,
3273   0.161358f,    -0.389384f,  -0.0455192f,  -0.61252f,   -0.174173f,
3274   -0.00788878f, -1.22487f,   0.332233f,    -0.0457021f, -0.225918f,
3275   -0.197657f,   -0.115408f,  -0.240589f,   -2.05681f,   0.00914629f,
3276   -1.92213f,    0.0268578f,  -0.49076f,    -0.0120123f, 0.291157f,
3277   0.267116f,    -0.0775724f, 0.181115f,    -0.392441f,  -0.488114f,
3278   -0.28842f,    -0.115465f,  0.128974f,    -0.0829899f, -0.14096f,
3279   -0.140145f,   -0.700281f,  0.0368945f,   -0.437598f,  0.243485f,
3280   -1.00301f,    0.332324f,   0.125014f,    -0.0604481f, -0.0652028f,
3281   -0.207295f,   -1.0209f,    -0.341525f,   0.191326f,   -0.147578f,
3282   0.0878327f,   0.129827f,   -0.0848319f,  0.187381f,   -1.28663f,
3283   0.00537885f,  -0.134277f,  -0.0411126f,  -0.3434f,    -0.0456494f,
3284   0.37861f,     0.409095f,   0.237177f,    -0.396855f,  -0.205418f,
3285   -1.31701f,    -0.319032f,  -0.123404f,   -0.240005f,  -0.305206f,
3286   -0.0258176f,  -0.26367f,   -0.142396f,   0.191672f,   -1.44061f,
3287   0.0554776f,   -0.571839f,  -0.284789f,   -0.425677f,  -0.0307376f,
3288   0.20275f,     -0.223146f,  0.144612f,    0.0212636f,  0.0238303f,
3289   -0.253802f,   -0.188922f,  -0.0637066f,  -0.340836f,  0.124774f,
3290   0.130474f,    -0.154099f,  -0.0292733f,  0.158148f,   -0.246989f,
3291   -0.259059f,   0.220224f,   0.228449f,    -0.41956f,   -0.321848f,
3292   -0.2396f,     -0.316449f,  -1.3363f,     0.0264099f,  -1.46865f,
3293   0.113073f,    0.0722885f,  -0.166986f,   -0.164877f,  0.0360911f,
3294   0.534472f,    -0.551152f,  -0.328501f,   0.0781121f,  -0.378112f,
3295   -0.459502f,   0.28015f,    -0.212302f,   -0.521641f,  0.618993f,
3296   -0.347709f,   0.266253f,   -0.0280894f,  0.348511f,   -0.0155031f,
3297   -0.100693f,   0.0447673f,  0.277519f,    -0.233998f,  -0.0796738f,
3298   -1.73644f,    -0.160776f,  0.53092f,     -0.180406f,  0.056447f,
3299   0.385356f,    -0.262337f,  -0.241479f,   -0.271426f,  -0.457354f,
3300   -0.266788f,   0.367371f,   -0.103065f,   0.47783f,    -0.188327f,
3301   -0.159636f,   0.00142907f, -0.409756f,   0.454889f,   -0.24566f,
3302   -0.0760084f,  0.286355f,   0.462102f,    0.0431695f,  -0.127395f,
3303   -0.200476f,   -0.350557f,  0.217275f,    -0.23975f,   0.255148f,
3304   -0.280626f,   0.42476f,    0.157411f,    0.0358675f,  -0.192591f
3305 };
3306 
3307 static const float av1_simple_motion_search_prune_part_layer_0_bias_32[] = {
3308   0.940498f,  0.15602f,   -0.234831f, 0.0268585f, 0.144769f,  0.243081f,
3309   0.611406f,  0.366093f,  0.361868f,  0.39668f,   0.401479f,  0.369467f,
3310   0.0909503f, 0.710595f,  0.032786f,  0.525891f,  -1.0232f,   0.732557f,
3311   -0.064425f, 0.865222f,  -0.042917f, -0.237191f, -0.527006f, -0.0172101f,
3312   0.59681f,   -0.472405f, 0.0969218f, -0.250624f
3313 };
3314 
3315 static const float av1_simple_motion_search_prune_part_layer_0_kernel_32[] = {
3316   0.355607f,    0.126701f,    -0.0825159f,  0.200675f,     -0.011308f,
3317   -0.280057f,   0.559816f,    0.142689f,    0.0422419f,    -0.151692f,
3318   -0.0275637f,  -0.283101f,   -0.20822f,    -0.200394f,    0.465427f,
3319   0.344491f,    -0.525319f,   -0.358813f,   -0.39767f,     0.0974486f,
3320   0.00559058f,  -0.00546089f, 0.0506486f,   0.114475f,     -0.0436463f,
3321   -0.574152f,   -0.376294f,   0.16563f,     -0.0967032f,   0.00579838f,
3322   0.0639909f,   -0.037129f,   0.407574f,    -0.231428f,    0.489326f,
3323   -0.221566f,   -0.270382f,   -0.784628f,   -0.155502f,    0.481698f,
3324   -0.0296057f,  0.431855f,    0.840807f,    0.112291f,     0.773874f,
3325   -0.0610936f,  -0.012892f,   0.365154f,    0.0267687f,    -0.0751114f,
3326   0.25043f,     0.516472f,    -0.186133f,   -0.12762f,     -0.168804f,
3327   -0.146309f,   0.139314f,    -0.367113f,   -0.601079f,    0.0559856f,
3328   0.176081f,    0.22397f,     0.434113f,    0.0363256f,    0.313051f,
3329   0.0143976f,   0.190076f,    0.474607f,    -0.681134f,    -0.0709097f,
3330   -0.253289f,   -0.216277f,   -0.0593789f,  -0.107795f,    -0.194842f,
3331   0.513945f,    0.239171f,    -0.720561f,   0.0136723f,    -0.391147f,
3332   -0.272043f,   -0.164766f,   0.124248f,    0.147178f,     -0.35497f,
3333   0.397725f,    -0.117603f,   0.262937f,    -0.331964f,    0.182418f,
3334   0.315671f,    -0.0385649f,  0.488769f,    -0.334568f,    0.00596018f,
3335   0.0661557f,   -0.0446985f,  -0.0928255f,  -0.0221032f,   -0.019045f,
3336   -0.20881f,    0.197907f,    -0.381881f,   0.0598071f,    -0.0434551f,
3337   0.159283f,    -0.110631f,   0.266996f,    -0.0265494f,   0.135199f,
3338   -0.00833162f, 0.804482f,    -0.114698f,   -0.15066f,     -0.479553f,
3339   0.448407f,    -0.344069f,   -0.0280952f,  -0.208211f,    -0.102269f,
3340   -0.679066f,   -0.37476f,    -0.0228875f,  0.0535049f,    0.111015f,
3341   -0.18125f,    -0.167584f,   0.0110497f,   0.262723f,     -0.413839f,
3342   -0.0611238f,  0.358499f,    0.0807514f,   0.208254f,     0.214499f,
3343   0.11137f,     -0.14262f,    -0.0513973f,  0.243718f,     -0.373716f,
3344   -0.00413366f, 0.216501f,    -0.164149f,   -0.064935f,    -0.0840282f,
3345   0.0566148f,   0.0377686f,   0.289835f,    0.769388f,     0.891198f,
3346   -0.592739f,   0.40744f,     -0.153095f,   0.657311f,     0.140737f,
3347   0.28209f,     0.158344f,    0.353546f,    0.0868246f,    0.116887f,
3348   0.402004f,    0.437184f,    0.589219f,    0.760594f,     -0.575419f,
3349   -0.754308f,   -0.709219f,   -0.297814f,   -0.418609f,    -0.0262104f,
3350   0.0411959f,   0.0597708f,   -0.143728f,   -0.136642f,    0.099614f,
3351   -0.257601f,   -0.2404f,     0.305893f,    0.254009f,     -0.0301398f,
3352   -0.0653091f,  -0.459002f,   -0.163404f,   0.123152f,     -0.0284252f,
3353   -0.457272f,   0.00788622f,  -0.828399f,   -0.0534199f,   0.586877f,
3354   0.982728f,    0.424581f,    0.0891856f,   0.383182f,     -0.122053f,
3355   0.0808408f,   -0.00384914f, -0.0560201f,  -0.0524772f,   -0.263444f,
3356   -0.239287f,   -0.882777f,   0.0180592f,   -0.0948711f,   -0.177946f,
3357   0.0296473f,   0.096082f,    0.0455604f,   -0.108608f,    0.00777951f,
3358   -0.140896f,   0.117187f,    -0.342467f,   -0.0691604f,   0.0761611f,
3359   -0.0892053f,  0.111386f,    -0.167456f,   1.40616f,      -0.00478793f,
3360   0.00547665f,  -0.0441829f,  0.0151323f,   -0.0674099f,   -0.0380578f,
3361   0.16072f,     0.31882f,     0.245486f,    -0.424318f,    0.101845f,
3362   -0.203343f,   -0.197402f,   -0.163025f,   -0.0771961f,   -0.264435f,
3363   0.319429f,    0.250076f,    0.782726f,    0.386003f,     0.00700673f,
3364   -0.375715f,   0.151453f,    -0.296265f,   -0.560183f,    -0.00767249f,
3365   -0.109593f,   -0.119419f,   -0.0161516f,  0.0380283f,    -0.156417f,
3366   0.131708f,    0.396268f,    -0.221796f,   0.232099f,     0.128852f,
3367   0.0567268f,   0.297297f,    0.173269f,    0.213411f,     0.0384426f,
3368   -0.290985f,   -0.0426841f,  -0.488292f,   -0.087101f,    -0.311582f,
3369   0.83009f,     -0.153163f,   0.903335f,    -1.15644f,     -0.0378635f,
3370   -0.0552129f,  -0.126362f,   -0.176945f,   0.0653115f,    0.0989368f,
3371   -0.333543f,   -0.330586f,   0.29775f,     -0.103535f,    0.210824f,
3372   -0.00300509f, 0.317105f,    0.216852f,    0.479718f,     0.0485808f,
3373   -0.15662f,    0.718199f,    0.327513f,    0.115169f,     -0.423598f,
3374   -0.456633f,   -0.575814f,   -0.494454f,   0.304411f,     0.0493055f,
3375   -0.381171f,   0.467251f,    -0.122872f,   -0.167441f,    0.017253f,
3376   -0.0583646f,  -0.1586f,     0.214046f,    -0.0284424f,   -0.217112f,
3377   0.606567f,    -0.107533f,   0.36615f,     -0.0709227f,   0.604761f,
3378   -0.244657f,   -0.296651f,   -0.595611f,   -0.156629f,    -0.693468f,
3379   -0.310603f,   0.499272f,    0.282941f,    0.295043f,     -0.178704f,
3380   0.281186f,    0.014329f,    -0.120819f,   0.154234f,     0.0131325f,
3381   -0.472231f,   -0.631281f,   0.422955f,    0.711432f,     -0.118025f,
3382   0.0864996f,   0.343971f,    -0.301477f,   -0.246638f,    0.165068f,
3383   0.218044f,    0.224236f,    -0.0848522f,  0.00671216f,   0.401141f,
3384   -0.218857f,   -0.0298495f,  -0.135725f,   -0.377618f,    0.022473f,
3385   0.106955f,    -0.0582005f,  0.0468484f,   -0.0217442f,   0.130911f,
3386   -0.0926905f,  0.383007f,    -0.159353f,   -0.222711f,    -0.0286419f,
3387   0.372315f,    -0.469095f,   0.797571f,    -0.301315f,    0.239327f,
3388   -0.997507f,   -0.363409f,   0.353717f,    0.676686f,     -0.0500028f,
3389   0.0638539f,   -0.431927f,   0.243852f,    0.000884826f,  -0.00166585f,
3390   0.0613292f,   -0.029558f,   -0.0248432f,  -0.0125607f,   -0.0309674f,
3391   -0.743308f,   0.0409806f,   0.0921015f,   0.167816f,     0.406849f,
3392   0.095677f,    0.0308913f,   0.139956f,    -0.400472f,    0.396617f,
3393   0.936517f,    0.355057f,    -0.423816f,   -0.232472f,    -0.220188f,
3394   -0.399746f,   -0.409623f,   -0.158797f,   0.361153f,     0.0327019f,
3395   0.0690844f,   -0.032197f,   0.0248558f,   0.00438518f,   0.0222724f,
3396   -0.326832f,   -0.314295f,   0.156563f,    0.0562703f,    0.332694f,
3397   0.299424f,    0.228206f,    0.322038f,    0.0136098f,    0.0060297f,
3398   -0.165851f,   -0.306512f,   0.0796508f,   -0.37158f,     0.239395f,
3399   -0.349442f,   0.198515f,    -0.253854f,   -1.13694f,     0.0202873f,
3400   -0.0504009f,  -0.130528f,   -0.017126f,   -0.0370001f,   -0.087458f,
3401   -0.119952f,   -0.130404f,   0.0333733f,   -0.184736f,    0.182162f,
3402   0.227776f,    -0.166563f,   -0.156162f,   0.118215f,     -0.220183f,
3403   0.00474779f,  -0.107792f,   0.260493f,    0.11884f,      0.156587f,
3404   0.303936f,    -0.131788f,   -0.314774f,   0.310606f,     0.0935523f,
3405   0.790767f,    0.26461f,     0.0236426f,   0.0629469f,    0.0344072f,
3406   -0.151513f,   0.211498f,    0.0245435f,   0.0629973f,    0.052019f,
3407   -0.03308f,    0.123487f,    0.0885027f,   0.159172f,     -0.0510615f,
3408   0.0298033f,   -0.130515f,   -0.121799f,   -0.104915f,    0.208822f,
3409   -0.310496f,   -0.314106f,   0.303307f,    -0.0196736f,   0.0420045f,
3410   0.461777f,    -0.433699f,   0.00345407f,  0.703139f,     -0.655637f,
3411   -0.210767f,   -0.201278f,   0.163694f,    -0.236534f,    0.300877f,
3412   0.0769982f,   -0.282453f,   0.149721f,    -0.0303466f,   -0.191473f,
3413   -0.406056f,   -0.213472f,   0.1619f,      -0.245953f,    0.00544399f,
3414   -0.121434f,   0.193012f,    -0.307165f,   1.45431f,      -0.161468f,
3415   -0.12444f,    -0.146129f,   -0.0528212f,  -0.0925165f,   -0.134528f,
3416   -0.479475f,   0.315525f,    0.133845f,    0.382158f,     -0.0799693f,
3417   -0.151041f,   0.255772f,    0.409536f,    -0.240663f,    -0.323741f,
3418   -0.205876f,   0.03699f,     -0.217541f,   0.108511f,     0.640628f,
3419   0.705993f,    -0.423899f,   -0.78314f,    -0.100733f,    -0.00859087f,
3420   0.0251879f,   0.0458335f,   0.00210128f,  -0.047576f,    -0.0560518f,
3421   -1.23869f,    -0.829914f,   0.0346551f,   0.350505f,     0.193688f,
3422   0.459154f,    0.137898f,    0.503818f,    0.260867f,     0.649539f,
3423   0.0150802f,   0.0239274f,   -0.276069f,   -0.0621478f,   -0.193106f,
3424   -0.0375665f,  -0.654529f,   0.189493f,    0.446625f,     -0.0208265f,
3425   0.019838f,    -0.0201955f,  0.00180428f,  -0.0110678f,   -0.0172414f,
3426   0.0276489f,   -0.252882f,   -0.0351807f,  -0.0518874f,   0.279098f,
3427   -0.245122f,   0.101287f,    -0.114202f,   -0.0812187f,   0.572429f,
3428   -0.0821731f,  0.564183f,    0.0222552f,   0.190111f,     -0.0417497f,
3429   -0.00385925f, -0.182995f,   -0.240482f,   -0.291572f,    -0.0450444f,
3430   0.0962974f,   -0.165973f,   -0.0954637f,  -0.163841f,    -0.833405f,
3431   -1.31541f,    -0.336473f,   -0.0920702f,  0.816105f,     0.393377f,
3432   0.0340241f,   -0.0844545f,  0.61729f,     -0.17596f,     0.241149f,
3433   -0.42825f,    -0.59091f,    -0.290702f,   0.0796465f,    0.0982819f,
3434   0.466934f,    0.261666f,    0.0373333f,   0.332509f,     -0.0266694f,
3435   -0.0476951f,  -0.00642167f, -0.0132542f,  -0.000320841f, 0.00475532f,
3436   0.000502778f, 0.296534f,    -0.13297f,    -0.113082f,    -0.327923f,
3437   0.35901f,     -0.302246f,   0.189799f,    -0.37994f,     0.16107f,
3438   -0.20414f,    0.548575f,    -0.460821f,   0.591878f,     -0.213113f,
3439   -0.169373f,   -0.07332f,    0.228841f,    0.682302f,     -0.0665316f,
3440   -0.142456f,   -0.0873117f,  0.00607451f,  0.0376443f,    0.0536673f,
3441   -0.0109536f,  -0.400279f,   0.550058f,    0.820871f,     -0.666373f,
3442   -0.471962f,   -0.315925f,   -0.313142f,   0.952742f,     0.473928f,
3443   -0.119006f,   0.153241f,    -0.0383078f,  0.631869f,     -0.343423f,
3444   -0.233473f,   -0.218195f,   -0.077688f,   -0.728291f,    0.0382408f,
3445   -0.00662886f, -0.0419666f,  0.0309776f,   -0.0281592f,   0.0154229f,
3446   -0.198534f,   0.0206324f,   0.0152272f,   -0.235067f,    0.0330486f,
3447   0.139198f,    -0.0612118f,  0.133154f,    -0.258675f,    0.0900275f,
3448   -0.127771f,   0.157322f,    -0.00767807f, -0.329258f,    0.327458f,
3449   0.0528581f,   -0.181125f,   0.409995f,    -0.162979f,    -0.0193475f,
3450   0.186009f,    0.0519501f,   0.651877f,    -0.37821f,     -1.10341f,
3451   -0.189776f,   -0.0922788f,  0.460256f,    0.168011f,     0.440295f,
3452   0.478135f,    0.374573f,    0.384048f,    0.116953f,     0.68886f,
3453   -0.427727f,   -0.36676f,    -0.500013f,   -0.228685f,    -0.218859f,
3454   0.208396f,    -0.0173765f,  -0.0680241f,  -0.00538013f,  -0.0674409f,
3455   -0.092764f,   0.0295707f,   -0.0462887f,  -0.00636006f,  0.0334169f
3456 };
3457 
3458 static const float av1_simple_motion_search_prune_part_logits_bias_32[] = {
3459   0.176459f,  0.154405f, 0.281821f,  0.375264f,  -0.882863f,
3460   -0.240261f, -1.17075f, -0.280216f, -0.743836f, -0.317511f
3461 };
3462 
3463 static const NN_CONFIG av1_simple_motion_search_prune_part_nn_config_32 = {
3464   NUM_FEATURES_32,
3465   NUM_LOGITS_32,
3466   NUM_HIDDEN_LAYERS_32,
3467   {
3468       NUM_LAYER_0_UNITS_32,
3469   },
3470   {
3471       av1_simple_motion_search_prune_part_layer_0_kernel_32,
3472       av1_simple_motion_search_prune_part_logits_kernel_32,
3473   },
3474   {
3475       av1_simple_motion_search_prune_part_layer_0_bias_32,
3476       av1_simple_motion_search_prune_part_logits_bias_32,
3477   },
3478 };
3479 
3480 #undef NUM_HIDDEN_LAYERS_32
3481 #undef NUM_FEATURES_32
3482 #undef NUM_LAYER_0_UNITS_32
3483 #undef NUM_LOGITS_32
3484 
3485 #define NUM_HIDDEN_LAYERS_16 1
3486 #define NUM_FEATURES_16 25
3487 #define NUM_LAYER_0_UNITS_16 32
3488 #define NUM_LOGITS_16 10
3489 
3490 static const float av1_simple_motion_search_prune_part_logits_kernel_16[] = {
3491   -0.520913f,   0.395611f,    0.0369091f,   -0.318591f,  -0.463252f,
3492   0.134992f,    -0.43154f,    -0.0739112f,  -0.118817f,  0.476373f,
3493   -0.281406f,   0.3413f,      0.456255f,    0.33307f,    0.2942f,
3494   0.1317f,      0.498113f,    1.95406f,     -0.165726f,  -0.219306f,
3495   -0.302656f,   -1.31157f,    -0.433662f,   0.151716f,   -0.214817f,
3496   0.504523f,    -0.710049f,   0.359616f,    -0.412695f,  -0.103193f,
3497   0.341912f,    0.351378f,    -0.181486f,   0.573862f,   -0.0396254f,
3498   -0.17855f,    -0.276163f,   0.0367465f,   -0.353905f,  -0.204689f,
3499   0.309581f,    -0.0439686f,  -0.147855f,   0.152745f,   0.290871f,
3500   0.131049f,    -0.27808f,    -0.142997f,   0.207843f,   -1.23074f,
3501   -0.267714f,   -0.336923f,   0.313781f,    -0.61488f,   -0.161984f,
3502   0.238059f,    -0.0879942f,  -0.085543f,   -0.260156f,  -0.13614f,
3503   -0.242196f,   0.201216f,    -0.248691f,   0.0936671f,  -0.350522f,
3504   -0.35002f,    -0.156583f,   -0.00579001f, 0.300578f,   -0.341269f,
3505   -0.290712f,   0.354802f,    -0.31629f,    0.509107f,   -0.236953f,
3506   -0.0923519f,  0.544509f,    -0.280991f,   -0.017437f,  -0.202721f,
3507   -0.116388f,   -0.7191f,     0.324586f,    0.254249f,   0.125505f,
3508   0.00658697f,  -0.333322f,   -0.126537f,   -0.140004f,  -0.0241202f,
3509   -0.172466f,   0.210035f,    -0.270833f,   0.0579044f,  0.0950352f,
3510   -0.120382f,   0.063292f,    -0.394925f,   0.482165f,   0.147753f,
3511   0.331465f,    -0.187444f,   0.1083f,      0.414028f,   0.279238f,
3512   -0.486889f,   -0.674349f,   -0.313656f,   -0.131186f,  -0.100662f,
3513   0.238191f,    -1.19083f,    -0.30667f,    -2.4324f,    0.235311f,
3514   0.108605f,    1.67197f,     0.476157f,    0.30055f,    0.0839538f,
3515   0.408469f,    -0.473517f,   0.560283f,    -0.0188136f, 0.273824f,
3516   -0.43707f,    -0.0346978f,  -0.438315f,   -0.0196275f, -0.0567921f,
3517   -0.220166f,   0.216175f,    -0.0180461f,  0.0116429f,  -0.0096949f,
3518   -0.32613f,    0.176829f,    -0.243563f,   -0.240972f,  -0.621819f,
3519   -0.00619648f, -0.145525f,   0.124324f,    -0.0306925f, 0.172208f,
3520   -2.04631f,    -0.200087f,   -0.594135f,   -0.352303f,  -0.309826f,
3521   0.0922786f,   -0.698371f,   -0.0366823f,  0.0244036f,  0.338775f,
3522   -0.115947f,   0.144971f,    -0.0607037f,  -0.762412f,  0.0125584f,
3523   -0.262427f,   -0.0830273f,  -0.291252f,   -0.176059f,  -0.203983f,
3524   0.0871455f,   -0.0894925f,  0.0426263f,   -0.060001f,  -0.542355f,
3525   -0.407837f,   -0.0419273f,  0.226608f,    -0.114844f,  0.158733f,
3526   -0.187237f,   0.113163f,    -1.86337f,    -0.367544f,  -0.547048f,
3527   -0.24192f,    -0.226764f,   0.090912f,    0.819604f,   0.433766f,
3528   -0.841657f,   0.446987f,    -0.622761f,   -0.0296385f, -0.130176f,
3529   -0.0518136f,  -0.640326f,   -0.330107f,   -0.137832f,  -0.0119033f,
3530   0.39401f,     0.111331f,    -0.141367f,   -0.230289f,  0.171054f,
3531   -0.924059f,   -0.107317f,   -0.347983f,   0.0261109f,  0.423002f,
3532   -0.305817f,   0.247696f,    0.0436002f,   0.0305862f,  -1.52448f,
3533   -0.595587f,   -0.155552f,   -1.11949f,    -0.513937f,  0.138347f,
3534   -0.301487f,   0.352144f,    -0.615801f,   0.0326701f,  -0.215322f,
3535   -0.0608176f,  -0.416557f,   -0.306073f,   -0.441512f,  -0.0569277f,
3536   -0.709768f,   -0.602527f,   -0.311134f,   0.152471f,   -0.255299f,
3537   0.354505f,    0.194464f,    0.0144251f,   0.110732f,   -0.4452f,
3538   -0.804814f,   0.205325f,    -0.0957486f,  0.502684f,   0.09112f,
3539   -0.533087f,   -1.77979f,    0.556992f,    -0.176157f,  -0.642633f,
3540   0.11553f,     -0.232561f,   0.161277f,    -0.0631125f, -0.20759f,
3541   0.489253f,    -0.067533f,   0.0231024f,   -0.179831f,  -0.272985f,
3542   -0.390059f,   0.3089f,      0.185733f,    -0.257065f,  -0.508838f,
3543   -0.550028f,   0.0665621f,   -0.138288f,   -0.413188f,  0.191193f,
3544   -1.32969f,    -0.431025f,   0.270242f,    -0.340062f,  0.0817257f,
3545   0.0376051f,   -0.18633f,    0.0828274f,   0.00670051f, -0.431295f,
3546   -0.450316f,   -0.173042f,   -0.322248f,   0.370628f,   0.10019f,
3547   0.317293f,    -0.266613f,   0.0752441f,   -0.425656f,  -0.112223f,
3548   0.557991f,    -0.324368f,   -0.195261f,   -0.0526129f, -0.807472f,
3549   -0.387466f,   0.192186f,    0.353213f,    -0.120238f,  0.107686f,
3550   0.200678f,    -0.75363f,    0.466857f,    -0.282345f,  -0.0849236f,
3551   -0.0490695f,  -0.00643182f, 0.123047f,    -0.207805f,  -0.130456f,
3552   -1.09455f,    0.340973f,    0.334784f,    0.0706643f,  -1.65681f,
3553   -0.319952f,   -0.198514f,   -0.0787972f,  0.089524f,   0.0531034f,
3554   -0.202705f,   -0.0852339f,  -0.62572f,    -0.0734234f, -0.838088f
3555 };
3556 
3557 static const float av1_simple_motion_search_prune_part_layer_0_bias_16[] = {
3558   -0.0616197f, 0.939947f, 0.521161f,  0.213886f,  0.130324f,  -0.127443f,
3559   -0.0538715f, 0.708746f, 0.445031f,  0.418781f,  -0.114539f, 0.521941f,
3560   1.13719f,    0.606545f, -0.32193f,  -0.150788f, 0.158487f,  -0.224005f,
3561   0.654715f,   0.115729f, -0.286506f, -2.06223f,  0.0117697f, 0.503905f,
3562   -0.102339f,  0.653256f, -0.813561f, 0.905235f,  -0.417269f, -0.206265f,
3563   0.661496f,   0.95533f
3564 };
3565 
3566 static const float av1_simple_motion_search_prune_part_layer_0_kernel_16[] = {
3567   -0.203489f,   0.00686229f,  -0.161414f,   0.0637276f,   0.27516f,
3568   0.512219f,    0.164205f,    0.00326062f,  -0.41914f,    -0.400334f,
3569   0.554419f,    0.715772f,    -0.295569f,   -0.703503f,   0.0137744f,
3570   -0.0934259f,  0.174234f,    -0.148618f,   -0.0360558f,  -0.0986598f,
3571   -0.138502f,   -0.0770713f,  0.122922f,    -0.00784415f, 0.0953234f,
3572   -0.255754f,   -0.310967f,   0.185306f,    0.464554f,    0.147338f,
3573   -0.0612304f,  0.164783f,    0.301097f,    0.161364f,    -0.12723f,
3574   -0.0265984f,  -0.471361f,   0.0578776f,   -0.362865f,   0.425789f,
3575   0.402758f,    -0.190235f,   0.00549738f,  -0.570908f,   1.27206f,
3576   0.048868f,    -0.0097675f,  0.0708324f,   0.0456103f,   0.0149062f,
3577   -0.563032f,   -0.420573f,   0.107278f,    0.0938258f,   0.142712f,
3578   -0.00251036f, -0.250583f,   0.522272f,    0.0113175f,   0.126751f,
3579   -0.433028f,   -0.035542f,   -0.536686f,   -0.0668722f,  0.253094f,
3580   0.254007f,    -0.435505f,   0.343001f,    0.0531542f,   -0.361914f,
3581   -0.102664f,   0.0404874f,   0.132686f,    0.0762298f,   0.0236971f,
3582   -0.419454f,   0.230877f,    -0.223714f,   0.037813f,    0.0818604f,
3583   0.383705f,    -0.235028f,   -0.0554801f,  0.429851f,    0.0845829f,
3584   0.166295f,    0.355111f,    -0.421197f,   0.298949f,    0.0218224f,
3585   0.445705f,    -0.392217f,   -0.429578f,   -0.076276f,   -0.0963531f,
3586   -0.631425f,   -0.225977f,   8.06349e-06f, 0.0676679f,   0.0779651f,
3587   0.0706891f,   0.101377f,    0.517103f,    0.0945502f,   -0.52522f,
3588   -0.312022f,   0.0358089f,   0.616509f,    -0.0507444f,  -0.465814f,
3589   -0.0326024f,  0.591298f,    0.188544f,    -0.0633316f,  -0.199987f,
3590   0.403118f,    -0.511281f,   -0.696263f,   0.112996f,    0.103875f,
3591   0.0495595f,   -0.0107449f,  0.521539f,    -0.0123823f,  -0.0642751f,
3592   0.08548f,     -0.0679207f,  0.526558f,    0.0651114f,   -0.342643f,
3593   -0.349934f,   0.307437f,    0.368763f,    -0.194851f,   -0.134117f,
3594   0.102448f,    -0.0520666f,  0.0415824f,   -0.175085f,   0.272685f,
3595   0.0675856f,   0.120627f,    0.391408f,    -0.135249f,   -0.357024f,
3596   0.019666f,    -0.0622677f,  0.407427f,    0.22655f,     -0.129432f,
3597   -0.165327f,   0.004893f,    0.5479f,      0.0613981f,   -0.479682f,
3598   -0.144228f,   -0.130106f,   0.206458f,    -0.342086f,   0.12691f,
3599   -0.113554f,   0.231164f,    -0.051419f,   0.0401286f,   -0.560429f,
3600   -0.070609f,   0.420232f,    0.442465f,    -0.237501f,   -0.000293732f,
3601   -1.017f,      -0.210222f,   0.0157063f,   0.0488178f,   0.0734721f,
3602   -0.52626f,    -0.276441f,   -0.521579f,   0.443532f,    -0.0819051f,
3603   -0.0732633f,  -0.17999f,    0.258525f,    -0.0374872f,  0.150115f,
3604   0.0510939f,   0.168116f,    0.473372f,    0.824489f,    0.302195f,
3605   -0.348613f,   0.238569f,    0.176444f,    -0.633945f,   -0.0567195f,
3606   -0.0305827f,  -0.0551851f,  0.85822f,     -0.0628099f,  0.0364294f,
3607   -0.234823f,   0.179067f,    0.143208f,    -0.0511014f,  -0.404191f,
3608   0.428035f,    0.0235506f,   0.371991f,    -0.312909f,   0.550933f,
3609   -0.389265f,   -0.271813f,   -0.293461f,   -0.583752f,   0.179991f,
3610   0.191698f,    0.659094f,    1.07941f,     -0.509555f,   -0.100638f,
3611   0.079988f,    -0.0519107f,  -0.112723f,   -0.0663326f,  0.0353569f,
3612   -0.795055f,   -0.465999f,   0.283579f,    0.340913f,    0.152738f,
3613   0.294664f,    0.527839f,    0.187735f,    0.359461f,    0.164629f,
3614   0.107512f,    0.390402f,    0.236702f,    0.114674f,    -0.525655f,
3615   -0.555476f,   -0.6589f,     -0.266601f,   -0.0946547f,  0.6306f,
3616   0.0248513f,   0.038497f,    0.432706f,    -0.0715465f,  0.0410172f,
3617   -0.115313f,   -0.428684f,   0.136283f,    0.0913185f,   0.11277f,
3618   0.0968689f,   -0.00437052f, 0.0888981f,   0.10304f,     0.02442f,
3619   -0.211315f,   0.00981596f,  -0.0974827f,  0.208611f,    0.140644f,
3620   0.0315567f,   0.350332f,    -0.291049f,   -0.0715449f,  -0.352992f,
3621   -0.858004f,   0.828658f,    0.439092f,    0.0151291f,   0.0503828f,
3622   0.0656112f,   -0.710749f,   -0.0951757f,  0.193908f,    0.00908018f,
3623   0.141486f,    -0.0657711f,  0.099791f,    0.153729f,    -0.419576f,
3624   -0.892636f,   -0.0449268f,  -0.170786f,   -0.156564f,   0.384511f,
3625   0.296565f,    0.0569815f,   -0.103938f,   1.27479f,     -0.0406475f,
3626   0.154083f,    -0.186442f,   0.0282588f,   0.0312102f,   -0.188994f,
3627   0.284243f,    -0.564693f,   0.425525f,    -0.00924596f, 0.810003f,
3628   0.233812f,    -0.0180273f,  0.121082f,    -0.209096f,   0.151437f,
3629   0.286921f,    -0.348095f,   0.174813f,    -0.413798f,   0.108994f,
3630   -0.34266f,    -0.0337981f,  -0.459f,      -0.409812f,   -0.0890104f,
3631   0.0834802f,   -0.00259191f, -0.105914f,   -0.164207f,   0.0697689f,
3632   -0.312098f,   -0.00650536f, -0.486758f,   -0.248486f,   0.24314f,
3633   -0.0857144f,  0.0884781f,   -0.65615f,    -0.121744f,   0.0709335f,
3634   -0.0237193f,  0.10764f,     -0.0409452f,  -0.0824305f,  0.42329f,
3635   0.138258f,    0.502607f,    0.228545f,    0.0687789f,   0.0361586f,
3636   0.39074f,     0.0722654f,   -0.0133148f,  0.283278f,    0.0743384f,
3637   0.310292f,    -0.297675f,   -0.359935f,   0.521021f,    -0.10082f,
3638   -0.272333f,   0.0120283f,   0.138118f,    -0.123711f,   -0.0711386f,
3639   0.0170747f,   0.831039f,    0.0509626f,   0.790608f,    -0.0863406f,
3640   -0.31962f,    0.0631013f,   0.0873453f,   -0.472331f,   -0.0826027f,
3641   -0.241722f,   0.148835f,    -0.131611f,   0.000195347f, -0.0615804f,
3642   -0.838663f,   -0.586979f,   0.247713f,    0.362254f,    0.492727f,
3643   -0.132163f,   0.0516545f,   0.477838f,    -0.0395182f,  0.0124993f,
3644   -0.771514f,   0.0386912f,   -0.118525f,   -0.346172f,   -0.265905f,
3645   -0.175257f,   -0.406287f,   0.393837f,    0.409096f,    -0.408501f,
3646   -0.0207146f,  0.0487809f,   0.0636982f,   0.0276368f,   0.0878249f,
3647   0.0425889f,   0.0868633f,   0.17423f,     -0.128217f,   -0.477068f,
3648   -0.321294f,   0.0393771f,   0.00812823f,  -0.350529f,   -0.129012f,
3649   0.439953f,    0.396662f,    0.410475f,    -0.123129f,   -0.565966f,
3650   0.0298635f,   -0.614611f,   -0.477514f,   0.453651f,    0.0617068f,
3651   0.0530563f,   0.0479074f,   0.213551f,    0.039034f,    0.0449095f,
3652   -1.06868f,    -1.2654f,     -0.175482f,   0.595068f,    -0.230095f,
3653   0.719838f,    -0.272148f,   0.696564f,    0.0485396f,   0.468584f,
3654   0.0695439f,   -0.0842122f,  -0.228978f,   0.161397f,    -0.000441421f,
3655   -0.0297514f,  -0.250599f,   0.196656f,    0.608423f,    -0.0112096f,
3656   0.0236881f,   -0.00167311f, 0.0040709f,   0.015495f,    0.00757698f,
3657   -0.165886f,   0.359767f,    -0.0214696f,  0.377208f,    0.0303547f,
3658   0.0657094f,   0.140775f,    0.21867f,     -0.203922f,   0.263878f,
3659   -0.0529099f,  0.202438f,    -0.243226f,   0.156659f,    -0.627056f,
3660   -0.845036f,   -0.500873f,   0.172588f,    0.402972f,    -0.147734f,
3661   0.151792f,    -0.075579f,   0.443519f,    0.0311335f,   -0.0328222f,
3662   -0.0299781f,  0.435956f,    -0.0987376f,  0.288402f,    0.135902f,
3663   -0.173584f,   -0.186255f,   0.224524f,    -0.249645f,   0.123702f,
3664   -0.0846244f,  0.491317f,    0.544846f,    0.338677f,    -0.258885f,
3665   -0.617434f,   -0.629003f,   -0.347233f,   0.181262f,    -0.0606015f,
3666   -0.537766f,   0.215089f,    -0.334527f,   0.0488534f,   0.0577997f,
3667   -1.12431f,    -0.932292f,   -0.11559f,    0.573715f,    0.151128f,
3668   0.693818f,    -0.16956f,    0.802591f,    -0.231531f,   1.04318f,
3669   -0.476417f,   0.293452f,    -0.610136f,   0.27506f,     -0.384012f,
3670   0.305366f,    -0.0540464f,  -0.337583f,   -0.174285f,   0.157248f,
3671   0.0477345f,   -0.0229535f,  0.0475766f,   -0.00603319f, 0.00856119f,
3672   -0.702893f,   -0.0579673f,  0.183024f,    -0.166222f,   0.109763f,
3673   -0.148019f,   -0.258873f,   -0.0820157f,  -0.186716f,   -0.449265f,
3674   -0.0534138f,  0.15732f,     0.46357f,     0.00502591f,  -0.0282085f,
3675   0.152277f,    -0.855199f,   -0.357115f,   0.0366159f,   0.0131101f,
3676   -0.0407758f,  0.0462835f,   0.146309f,    -0.00276278f, -0.0591814f,
3677   -0.109437f,   0.506764f,    -0.044421f,   0.465907f,    0.114444f,
3678   -0.241053f,   -0.362649f,   -0.432615f,   0.199989f,    -0.00635866f,
3679   -0.521886f,   0.0958924f,   -0.485725f,   0.0430527f,   0.069746f,
3680   0.681091f,    -0.288144f,   0.505671f,    0.0489065f,   -0.0373836f,
3681   0.266079f,    0.145173f,    -0.011481f,   -0.225074f,   -0.754501f,
3682   -0.122939f,   -0.294213f,   0.334738f,    0.281561f,    0.558977f,
3683   -0.21551f,    -0.346507f,   -0.0625635f,  0.0782034f,   -0.236999f,
3684   -0.803783f,   -0.601117f,   0.091192f,    0.636122f,    -0.250626f,
3685   0.0354961f,   0.103915f,    0.508571f,    0.329911f,    -0.0425999f,
3686   -0.0867587f,  -0.0385824f,  1.13914f,     -0.0261992f,  0.00484478f,
3687   0.124603f,    -0.012173f,   -0.377358f,   -0.243563f,   0.236094f,
3688   0.145663f,    -0.132752f,   0.347497f,    -0.529315f,   0.271632f,
3689   -0.372805f,   0.0261836f,   0.126169f,    0.0941008f,   0.283773f,
3690   0.765701f,    -0.226477f,   -0.181549f,   -0.306896f,   0.110165f,
3691   -0.0784234f,  -0.0827892f,  -0.0374252f,  -0.0950872f,  -0.451015f,
3692   -0.995793f,   -0.452663f,   0.293338f,    -0.380865f,   0.032683f,
3693   0.0178248f,   0.0699194f,   -0.0811722f,  -0.0866096f,  0.139289f,
3694   0.296604f,    0.192293f,    -0.0589607f,  -0.179878f,   0.00360266f,
3695   -0.0905794f,  0.136744f,    -0.191555f,   1.31877f,     -0.0592033f,
3696   -0.158766f,   0.0214746f,   -0.190113f,   -0.116671f,   0.0449292f,
3697   -0.109533f,   -0.709307f,   0.386424f,    0.40201f,     0.262211f,
3698   -0.155244f,   0.233988f,    -0.0166317f,  0.462665f,    0.0484462f,
3699   0.210902f,    -0.352798f,   0.38698f,     -0.228261f,   -0.084309f,
3700   -0.220751f,   -0.170879f,   -0.352617f,   -1.24277f,    0.266004f,
3701   -0.0125749f,  -0.0380073f,  0.101838f,    -0.0483024f,  -0.0629178f,
3702   -0.0695577f,  -0.103439f,   0.242131f,    -0.0796858f,  0.349718f,
3703   -0.332045f,   0.0138352f,   -0.380235f,   -0.28717f,    -0.176276f,
3704   0.865903f,    0.36593f,     0.243925f,    -0.422289f,   -0.117327f,
3705   0.21876f,     0.245393f,    -0.426134f,   -0.186077f,   0.0352515f,
3706   -0.123742f,   0.249376f,    1.3281f,      0.0707771f,   0.071415f,
3707   -0.286827f,   -0.131691f,   -0.270881f,   -0.434378f,   0.376064f,
3708   0.35966f,     0.513374f,    0.439378f,    -0.222716f,   -0.5874f,
3709   0.487997f,    -0.293271f,   -0.184245f,   -0.037256f,   0.17723f,
3710   -0.438651f,   0.428184f,    0.112983f,    -0.449287f,   -0.0451963f,
3711   0.0854929f,   0.0735442f,   -0.0148642f,  -0.0586782f,  -0.176455f,
3712   -0.438979f,   -0.127109f,   0.211478f,    0.388035f,    -0.0372021f,
3713   0.220575f,    0.382144f,    0.302121f,    0.0857121f,   0.193445f,
3714   -0.488858f,   -0.195288f,   -0.316184f,   -0.314026f,   -0.111956f,
3715   0.0744768f,   0.292709f,    0.30187f,     -0.285506f,   -0.105006f,
3716   0.0851402f,   -0.082318f,   0.277518f,    0.725294f,    -0.756304f,
3717   0.0155309f,   -0.378542f,   0.293377f,    -0.347252f,   -0.338458f,
3718   0.221449f,    -0.176443f,   -0.131972f,   0.0129163f,   -0.290649f,
3719   0.198596f,    -0.0721333f,  0.620591f,    0.568736f,    0.174001f,
3720   -0.205186f,   -0.265606f,   -0.249155f,   0.299163f,    1.11842f,
3721   0.17423f,     0.196417f,    -0.014484f,   0.0735422f,   0.26329f,
3722   0.12284f,     -0.750305f,   -0.351337f,   0.121994f,    -0.00542878f,
3723   -0.295707f,   -0.094124f,   0.300993f,    0.412408f,    -0.170761f,
3724   -0.0676329f,  -0.106638f,   -0.419785f,   -0.43878f,    0.22421f,
3725   0.0339903f,   0.619851f,    0.0615381f,   0.514631f,    1.35424f,
3726   -0.0679228f,  -0.203457f,   0.131948f,    -0.0041251f,  -0.209054f
3727 };
3728 
3729 static const float av1_simple_motion_search_prune_part_logits_bias_16[] = {
3730   0.304025f,  0.131887f, 0.259279f,  -0.561564f, -0.161729f,
3731   -0.208036f, 0.102206f, -0.162937f, -1.42311f,  -0.708305f
3732 };
3733 
3734 static const NN_CONFIG av1_simple_motion_search_prune_part_nn_config_16 = {
3735   NUM_FEATURES_16,
3736   NUM_LOGITS_16,
3737   NUM_HIDDEN_LAYERS_16,
3738   {
3739       NUM_LAYER_0_UNITS_16,
3740   },
3741   {
3742       av1_simple_motion_search_prune_part_layer_0_kernel_16,
3743       av1_simple_motion_search_prune_part_logits_kernel_16,
3744   },
3745   {
3746       av1_simple_motion_search_prune_part_layer_0_bias_16,
3747       av1_simple_motion_search_prune_part_logits_bias_16,
3748   },
3749 };
3750 
3751 #undef NUM_HIDDEN_LAYERS_16
3752 #undef NUM_FEATURES_16
3753 #undef NUM_LAYER_0_UNITS_16
3754 #undef NUM_LOGITS_16
3755 
3756 #define NUM_HIDDEN_LAYERS_8 1
3757 #define NUM_FEATURES_8 25
3758 #define NUM_LAYER_0_UNITS_8 32
3759 #define NUM_LOGITS_8 4
3760 
3761 static const float av1_simple_motion_search_prune_part_logits_kernel_8[] = {
3762   -0.266303f,  -0.387676f,  0.204501f,   -0.120842f,  -0.0752326f, 0.0337739f,
3763   0.0243477f,  -0.356748f,  0.0143051f,  -0.16403f,   -0.139013f,  0.175003f,
3764   -0.206754f,  0.349059f,   0.181763f,   0.212768f,   -0.313783f,  0.182829f,
3765   0.00205376f, -0.939525f,  -0.0992424f, 0.306254f,   0.083329f,   -0.133137f,
3766   -0.179022f,  -0.0237902f, 0.0601026f,  -0.216698f,  -0.551149f,  0.081711f,
3767   -0.442191f,  0.0680832f,  -0.0353678f, 0.237704f,   0.23155f,    -0.36097f,
3768   0.123389f,   -0.288927f,  0.178133f,   -0.152222f,  -0.235648f,  -0.0495293f,
3769   -0.316522f,  0.034207f,   0.0463139f,  -0.817825f,  0.417443f,   -0.110984f,
3770   -0.402371f,  0.0341694f,  -0.37383f,   0.414532f,   0.093993f,   0.0039505f,
3771   0.0803175f,  -0.511859f,  -0.0154802f, 0.0979595f,  0.0909049f,  -0.120938f,
3772   -0.577382f,  -0.155041f,  -0.404295f,  0.122223f,   -0.084703f,  0.00415336f,
3773   0.149135f,   0.113219f,   0.124236f,   -0.240905f,  0.163909f,   -0.154202f,
3774   -0.208917f,  0.00200158f, -0.71796f,   0.105984f,   -0.131996f,  -0.539603f,
3775   0.223768f,   -0.0710733f, -0.346679f,  -0.0745909f, 0.171032f,   0.215701f,
3776   0.218519f,   0.105981f,   -0.096209f,  -0.166453f,  -0.468894f,  -0.401578f,
3777   -0.239222f,  0.111382f,   0.38747f,    -0.164734f,  -0.175955f,  0.336621f,
3778   -0.0305501f, -0.0576765f, 0.0672671f,  -0.183692f,  0.412082f,   -0.262951f,
3779   -0.153429f,  -0.128589f,  -0.530472f,  0.0936412f,  -1.08296f,   -0.45147f,
3780   0.0714904f,  -3.96842f,   0.438125f,   -0.313945f,  0.231104f,   -0.00183851f,
3781   -0.0192768f, -0.637531f,  -0.109296f,  0.0531702f,  0.00262162f, -0.615951f,
3782   -0.546241f,  -0.635305f,  -0.0762367f, 0.0122019f,  0.423693f,   -0.129142f,
3783   -0.112242f,  0.295184f
3784 };
3785 
3786 static const float av1_simple_motion_search_prune_part_layer_0_bias_8[] = {
3787   -2.16023f,  -3.12831f, -0.213206f,  -2.97875f, -1.83791f, -2.84713f,
3788   -0.909636f, -2.05893f, 0.00525274f, -1.51672f, -3.95017f, 1.82847f,
3789   -0.853224f, -3.29503f, -0.537517f,  0.923106f, -3.18665f, -1.29905f,
3790   1.64506f,   -1.99848f, -2.24315f,   0.408613f, 0.503671f, -3.83393f,
3791   -2.88388f,  -3.52337f, 1.46818f,    -1.67169f, -3.83253f, 1.52644f,
3792   -0.490783f, -0.415782f
3793 };
3794 
3795 static const float av1_simple_motion_search_prune_part_layer_0_kernel_8[] = {
3796   -0.702198f,  -0.102148f,   0.0564545f,   -0.0555548f,  0.16184f,
3797   0.0950792f,  0.136974f,    -0.00824146f, 0.05746f,     0.0447542f,
3798   0.145978f,   0.0855769f,   -0.041449f,   0.301347f,    -0.0206691f,
3799   -0.0662514f, -0.0525079f,  -0.0998387f,  -0.0891438f,  0.110545f,
3800   -0.863098f,  -1.83798f,    0.238818f,    0.127797f,    0.116872f,
3801   -0.270655f,  -0.21057f,    0.197013f,    -0.123332f,   0.137104f,
3802   -0.174766f,  -0.00803025f, 0.0234369f,   -0.0894175f,  -0.0380927f,
3803   0.00827928f, -0.134148f,   0.110575f,    -0.250173f,   0.116273f,
3804   0.0197749f,  0.270391f,    0.108437f,    0.173197f,    -0.0650348f,
3805   0.0884626f,  0.262792f,    0.0649228f,   0.5573f,      -2.81315f,
3806   -0.479801f,  -1.15825f,    0.0807932f,   -0.19144f,    0.404016f,
3807   -0.211521f,  0.233269f,    -0.391414f,   0.160381f,    -0.277233f,
3808   0.426354f,   0.156839f,    0.494315f,    -0.214259f,   -0.0132062f,
3809   0.148628f,   -0.0899568f,  0.161845f,    0.467689f,    0.229474f,
3810   0.590634f,   -0.705793f,   -0.0486113f,  -0.439088f,   0.994566f,
3811   0.679065f,   0.777869f,    -0.225291f,   -0.0303006f,  -0.638782f,
3812   -0.0824632f, -0.128561f,   -0.327603f,   0.105624f,    0.567581f,
3813   -0.396135f,  -0.471028f,   0.181286f,    0.274604f,    0.180169f,
3814   0.0612144f,  -0.865004f,   0.0306804f,   0.142985f,    -0.0914358f,
3815   -0.243284f,  0.358359f,    -0.443847f,   -0.371978f,   0.606933f,
3816   -0.900408f,  -0.52076f,    0.472118f,    0.0610973f,   0.152526f,
3817   -0.550379f,  0.309331f,    -0.141573f,   0.203046f,    -0.231485f,
3818   0.505156f,   0.393224f,    0.435487f,    -0.218681f,   0.123707f,
3819   -0.270383f,  -0.033565f,   0.210373f,    -2.33967f,    0.367434f,
3820   0.0308118f,  -0.205771f,   0.546141f,    0.19837f,     0.035648f,
3821   -0.467007f,  -1.50995f,    -0.0314176f,  0.11762f,     -0.15307f,
3822   0.618257f,   -0.139502f,   0.303386f,    -0.00758681f, 0.228107f,
3823   -0.594499f,  -0.201984f,   -0.239666f,   0.114878f,    -0.922174f,
3824   -0.530137f,  -0.379366f,   -0.319582f,   0.0889624f,   -0.00544663f,
3825   0.316264f,   -0.204262f,   -0.0959358f,  0.23552f,     0.141369f,
3826   -0.207129f,  -1.04067f,    -0.0780501f,  0.226768f,    -0.246752f,
3827   0.0823105f,  0.114783f,    0.49315f,     0.0197732f,   0.705433f,
3828   0.158076f,   -0.250584f,   -0.157326f,   -0.0439547f,  -0.139047f,
3829   0.090531f,   -0.38833f,    0.743143f,    -1.47418f,    -0.155009f,
3830   0.511466f,   -0.726716f,   -0.181075f,   0.450133f,    -0.390204f,
3831   0.292725f,   0.00811462f,  -0.347738f,   0.613381f,    -0.237124f,
3832   0.750748f,   -0.383123f,   0.410309f,    -0.204166f,   0.667199f,
3833   -0.313197f,  0.436059f,    -0.607571f,   0.193681f,    0.409399f,
3834   0.631747f,   -0.0454149f,  0.198232f,    0.345591f,    -0.0137374f,
3835   -0.307014f,  -0.535515f,   0.764678f,    -0.225686f,   -0.451621f,
3836   -2.75564f,   -1.52877f,    0.0511933f,   0.905979f,    0.145029f,
3837   0.759615f,   0.130166f,    0.83827f,     0.0655081f,   1.07555f,
3838   -0.529777f,  0.682967f,    -0.412052f,   0.611947f,    -0.83676f,
3839   0.940695f,   -0.465681f,   0.51505f,     -0.883659f,   -0.105524f,
3840   -0.0344173f, -0.0683618f,  -0.00698688f, -0.139349f,   0.135741f,
3841   -0.294455f,  -0.377834f,   -0.602084f,   -1.00128f,    0.483291f,
3842   1.25327f,    0.178987f,    0.75068f,     -0.520731f,   -0.325517f,
3843   0.272032f,   0.144144f,    -0.279453f,   0.564907f,    0.144036f,
3844   0.297448f,   -0.504243f,   -0.250508f,   -1.26395f,    0.4816f,
3845   0.392771f,   -0.389961f,   -0.261585f,   -0.127124f,   -0.202945f,
3846   -0.709716f,  -0.174719f,   0.113613f,    0.477753f,    -0.226659f,
3847   0.0697828f,  -0.177994f,   0.300726f,    -0.185504f,   0.339424f,
3848   -0.316746f,  0.369693f,    -0.339723f,   -0.143886f,   -0.0326589f,
3849   -0.268761f,  -0.241094f,   0.284876f,    -0.0270867f,  -0.207397f,
3850   -1.42738f,   0.495612f,    -0.0277732f,  0.199675f,    1.48638f,
3851   -0.659257f,  -1.28199f,    0.498702f,    0.140695f,    0.571152f,
3852   0.416368f,   0.14153f,     0.126876f,    0.521114f,    -0.00150571f,
3853   0.375581f,   0.00537624f,  0.1286f,      -0.332227f,   0.417663f,
3854   -0.539023f,  0.217124f,    -0.787111f,   -0.0335266f,  1.56751f,
3855   0.0640563f,  -0.158791f,   0.118195f,    0.000970493f, -0.0403852f,
3856   -0.0572557f, -0.0201181f,  -0.10255f,    0.63237f,     0.156662f,
3857   0.418696f,   -0.274802f,   -0.663923f,   -0.375232f,   -0.40846f,
3858   0.462092f,   1.2176f,      -0.301532f,   -0.779704f,   -0.112876f,
3859   0.0806591f,  -0.0141923f,  0.00960801f,  -0.663557f,   0.0979948f,
3860   -0.0575999f, -0.012847f,   0.0403853f,   -0.133666f,   -0.00330217f,
3861   -0.931518f,  -0.774599f,   -0.21391f,    0.377601f,    -0.183365f,
3862   0.299094f,   0.0238552f,   0.206716f,    -0.18959f,    0.346013f,
3863   -0.150991f,  -0.192817f,   -0.293962f,   -0.0537604f,  -0.0648171f,
3864   -0.275941f,  -0.144854f,   -0.224092f,   2.43113f,     0.0422494f,
3865   -0.047236f,  -0.0262028f,  0.0282119f,   -0.175553f,   0.0888502f,
3866   0.580682f,   0.951055f,    -0.284441f,   -0.120133f,   -0.268058f,
3867   -0.312083f,  -0.411556f,   0.21431f,     -0.28033f,    0.324851f,
3868   -1.02787f,   -0.936816f,   -0.577628f,   0.544743f,    0.295807f,
3869   0.406157f,   0.447927f,    0.25369f,     -0.811421f,   -0.0424979f,
3870   -0.189867f,  0.00778673f,  -0.113587f,   -0.116175f,   -0.0542222f,
3871   -1.80089f,   -1.44175f,    -0.35332f,    0.191314f,    -0.236691f,
3872   -0.0261926f, -0.502363f,   0.252278f,    -0.485478f,   0.296495f,
3873   0.455612f,   -0.0489631f,  0.227255f,    0.170975f,    0.473487f,
3874   0.257812f,   0.178048f,    0.2506f,      2.04637f,     -0.173857f,
3875   0.0583379f,  0.00765589f,  -0.025772f,   -0.162666f,   -0.016214f,
3876   -0.607486f,  -0.0808025f,  0.0551611f,   -0.0772291f,  0.126421f,
3877   0.10869f,    -0.0877463f,  -0.111527f,   -0.0775766f,  0.503886f,
3878   -0.002757f,  -0.0421354f,  -0.247857f,   0.140827f,    0.383576f,
3879   0.228232f,   -0.157877f,   -0.0927911f,  0.344687f,    0.191181f,
3880   0.236533f,   0.00102869f,  -0.0184502f,  -1.4509f,     -1.15945f,
3881   -0.521978f,  -0.643225f,   0.133139f,    0.0660321f,   0.0851957f,
3882   0.0303648f,  0.0296239f,   0.0455713f,   0.175647f,    0.080532f,
3883   0.0445691f,  -0.257356f,   -0.125602f,   -0.138829f,   -0.167057f,
3884   -0.0992552f, -0.13944f,    0.507531f,    0.444997f,    0.221452f,
3885   -0.308384f,  -0.327554f,   0.13235f,     2.1487f,      -1.15453f,
3886   -0.280239f,  -0.363582f,   -0.00358745f, 0.012866f,    0.251088f,
3887   0.0676416f,  0.178492f,    -0.136631f,   0.197938f,    -0.078198f,
3888   0.812439f,   1.1173f,      0.712113f,    1.10124f,     -0.836503f,
3889   -1.22433f,   -1.07894f,    -1.29215f,    0.56057f,     2.23928f,
3890   -0.419029f,  0.282178f,    -0.0719266f,  -0.172192f,   0.28034f,
3891   -2.99124f,   -2.01481f,    0.0688982f,   0.697466f,    0.00635555f,
3892   0.566069f,   0.047534f,    0.507755f,    -0.00690707f, 0.712594f,
3893   -0.191467f,  0.355733f,    -0.480016f,   0.664669f,    -0.390619f,
3894   0.351199f,   -0.482342f,   0.325005f,    1.9089f,      0.155987f,
3895   0.17032f,    0.132729f,    0.0402649f,   0.146991f,    0.0314905f,
3896   -0.775316f,  -0.208892f,   -0.105993f,   0.0181653f,   -0.12735f,
3897   0.0897852f,  0.0470231f,   0.25807f,     0.127406f,    -0.0893252f,
3898   -0.279776f,  0.190844f,    0.110384f,    -0.148833f,   0.025293f,
3899   0.239838f,   0.00932245f,  0.35103f,     -0.128268f,   -0.0536754f,
3900   0.506899f,   -0.16793f,    0.0955582f,   -2.01108f,    0.721433f,
3901   -2.31413f,   -2.08646f,    0.033315f,    0.689828f,    -0.271213f,
3902   0.790425f,   -0.114234f,   0.755325f,    -0.211533f,   0.774544f,
3903   -0.263268f,  0.795762f,    -0.551455f,   0.953602f,    -0.168454f,
3904   0.529055f,   -0.768991f,   0.882371f,    0.29763f,     -0.155017f,
3905   0.00464101f, 0.121093f,    0.948271f,    0.113138f,    -0.110332f,
3906   -2.0492f,    -1.31322f,    -0.129212f,   0.464778f,    -0.181465f,
3907   0.618403f,   0.0627984f,   0.465228f,    0.165729f,    0.278277f,
3908   -0.563276f,  -0.358358f,   -0.590638f,   0.0104993f,   0.731206f,
3909   0.752569f,   0.631615f,    0.811822f,    0.129804f,    -0.0558327f,
3910   0.570081f,   -0.417922f,   -0.168275f,   0.0703671f,   0.269127f,
3911   0.240457f,   -0.197159f,   -0.00179261f, 0.220065f,    0.463511f,
3912   0.0714626f,  -0.716477f,   -0.441865f,   -0.717028f,   -0.149176f,
3913   0.452182f,   0.662699f,    -0.906534f,   -0.817133f,   0.237747f,
3914   0.26024f,    -7.7441e-05f, 0.0934616f,   0.824641f,    -0.0404494f,
3915   -0.088297f,  -0.157899f,   0.037408f,    0.132435f,    -0.316155f,
3916   -0.276785f,  0.0117868f,   0.185008f,    0.32369f,     -0.465855f,
3917   -0.302127f,  0.303289f,    0.338597f,    -0.665408f,   -0.507594f,
3918   0.526979f,   0.532091f,    0.234395f,    0.754063f,    0.116769f,
3919   0.0800309f,  -0.939344f,   -1.51269f,    1.4583f,      0.178444f,
3920   0.0106756f,  -0.213468f,   -0.00369439f, 0.071015f,    -0.192798f,
3921   -0.0933147f, -0.129901f,   -0.368279f,   -0.246564f,   0.126966f,
3922   0.478565f,   -0.476246f,   -0.762863f,   0.168883f,    0.536136f,
3923   -0.272969f,  0.2573f,      -0.161577f,   0.311428f,    -0.777994f,
3924   -1.29752f,   0.216046f,    0.329016f,    1.57265f,     0.168075f,
3925   -0.192518f,  0.0829308f,   -0.073533f,   -0.0202034f,  0.114716f,
3926   -0.34888f,   -0.519215f,   0.190809f,    0.0138507f,   0.133635f,
3927   0.14194f,    0.410618f,    -0.165106f,   0.214438f,    0.0438265f,
3928   -0.8481f,    -1.19182f,    -1.07878f,    -0.882217f,   0.45616f,
3929   0.977385f,   0.74929f,     0.918466f,    0.904704f,    0.041938f,
3930   0.0362776f,  0.0757255f,   1.14007f,     0.0516825f,   -0.160068f,
3931   0.219535f,   0.638634f,    -0.0284544f,  -0.222849f,   -0.0344915f,
3932   -0.0350256f, -0.0504452f,  -0.0458416f,  0.146099f,    0.0783083f,
3933   0.206579f,   0.241264f,    0.28401f,     0.0425312f,   -0.802049f,
3934   -0.746271f,  -0.578969f,   -0.078218f,   0.436176f,    -0.281465f,
3935   -2.5539f,    0.237868f,    -0.121796f,   0.0715619f,   0.106992f,
3936   -0.621862f,  -0.167142f,   0.153716f,    0.0570912f,   -0.06525f,
3937   -0.923773f,  0.130759f,    0.0517066f,   0.0729862f,   -0.873064f,
3938   0.0403328f,  -0.186499f,   -0.0831918f,  -0.223723f,   0.144697f,
3939   0.212845f,   0.416876f,    0.361598f,    0.138229f,    0.0728777f,
3940   -1.95419f,   -0.00382816f, -0.0440387f,  0.433627f,    0.44781f,
3941   -1.05229f,   -1.54506f,    0.564827f,    -0.263456f,   0.296105f,
3942   -0.158055f,  0.388274f,    -0.366639f,   0.212006f,    -0.245619f,
3943   0.593064f,   0.088727f,    0.410632f,    -0.263462f,   0.507075f,
3944   -0.0974155f, 0.275268f,    -0.1293f,     0.136679f,    1.98276f,
3945   0.411766f,   0.391987f,    0.34283f,     -0.114077f,   0.258462f,
3946   -0.302443f,  0.301138f,    -0.00726621f, 0.276441f,    -0.291582f,
3947   0.66498f,    -0.321451f,   -0.332805f,   0.0943272f,   0.572253f,
3948   -0.45818f,   -0.0219593f,  -0.151679f,   0.402033f,    -1.15502f,
3949   -0.882955f,  0.772904f,    0.88126f,     -0.149555f,   0.709525f,
3950   0.350116f,   -0.21531f,    0.797893f,    0.0230234f,   0.0203034f,
3951   0.2744f,     1.08273f,     0.039349f,    0.503909f,    -0.45892f,
3952   -0.579516f,  -0.344058f,   0.390628f,    -0.386941f,   -0.430317f,
3953   -0.0807066f, 0.435906f,    0.522996f,    0.724476f,    -0.74371f,
3954   -0.05376f,   -0.340898f,   -0.962646f,   -0.0278005f,  0.0981149f,
3955   -0.0811161f, 0.00237994f,  0.850042f,    0.0665473f,   0.134413f
3956 };
3957 
3958 static const float av1_simple_motion_search_prune_part_logits_bias_8[] = {
3959   1.63404f, -0.715866f, -1.0132f, -2.08745f
3960 };
3961 
3962 static const NN_CONFIG av1_simple_motion_search_prune_part_nn_config_8 = {
3963   NUM_FEATURES_8,
3964   NUM_LOGITS_8,
3965   NUM_HIDDEN_LAYERS_8,
3966   {
3967       NUM_LAYER_0_UNITS_8,
3968   },
3969   {
3970       av1_simple_motion_search_prune_part_layer_0_kernel_8,
3971       av1_simple_motion_search_prune_part_logits_kernel_8,
3972   },
3973   {
3974       av1_simple_motion_search_prune_part_layer_0_bias_8,
3975       av1_simple_motion_search_prune_part_logits_bias_8,
3976   },
3977 };
3978 
3979 #undef NUM_HIDDEN_LAYERS_8
3980 #undef NUM_FEATURES_8
3981 #undef NUM_LAYER_0_UNITS_8
3982 #undef NUM_LOGITS_8
3983 
3984 #define FEATURE_SIZE 19
3985 static const float av1_2pass_split_partition_weights_128[FEATURE_SIZE + 1] = {
3986   2.683936f, -0.193620f, -4.106470f, -0.141320f, -0.282289f,
3987   0.125296f, -1.134961f, 0.862757f,  -0.418799f, -0.637666f,
3988   0.016232f, 0.345013f,  0.018823f,  -0.393394f, -1.130700f,
3989   0.695357f, 0.112569f,  -0.341975f, -0.513882f, 5.7488966f,
3990 };
3991 
3992 static const float av1_2pass_split_partition_weights_64[FEATURE_SIZE + 1] = {
3993   2.990993f,  0.423273f,  -0.926544f, 0.454646f,  -0.292698f,
3994   -1.311632f, -0.284432f, 0.717141f,  -0.419257f, -0.574760f,
3995   -0.674444f, 0.669047f,  -0.374255f, 0.380624f,  -0.804036f,
3996   0.264021f,  0.004163f,  1.896802f,  0.924287f,  0.13490619f,
3997 };
3998 
3999 static const float av1_2pass_split_partition_weights_32[FEATURE_SIZE + 1] = {
4000   2.795181f,  -0.136943f, -0.924842f, 0.405330f,  -0.463505f,
4001   -0.584076f, -0.831472f, 0.382985f,  -0.597544f, -0.138915f,
4002   -1.354350f, 0.466035f,  -0.553961f, 0.213202f,  -1.166429f,
4003   0.010776f,  -0.096236f, 2.335084f,  1.699857f,  -0.58178353f,
4004 };
4005 
4006 static const float av1_2pass_split_partition_weights_16[FEATURE_SIZE + 1] = {
4007   1.987888f,  -0.431100f, -1.687703f, 0.262602f,  -0.425298f,
4008   -0.463870f, -1.493457f, 0.470917f,  -0.528457f, -0.087700f,
4009   -1.815092f, 0.152883f,  -0.337908f, 0.093679f,  -1.548267f,
4010   -0.042387f, -0.000861f, 2.556746f,  1.619192f,  0.03643292f,
4011 };
4012 
4013 static const float av1_2pass_split_partition_weights_8[FEATURE_SIZE + 1] = {
4014   2.188344f,  -0.817528f, -2.119219f, 0.000000f,  -0.348167f,
4015   -0.658074f, -1.960362f, 0.000000f,  -0.403080f, 0.282699f,
4016   -2.061088f, 0.000000f,  -0.431919f, -0.127960f, -1.099550f,
4017   0.000000f,  0.121622f,  2.017455f,  2.058228f,  -0.15475988f,
4018 };
4019 
4020 static const float av1_2pass_none_partition_weights_128[FEATURE_SIZE + 1] = {
4021   -1.006689f, 0.777908f,  4.461072f,  -0.395782f, -0.014610f,
4022   -0.853863f, 0.729997f,  -0.420477f, 0.282429f,  -1.194595f,
4023   3.181220f,  -0.511416f, 0.117084f,  -1.149348f, 1.507990f,
4024   -0.477212f, 0.202963f,  -1.469581f, 0.624461f,  -0.89081228f,
4025 };
4026 
4027 static const float av1_2pass_none_partition_weights_64[FEATURE_SIZE + 1] = {
4028   -1.241117f, 0.844878f,  5.638803f,  -0.489780f, -0.108796f,
4029   -4.576821f, 1.540624f,  -0.477519f, 0.227791f,  -1.443968f,
4030   1.586911f,  -0.505125f, 0.140764f,  -0.464194f, 1.466658f,
4031   -0.641166f, 0.195412f,  1.427905f,  2.080007f,  -1.98272777f,
4032 };
4033 
4034 static const float av1_2pass_none_partition_weights_32[FEATURE_SIZE + 1] = {
4035   -2.130825f, 0.476023f,  5.907343f,  -0.516002f, -0.097471f,
4036   -2.662754f, 0.614858f,  -0.576728f, 0.085261f,  -0.031901f,
4037   0.727842f,  -0.600034f, 0.079326f,  0.324328f,  0.504502f,
4038   -0.547105f, -0.037670f, 0.304995f,  0.369018f,  -2.66299987f,
4039 };
4040 
4041 static const float av1_2pass_none_partition_weights_16[FEATURE_SIZE + 1] = {
4042   -1.626410f, 0.872047f,  5.414965f,  -0.554781f, -0.084514f,
4043   -3.020550f, 0.467632f,  -0.382280f, 0.199568f,  0.426220f,
4044   0.829426f,  -0.467100f, 0.153098f,  0.662994f,  0.327545f,
4045   -0.560106f, -0.141610f, 0.403372f,  0.523991f,  -3.02891231f,
4046 };
4047 
4048 static const float av1_2pass_none_partition_weights_8[FEATURE_SIZE + 1] = {
4049   -1.463349f, 0.375376f,  4.751430f, 0.000000f, -0.184451f,
4050   -1.655447f, 0.443214f,  0.000000f, 0.127961f, 0.152435f,
4051   0.083288f,  0.000000f,  0.143105f, 0.438012f, 0.073238f,
4052   0.000000f,  -0.278137f, 0.186134f, 0.073737f, -1.6494962f,
4053 };
4054 #undef FEATURE_SIZE
4055 
4056 // nn model for predicting max square partition level of a superblock
4057 #define NUM_HIDDEN_LAYERS 1
4058 #define NUM_FEATURES 13
4059 #define NUM_LAYER_0_UNITS 48
4060 #define NUM_LOGITS 4
4061 
4062 static const float av1_max_part_pred_logits_kernel[] = {
4063   -0.304561f,   0.0885596f,   -0.988539f,   1.08147f,    0.215213f,
4064   0.202965f,    -0.828457f,   -0.233945f,   -0.0866977f, -0.115521f,
4065   0.02079f,     0.196491f,    -0.0285075f,  0.05067f,    -0.00872862f,
4066   0.00281844f,  -0.238954f,   0.0253801f,   0.0257775f,  0.339269f,
4067   0.176174f,    -0.152545f,   -0.0588704f,  -1.62275f,   -0.189329f,
4068   0.0808033f,   0.233844f,    -4.53798f,    0.674968f,   -0.0361688f,
4069   -0.0754075f,  1.16129f,     -0.0188879f,  0.113255f,   -3.04378f,
4070   0.814728f,    -0.568517f,   -0.00179383f, -3.61223f,   -1.67535f,
4071   -2.20417f,    -0.197196f,   0.0507745f,   -0.0909394f, -0.0507879f,
4072   -1.27999f,    -0.055623f,   0.0318497f,   0.192867f,   0.138726f,
4073   0.0443392f,   -0.595075f,   -0.166774f,   0.0882958f,  -0.348161f,
4074   0.0214428f,   -0.0599275f,  -0.0995385f,  -0.82358f,   0.141205f,
4075   -0.053232f,   0.00508296f,  -1.90872f,    1.15004f,    -0.194219f,
4076   0.0229019f,   -0.00354318f, 0.22016f,     0.154101f,   -0.159231f,
4077   -0.0446647f,  -0.197503f,   0.0408453f,   0.197659f,   0.797858f,
4078   -0.189722f,   0.343653f,    0.124666f,    -1.03083f,   0.603059f,
4079   0.101565f,    0.0932993f,   0.462484f,    0.295984f,   1.11198f,
4080   0.143709f,    -0.846232f,   -0.464392f,   -1.06058f,   -0.124889f,
4081   0.0727475f,   1.18446f,     -0.100302f,   0.0641918f,  -0.101622f,
4082   0.10219f,     0.130189f,    0.0915623f,   -0.166904f,  -1.10606f,
4083   -0.16726f,    -0.146152f,   0.145443f,    -0.177091f,  -0.0215214f,
4084   0.0158506f,   -0.553294f,   0.0784749f,   -0.0416628f, -0.027785f,
4085   0.280027f,    0.484898f,    -0.164225f,   0.0238317f,  -0.0345254f,
4086   0.0410244f,   0.131529f,    0.0239622f,   -0.0749436f, -0.0224914f,
4087   0.128926f,    0.224539f,    0.413297f,    0.0638572f,  0.103308f,
4088   0.0913242f,   -0.119274f,   0.0163103f,   0.113828f,   0.119809f,
4089   0.297057f,    -0.124889f,   -0.533108f,   -0.181408f,  -0.129896f,
4090   0.0221064f,   -0.0773281f,  -0.0386467f,  0.0342961f,  0.126575f,
4091   -0.24114f,    0.0735576f,   0.0524791f,   0.246896f,   -0.130674f,
4092   -0.03979f,    0.173639f,    1.95193f,     -0.113029f,  -0.0305852f,
4093   -0.00671737f, 0.157159f,    -0.00102858f, -0.543688f,  0.566772f,
4094   0.124124f,    -0.0294064f,  -0.0699021f,  -0.0704103f, -0.766097f,
4095   -0.0625802f,  -0.0906173f,  -0.0520414f,  -0.0272724f, 0.283064f,
4096   0.236213f,    -0.127319f,   0.019392f,    0.170042f,   -0.0214542f,
4097   0.0740938f,   0.356578f,    -0.236257f,   0.269021f,   0.114759f,
4098   -0.641166f,   0.136308f,    -0.0386959f,  -0.112024f,  -0.361209f,
4099   0.686095f,    0.183906f,    0.288656f,    0.182007f,   0.337458f,
4100   0.058974f,    -0.305512f,   -0.841708f,   -0.243779f,  -0.0614058f,
4101   0.208747f,    0.448697f
4102 };
4103 
4104 static const float av1_max_part_pred_layer_0_bias[] = {
4105   -0.776544f, -2.0022f,    -0.330294f, 2.47665f,  1.90206f,   -1.61571f,
4106   0.536246f,  1.00455f,    5.24561f,   1.55111f,  -0.816399f, -4.88703f,
4107   -1.06417f,  -1.15359f,   -0.145289f, 1.91831f,  0.630915f,  -1.94256f,
4108   -3.35239f,  -1.05007f,   -1.05186f,  1.36824f,  -5.2878f,   1.10482f,
4109   -5.00077f,  -0.0445198f, 3.41427f,   2.3439f,   -0.413306f, -1.88152f,
4110   -2.28638f,  8.24783f,    -1.91961f,  -1.49324f, 1.96599f,   -6.32309f,
4111   -0.332426f, -0.425506f,  4.06511f,   5.84386f,  4.15747f,   1.22402f,
4112   2.8512f,    2.53027f,    0.0170272f, -1.43966f, -0.997785f, 5.43064f
4113 };
4114 
4115 static const float av1_max_part_pred_logits_bias[] = { -4.25432f, 0.144758f,
4116                                                        1.96217f, 0.728905f };
4117 
4118 static const float av1_max_part_pred_layer_0_kernel[] = {
4119   0.992471f,    0.533006f,    0.143743f,     -2.51788f,    -0.468337f,
4120   -0.201376f,   -0.151834f,   0.479883f,     1.16061f,     -0.278878f,
4121   -0.814954f,   -0.152405f,   -0.0521608f,   0.797104f,    -2.08912f,
4122   0.385839f,    -2.22889f,    -0.106858f,    -0.239766f,   -0.951128f,
4123   -0.698753f,   0.0831051f,   1.1702f,       0.342834f,    -0.0352795f,
4124   -0.0847639f,  -0.802086f,   0.258982f,     1.14174f,     0.645885f,
4125   -1.19226f,    -0.592888f,   -0.343659f,    1.1912f,      1.45411f,
4126   -1.22927f,    0.152858f,    0.00373585f,   -1.60637f,    0.592611f,
4127   0.0857475f,   -0.346147f,   -0.150784f,    -0.0817408f,  -0.189918f,
4128   -0.804952f,   -1.33036f,    -1.03307f,     0.0248769f,   0.16607f,
4129   -2.896f,      -2.1293f,     0.12293f,      -0.173179f,   -0.212128f,
4130   -6.76221f,    0.033188f,    0.0231787f,    0.905957f,    0.0551327f,
4131   -0.356276f,   0.0181795f,   0.0977523f,    -0.0352873f,  -0.0396386f,
4132   2.3241f,      0.0632874f,   -0.11804f,     -6.32521f,    0.0224659f,
4133   -0.00188896f, 0.267992f,    0.272337f,     0.00936963f,  0.659969f,
4134   -2.25707f,    -0.0278229f,  -0.0185089f,   -1.14466f,    0.104827f,
4135   0.0435885f,   0.558586f,    -0.00697004f,  0.0312611f,   0.540574f,
4136   -0.568625f,   0.218608f,    0.378911f,     -0.0289192f,  -0.0734742f,
4137   -1.08782f,    -2.42069f,    -0.0127239f,   0.0493651f,   -1.15837f,
4138   0.261831f,    0.401824f,    -1.04545f,     0.284173f,    0.784972f,
4139   -0.511243f,   -0.982599f,   -0.106134f,    -0.325964f,   -1.44107f,
4140   -1.42434f,    -1.02402f,    -1.52034f,     0.0737116f,   0.0462242f,
4141   0.628722f,    -1.0405f,     -0.113718f,    2.20573f,     -4.33951f,
4142   -0.0192695f,  -0.0229314f,  -1.89156f,     0.645942f,    0.375708f,
4143   -1.97447f,    -0.267014f,   0.0989443f,    -0.450534f,   -1.01737f,
4144   -0.642416f,   -0.0897288f,  -2.08724f,     -0.190965f,   -0.279135f,
4145   -0.830178f,   0.808754f,    -0.139091f,    1.11004f,     -0.454439f,
4146   -0.479238f,   -1.44001f,    0.0888059f,    0.885689f,    -0.642505f,
4147   -0.00773651f, -0.0265721f,  -0.906346f,    1.68504f,     0.084257f,
4148   -0.951101f,   -8.06495f,    0.19231f,      0.16389f,     -0.193678f,
4149   0.729837f,    -1.98392f,    -5.98513f,     3.32638f,     -0.0658378f,
4150   -0.0910426f,  -0.666567f,   -0.315339f,    0.123124f,    -2.66375f,
4151   -0.714852f,   -0.136176f,   -0.460166f,    -0.567551f,   -1.06193f,
4152   -1.21389f,    -0.83865f,    0.00280695f,   -0.199519f,   -0.534704f,
4153   0.419311f,    -0.149008f,   -3.68707f,     0.00285113f,  -0.0718198f,
4154   -1.41026f,    -1.34155f,    -0.538687f,    -0.623666f,   -2.56462f,
4155   -0.0183333f,  -0.323532f,   -1.27141f,     -0.0212039f,  0.198633f,
4156   0.459554f,    -4.65103f,    -1.01293f,     -1.39512f,    -0.289026f,
4157   0.208724f,    -0.665226f,   1.13369f,      -1.96734f,    -1.45442f,
4158   -3.46172f,    0.810681f,    -0.603973f,    0.842764f,    -3.90371f,
4159   -0.394561f,   -3.61363f,    -2.88085f,     0.031645f,    -0.23125f,
4160   -2.63898f,    -1.35314f,    -0.46726f,     1.33145f,     1.20269f,
4161   1.38682f,     -0.331637f,   0.069021f,     0.149523f,    -1.24957f,
4162   -0.878857f,   -0.200368f,   0.465744f,     1.01365f,     -0.0122221f,
4163   -0.550586f,   -1.12581f,    -0.422132f,    -0.0744868f,  -2.4804f,
4164   -1.07072f,    -0.479006f,   0.101817f,     -0.118947f,   0.341576f,
4165   -1.0538f,     -0.812346f,   -1.13727f,     -0.00939806f, 10.1571f,
4166   -0.0441302f,  0.00280407f,  -21.5044f,     0.0181152f,   -0.0143246f,
4167   3.23462f,     -1.38624f,    -1.80416f,     4.89763f,     -2.67364f,
4168   2.31771e-05f, 0.000393989f, 0.352204f,     -0.193455f,   0.531455f,
4169   0.488757f,    -0.442555f,   -0.518528f,    0.431482f,    -2.67727f,
4170   -2.00626f,    -0.39729f,    -0.221494f,    -0.0188888f,  -0.0377649f,
4171   -1.80169f,    0.0810332f,   -0.0408335f,   -1.28675f,    -0.0353824f,
4172   -0.666723f,   -1.07281f,    0.252912f,     -1.24547f,    -1.7831f,
4173   -1.14354f,    -0.137662f,   0.00230182f,   0.736862f,    0.175872f,
4174   -0.187556f,   0.43963f,     -0.796524f,    0.056219f,    -0.387874f,
4175   0.0710224f,   -0.16548f,    -0.100993f,    0.931481f,    -3.20738f,
4176   -0.0197576f,  0.266148f,    -0.173909f,    -0.337795f,   -0.0682381f,
4177   0.176844f,    0.140286f,    1.12033f,      0.429064f,    -2.24192f,
4178   -1.54682f,    2.23646f,     -0.0371138f,   -0.0475339f,  -3.21766f,
4179   0.0412858f,   0.387811f,    6.6711f,       0.140649f,    0.0559547f,
4180   -0.802839f,   0.599977f,    0.64552f,      -2.08103f,    -0.503401f,
4181   -0.0407036f,  -0.0299199f,  0.0849445f,    -0.111657f,   -1.63462f,
4182   3.33762f,     0.0441394f,   0.0466889f,    -0.951806f,   0.0723954f,
4183   0.00348661f,  -1.36903f,    2.24625f,      -0.0348915f,  -0.0508893f,
4184   -0.240891f,   -0.120143f,   -0.17991f,     -2.09137f,    0.0150871f,
4185   0.0480333f,   1.72012f,     0.0309551f,    -0.0370507f,  -0.377075f,
4186   0.103916f,    -0.0169255f,  -0.0145395f,   -4.02144f,    0.83193f,
4187   -0.316502f,   6.3832f,      -1.70038f,     -1.97215f,    -1.94501f,
4188   1.45479f,     0.711725f,    -0.348496f,    -0.279056f,   -1.13396f,
4189   -1.51744f,    -0.853307f,   1.53131f,      -0.0032358f,  1.41808f,
4190   -1.32989f,    -0.245221f,   -0.161614f,    -0.500845f,   -0.449252f,
4191   0.0724151f,   -0.116333f,   -0.0946182f,   -2.0945f,     0.0564572f,
4192   0.393261f,    -1.06861f,    -0.111458f,    -0.839943f,   -0.0880348f,
4193   0.0365742f,   0.415339f,    -1.57494f,     -0.713697f,   1.02349f,
4194   -0.221371f,   -0.0446281f,  1.89223f,      -0.0811754f,  -0.402773f,
4195   -0.930987f,   0.0243194f,   0.0678332f,    -0.0233014f,  0.165372f,
4196   -0.44083f,    -1.2404f,     0.35675f,      -0.040916f,   -0.0512548f,
4197   -2.9071f,     0.861174f,    -0.778133f,    2.14436f,     -0.688427f,
4198   -0.480371f,   -1.69032f,    0.706687f,     -0.281982f,   -2.30451f,
4199   1.61541f,     -0.0213638f,  -0.740509f,    -0.266677f,   0.0268434f,
4200   -0.0116908f,  -3.17595f,    0.0114825f,    0.0196997f,   -0.144005f,
4201   0.0550181f,   -0.851459f,   -0.000285073f, -0.538441f,   -0.0254868f,
4202   -0.0104454f,  -0.0661998f,  -0.196469f,    -0.346372f,   -5.52892f,
4203   -0.643683f,   -0.622224f,   -0.31463f,     -0.555956f,   -0.520132f,
4204   -0.843166f,   -2.59479f,    -0.750195f,    0.00635995f,  -0.338615f,
4205   -0.216676f,   -0.391544f,   -1.62185f,     -0.718471f,   -0.475406f,
4206   -0.782041f,   -0.608824f,   -1.09633f,     -1.27308f,    -0.560719f,
4207   -0.207539f,   -0.0196445f,  -1.05519f,     -0.575249f,   -1.0642f,
4208   1.01615f,     -0.873633f,   -0.417953f,    -0.428051f,   0.350259f,
4209   -2.53833f,    -2.72203f,    0.672846f,     -0.503094f,   -1.1374f,
4210   0.214291f,    0.013305f,    0.0112064f,    1.10532f,     0.030455f,
4211   0.0239614f,   0.628072f,    0.0539135f,    -0.472441f,   -0.688439f,
4212   -0.32044f,    -0.0234867f,  -0.0158436f,   -0.949314f,   -0.0453161f,
4213   -1.18306f,    0.626845f,    -0.426925f,    -0.688371f,   0.415062f,
4214   0.0640985f,   -0.638387f,   -2.01399f,     -0.209744f,   -0.762892f,
4215   -0.0753296f,  -0.879315f,   -0.520433f,    -0.111375f,   0.389742f,
4216   -0.398862f,   -0.643227f,   -0.246396f,    0.0317051f,   1.06973f,
4217   0.413617f,    0.180506f,    -0.0507897f,   -0.00650435f, 0.620892f,
4218   0.046312f,    0.475032f,    0.906993f,     -0.0388061f,  -0.256271f,
4219   -1.03323f,    0.0125266f,   -0.31116f,     -0.377611f,   -0.0386407f,
4220   -0.0232745f,  -0.353644f,   -2.27289f,     0.0571779f,   -0.00865006f,
4221   1.65101f,     0.0175711f,   0.0184585f,    0.558458f,    0.2213f,
4222   -0.285089f,   0.433445f,    -0.427177f,    -0.0103682f,  -0.0101273f,
4223   0.214085f,    -0.0459885f,  0.00761981f,   0.836381f,    0.0175293f,
4224   0.02508f,     -1.51778f,    0.0143956f,    -0.162589f,   0.595418f,
4225   0.21445f,     -0.0335848f,  -0.0136684f,   -0.16686f,    -0.14612f,
4226   0.0816238f,   0.499636f,    0.12458f,      -2.41673f,    -0.261721f,
4227   -0.676805f,   -1.88366f,    0.730462f,     0.69196f,     -0.0288489f,
4228   -2.38272f,    0.329876f,    0.014517f,     -0.115145f,   -3.48151f,
4229   -0.00209072f, -0.0732377f,  0.820443f,     -0.0118701f,  0.112145f,
4230   0.272315f,    0.137531f,    -0.0200997f,   -0.0397883f,  -2.19458f,
4231   0.183554f,    -0.639716f,   0.481605f,     -0.621639f,   -0.0980299f,
4232   -0.710534f,   -0.143105f,   -6.77626f,     -1.65139f,    -2.37718f,
4233   -0.533127f,   -1.12574f,    3.34182f,      -0.0758663f,  0.0334238f,
4234   -9.48647f,    0.0674974f,   0.0507665f,    0.523007f,    -0.0668f,
4235   0.5736f,      -0.589761f,   -1.1692f,      -0.0236497f,  -0.00828928f,
4236   -0.265823f,   1.15284f,     0.307927f,     -0.695308f,   0.13725f,
4237   -0.20394f,    -0.363965f,   -0.331159f,    -1.50927f,    -1.20051f,
4238   -0.0205825f,  -0.0381859f,  -0.0579876f,   -1.6913f,     -1.94626f,
4239   3.4214f,      3.3922f,      -2.13798f,     -0.679848f,   -0.890735f,
4240   0.235017f,    -0.253202f,   -1.0571f,      1.40354f,     0.00719052f,
4241   -1.54365f,    -0.7289f,     -1.05492f,     0.0238169f,   -0.00543592f,
4242   -0.0510353f,  -0.175386f,   -0.724207f,    -0.788936f,   0.039976f,
4243   1.36966f,     0.869475f,    -0.0302774f,   -0.0537556f
4244 };
4245 
4246 static const NN_CONFIG av1_max_part_pred_nn_config = {
4247   NUM_FEATURES,
4248   NUM_LOGITS,
4249   NUM_HIDDEN_LAYERS,
4250   {
4251       NUM_LAYER_0_UNITS,
4252   },
4253   {
4254       av1_max_part_pred_layer_0_kernel,
4255       av1_max_part_pred_logits_kernel,
4256   },
4257   {
4258       av1_max_part_pred_layer_0_bias,
4259       av1_max_part_pred_logits_bias,
4260   },
4261 };
4262 
4263 #undef NUM_HIDDEN_LAYERS
4264 #undef NUM_FEATURES
4265 #undef NUM_LAYER_0_UNITS
4266 #undef NUM_LOGITS
4267 
4268 // Early termination in second pass
4269 static const float av1_simple_motion_search_term_none_mean_128[28] = {
4270   12.661922f, 12.638062f, 10.896497f, 10.865719f, 10.978963f, 10.940105f,
4271   11.012235f, 10.972760f, 11.069924f, 11.018533f, 11.773865f, 11.747426f,
4272   11.891315f, 11.858107f, 11.793916f, 11.766356f, 11.874997f, 11.840164f,
4273   5.940535f,  0.770746f,  4.292692f,  4.309581f,  0.848423f,  4.292334f,
4274   4.298179f,  8.514713f,  14.911736f, 19.825352f,
4275 };
4276 
4277 static const float av1_simple_motion_search_term_none_std_128[28] = {
4278   1.796731f, 1.797056f, 1.898383f, 1.900753f, 1.846624f, 1.846953f, 1.906632f,
4279   1.908089f, 1.836533f, 1.835967f, 1.840262f, 1.840671f, 1.816836f, 1.817103f,
4280   1.879846f, 1.881333f, 1.803102f, 1.802654f, 2.263402f, 0.420354f, 1.117165f,
4281   1.083779f, 0.358611f, 1.101183f, 1.084938f, 2.462638f, 1.577009f, 1.574711f,
4282 };
4283 
4284 static const float av1_simple_motion_search_term_none_mean_64[28] = {
4285   10.904455f, 10.853546f, 9.247903f,  9.184479f,  9.251985f,  9.186686f,
4286   9.253490f,  9.190190f,  9.270079f,  9.204357f,  10.086511f, 10.031060f,
4287   10.100875f, 10.045429f, 10.069688f, 10.013173f, 10.082980f, 10.024640f,
4288   4.888378f,  0.878113f,  3.598450f,  3.628491f,  0.925833f,  3.560971f,
4289   3.573322f,  8.807137f,  13.348477f, 18.269117f,
4290 };
4291 
4292 static const float av1_simple_motion_search_term_none_std_64[28] = {
4293   1.789300f, 1.787061f, 1.823519f, 1.820226f, 1.794643f, 1.788620f, 1.797194f,
4294   1.795135f, 1.777795f, 1.773634f, 1.794000f, 1.790377f, 1.772197f, 1.769692f,
4295   1.819050f, 1.817139f, 1.793577f, 1.789333f, 1.998251f, 0.327156f, 0.885748f,
4296   0.853767f, 0.262043f, 0.902435f, 0.860033f, 1.224865f, 1.603411f, 1.589296f,
4297 };
4298 
4299 static const float av1_simple_motion_search_term_none_mean_32[28] = {
4300   9.818970f, 9.751199f, 8.015079f, 7.927318f, 8.029113f, 7.938330f,  8.012570f,
4301   7.923719f, 8.033508f, 7.941911f, 8.933057f, 8.857422f, 8.935639f,  8.859187f,
4302   8.905495f, 8.829741f, 8.929428f, 8.851351f, 4.114069f, 0.954752f,  2.645082f,
4303   2.709703f, 0.964678f, 2.652077f, 2.673393f, 9.430499f, 11.922798f, 16.942251f,
4304 };
4305 
4306 static const float av1_simple_motion_search_term_none_std_32[28] = {
4307   1.737107f, 1.734327f, 1.727923f, 1.720244f, 1.721570f, 1.712775f, 1.718028f,
4308   1.710370f, 1.711612f, 1.702596f, 1.754856f, 1.748855f, 1.741871f, 1.736304f,
4309   1.722428f, 1.717380f, 1.713563f, 1.707582f, 1.761170f, 0.207847f, 0.900058f,
4310   0.862356f, 0.184593f, 0.903822f, 0.856120f, 1.529199f, 1.412085f, 1.453153f,
4311 };
4312 
4313 static const float av1_simple_motion_search_term_none_mean_16[28] = {
4314   8.998877f, 8.912468f, 7.085255f, 6.953476f, 7.086386f, 6.954091f,  7.088727f,
4315   6.955747f, 7.093955f, 6.960635f, 8.065050f, 7.961432f, 8.071631f,  7.967233f,
4316   8.041699f, 7.937715f, 8.046791f, 7.942183f, 3.833521f, 0.978421f,  1.901347f,
4317   1.950124f, 0.979418f, 1.928000f, 1.936727f, 9.773951f, 10.735227f, 15.949769f,
4318 };
4319 
4320 static const float av1_simple_motion_search_term_none_std_16[28] = {
4321   1.641193f, 1.640172f, 1.614794f, 1.608906f, 1.609571f, 1.603580f, 1.606928f,
4322   1.601246f, 1.599230f, 1.593529f, 1.633747f, 1.630219f, 1.625695f, 1.622547f,
4323   1.633827f, 1.630182f, 1.626607f, 1.622777f, 1.548838f, 0.145303f, 0.744550f,
4324   0.736552f, 0.141980f, 0.742979f, 0.736977f, 1.366255f, 1.258794f, 1.294309f,
4325 };
4326 
4327 static const float av1_simple_motion_search_term_none_model_128[] = {
4328   -0.6106842357f, -1.0402954455f, 0.6054417656f,  -0.2116623578f,
4329   0.2447714930f,  0.3782256209f,  0.5095592479f,  -0.3275620904f,
4330   0.3886188013f,  0.2629499420f,  -0.1979599415f, -0.5389565605f,
4331   0.1209207902f,  -0.4913347466f, 0.3798542731f,  -0.2812861709f,
4332   -0.1049824167f, -0.1088672020f, 0.4059596517f,  -0.1347896613f,
4333   0.2276868621f,  0.0506386970f,  0.0071088411f,  0.0467952100f,
4334   0.2091247458f,  -0.7371964736f, 0.1368935545f,  0.3175247786f,
4335   -0.5493146094f,
4336 };
4337 
4338 static const float av1_simple_motion_search_term_none_model_64[] = {
4339   -0.4150046575f, -0.3954358561f, 0.1997997444f,  0.3395826831f,
4340   0.2827215753f,  0.3395683652f,  0.2483140395f,  0.2722216476f,
4341   0.2610308009f,  0.3724974359f,  -0.0551479654f, -0.1721616359f,
4342   -0.3459358629f, -0.0952524186f, -0.1428993840f, -0.0415654914f,
4343   -0.3169539902f, -0.0269429900f, 0.9891530919f,  -0.0125084982f,
4344   0.0972182377f,  0.0008889801f,  0.0205418050f,  0.0057237854f,
4345   0.1005222691f,  -0.2851321920f, -1.5150336445f, 0.1893942436f,
4346   -0.4337360901f,
4347 };
4348 
4349 static const float av1_simple_motion_search_term_none_model_32[] = {
4350   -0.4667392852f, -0.3893302767f, 0.1603498635f,  0.2304974726f,
4351   0.1404975592f,  0.2505516225f,  0.1423053884f,  0.2189318406f,
4352   0.1379765409f,  0.2638241296f,  -0.1342865463f, -0.0549054345f,
4353   -0.1925223436f, -0.1142702769f, 0.0127811659f,  0.0868639997f,
4354   -0.0643197251f, 0.0279496470f,  0.9904395769f,  -0.0095178685f,
4355   0.1179410649f,  -0.0013411972f, 0.0095060660f,  0.0195730400f,
4356   0.0779717771f,  -0.2498860763f, -0.8168817125f, -0.4798397348f,
4357   -0.6609679881f,
4358 };
4359 
4360 static const float av1_simple_motion_search_term_none_model_16[] = {
4361   -0.3021081992f, -0.4620153673f, 0.0448577479f,  0.1738455035f,
4362   0.0663209177f,  0.1629614573f,  0.0555168744f,  0.1631870212f,
4363   0.0425805150f,  0.1688564954f,  0.0434083772f,  -0.0046603915f,
4364   -0.0271580056f, -0.0183879127f, 0.1073730471f,  0.0314201476f,
4365   0.0576891756f,  0.0119723753f,  0.9084332022f,  -0.0188429077f,
4366   0.0755089811f,  -0.0172550234f, 0.0037663075f,  0.0022094472f,
4367   0.0500247894f,  -0.2944572004f, -0.8908521199f, -0.2555515792f,
4368   -0.5396254205f,
4369 };
4370 
4371 // Early termination in firstpass
4372 static const float av1_fp_simple_motion_search_term_none_mean_32[20] = {
4373   10.216787f, 10.167575f, 8.405353f, 8.340786f,  8.436503f,
4374   8.373259f,  8.444113f,  8.379074f, 8.448215f,  8.384669f,
4375   4.107491f,  0.923902f,  2.702687f, 2.712742f,  0.953166f,
4376   2.703244f,  2.707070f,  9.549801f, 12.013671f, 17.059454f,
4377 };
4378 
4379 static const float av1_fp_simple_motion_search_term_none_std_32[20] = {
4380   1.886182f, 1.886638f, 1.884324f, 1.883410f, 1.851800f, 1.851652f, 1.847129f,
4381   1.848014f, 1.832187f, 1.832360f, 1.758185f, 0.265155f, 0.939592f, 0.932395f,
4382   0.211284f, 0.950024f, 0.945295f, 1.846744f, 1.453674f, 1.505994f,
4383 };
4384 
4385 static const float av1_fp_simple_motion_search_term_none_mean_16[20] = {
4386   9.131485f, 9.065489f, 7.254479f, 7.158092f, 7.274240f,  7.178158f,  7.278780f,
4387   7.182110f, 7.278793f, 7.182714f, 3.981902f, 0.964040f,  2.080875f,  2.087185f,
4388   0.973397f, 2.088189f, 2.090166f, 9.386505f, 10.826546f, 15.985614f,
4389 };
4390 
4391 static const float av1_fp_simple_motion_search_term_none_std_16[20] = {
4392   1.681172f, 1.688587f, 1.710854f, 1.717533f, 1.684010f, 1.691476f, 1.683537f,
4393   1.691523f, 1.674699f, 1.682130f, 1.639731f, 0.186191f, 0.796448f, 0.795075f,
4394   0.160921f, 0.791005f, 0.790048f, 1.430960f, 1.337976f, 1.370498f,
4395 };
4396 
4397 static const float av1_fp_simple_motion_search_term_none_mean_8[20] = {
4398   7.821461f, 7.714526f, 5.799360f, 5.606948f, 5.805885f, 5.614357f,  5.794252f,
4399   5.599669f, 5.798780f, 5.605399f, 4.069016f, 0.977720f, 1.577513f,  1.581266f,
4400   0.983371f, 1.524603f, 1.524952f, 9.221803f, 9.508886f, 14.972815f,
4401 };
4402 
4403 static const float av1_fp_simple_motion_search_term_none_std_8[20] = {
4404   1.618036f, 1.634415f, 1.652861f, 1.672006f, 1.646337f, 1.664935f, 1.650876f,
4405   1.670476f, 1.645141f, 1.664301f, 1.502258f, 0.147592f, 0.760353f, 0.762547f,
4406   0.127879f, 0.741096f, 0.742186f, 1.042003f, 1.292524f, 1.250398f,
4407 };
4408 
4409 #define NUM_HIDDEN_LAYERS_32 1
4410 #define NUM_FEATURES_32 20
4411 #define NUM_LAYER_0_UNITS_32 20
4412 #define NUM_LOGITS_32 1
4413 
4414 static const float
4415     av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_32[] = {
4416       -0.293987f,   0.796773f,     -0.0888487f, -0.00796495f, -0.343768f,
4417       0.0783252f,   0.0596814f,    -0.235432f,  -0.0780005f,  -0.409017f,
4418       -0.256821f,   -0.281654f,    1.00889f,    0.701893f,    -0.0181661f,
4419       0.119718f,    0.0956582f,    0.76792f,    0.235693f,    0.351628f,
4420       -1.28111f,    -1.45847f,     0.387732f,   0.476054f,    0.384561f,
4421       0.427465f,    0.11875f,      -0.0176598f, -0.0528453f,  0.395589f,
4422       -0.331994f,   0.0442108f,    0.195171f,   -0.0377402f,  -0.0736457f,
4423       -0.0490903f,  0.116165f,     -0.549512f,  0.12968f,     0.641055f,
4424       -1.03066f,    -0.601979f,    0.351981f,   -0.122019f,   0.00869275f,
4425       0.399222f,    -0.343995f,    -0.444257f,  -0.160805f,   -0.537537f,
4426       0.261478f,    -0.163785f,    0.218916f,   0.106506f,    -0.103819f,
4427       0.0121841f,   0.284757f,     -0.362989f,  1.10793f,     0.477236f,
4428       -0.424117f,   -0.884156f,    -0.468291f,  -0.510531f,   0.791441f,
4429       0.75243f,     0.839871f,     0.604127f,   -0.182956f,   -0.246703f,
4430       -1.25861f,    0.0546303f,    0.0811323f,  0.00655988f,  0.0286305f,
4431       -0.00938366f, -0.0291418f,   -0.231632f,  -0.331077f,   1.12479f,
4432       -0.635514f,   -0.146066f,    0.853122f,   0.923699f,    0.180011f,
4433       -0.252973f,   0.1474f,       -0.454344f,  0.354736f,    0.576872f,
4434       -1.43275f,    0.0327868f,    0.140849f,   -0.102523f,   0.0524867f,
4435       0.007091f,    -0.00232578f,  -0.536116f,  -0.700144f,   0.166646f,
4436       0.0636548f,   0.44645f,      -0.346062f,  -0.685779f,   -1.0792f,
4437       -0.999219f,   0.442744f,     0.371198f,   0.777914f,    0.719409f,
4438       -0.417984f,   0.0602868f,    0.0225539f,  0.0457407f,   0.0249501f,
4439       0.0126021f,   0.00450792f,   0.0485095f,  0.203485f,    0.584116f,
4440       -0.599426f,   -0.244633f,    0.168231f,   -0.00134934f, -0.106987f,
4441       -0.0490239f,  -0.22029f,     0.138017f,   0.373674f,    0.00638684f,
4442       -2.08003f,    0.106453f,     0.124456f,   -0.0286108f,  0.0422698f,
4443       0.013734f,    0.0780971f,    -0.40173f,   0.473453f,    1.16836f,
4444       -0.251035f,   0.0119074f,    0.319241f,   0.0422023f,   -0.730454f,
4445       -0.745948f,   0.796709f,     0.277634f,   0.09711f,     -0.212224f,
4446       0.825348f,    0.0208521f,    -0.0238098f, 0.00929265f,  0.0516351f,
4447       -0.02329f,    0.0983163f,    -0.180721f,  0.0122096f,   -0.246159f,
4448       0.61468f,     0.923765f,     0.240435f,   -0.294845f,   -0.495317f,
4449       -0.0563837f,  -0.417936f,    0.154874f,   -0.604407f,   -0.0681337f,
4450       -0.65738f,    -0.0270073f,   0.0920023f,  -0.0742724f,  0.820862f,
4451       -0.602758f,   -1.20617f,     -0.201707f,  0.869499f,    -0.0539076f,
4452       0.403097f,    0.429168f,     -0.938227f,  -0.830894f,   -0.362462f,
4453       -0.0658648f,  0.471469f,     -0.264827f,  0.610275f,    0.367995f,
4454       0.735662f,    -0.0473157f,   -0.0380545f, -0.0848067f,  -0.146108f,
4455       -0.125875f,   -0.0576117f,   -0.296198f,  -0.100443f,   -0.212971f,
4456       0.593524f,    1.23111f,      -0.810009f,  -0.604572f,   0.203021f,
4457       0.256285f,    -1.17049f,     -1.19156f,   0.24365f,     0.727876f,
4458       -0.466826f,   0.0298762f,    -0.0331735f, -0.0109056f,  0.0114862f,
4459       0.00396703f,  0.0385985f,    -0.0587946f, 0.821079f,    0.0582033f,
4460       0.349156f,    1.03529f,      -0.407036f,  0.200308f,    -0.265649f,
4461       -0.104567f,   0.161149f,     -0.0717528f, -0.0112724f,  0.0681578f,
4462       0.103809f,    -0.0807997f,   0.0316814f,  -0.332323f,   0.112254f,
4463       -0.163981f,   0.118988f,     -0.777055f,  -1.34047f,    -0.910482f,
4464       0.74599f,     -0.59633f,     0.165649f,   -0.594998f,   0.0845802f,
4465       0.00440975f,  0.122606f,     -0.463991f,  0.418502f,    -0.339126f,
4466       1.41847f,     -0.109594f,    -0.411879f,  -0.444865f,   -0.0404821f,
4467       -0.0607352f,  -0.663753f,    -0.724327f,  -0.138642f,   0.834144f,
4468       -0.811695f,   -0.930264f,    0.150993f,   -0.325565f,   0.0615853f,
4469       -0.473993f,   0.0966587f,    0.315197f,   1.0345f,      0.35441f,
4470       0.703234f,    -0.335715f,    0.783153f,   0.467976f,    -0.0234736f,
4471       0.549724f,    0.539107f,     -0.510182f,  -0.154442f,   0.0126656f,
4472       1.66711f,     0.884555f,     0.118675f,   -0.341705f,   0.195316f,
4473       -0.0366564f,  -0.619244f,    -0.634092f,  -0.559951f,   0.0564255f,
4474       0.765917f,    0.0510238f,    0.0667615f,  0.0699302f,   -0.0351751f,
4475       -0.0484402f,  -0.000792665f, -0.10775f,   -0.337121f,   -0.983947f,
4476       0.517793f,    1.34977f,      -0.567602f,  0.129921f,    -0.443722f,
4477       -0.276277f,   -0.501404f,    -0.183234f,  -0.553055f,   -0.447434f,
4478       -0.35529f,    -0.0444689f,   0.0192031f,  0.0372702f,   -0.195202f,
4479       -0.020753f,   -0.0247035f,   0.420298f,   1.39373f,     0.203699f,
4480       -0.218818f,   0.250734f,     -0.0282348f, 0.411986f,    -0.262946f,
4481       0.526339f,    0.242769f,     -0.159857f,  -0.546788f,   -0.0410147f,
4482       0.954238f,    -0.0252765f,   0.639488f,   -0.491367f,   -0.0572638f,
4483       0.285763f,    -0.45764f,     0.121657f,   -1.24374f,    -0.372479f,
4484       -0.111521f,   0.194134f,     -0.271364f,  0.179678f,    0.121237f,
4485       -0.14305f,    -0.205662f,    0.216891f,   0.344568f,    -0.523745f,
4486       -1.00908f,    0.180965f,     0.0263031f,  -0.0556144f,  0.0831083f,
4487       -0.0623274f,  0.112748f,     0.597137f,   -0.502616f,   -1.10624f,
4488       -0.0487462f,  -1.10744f,     -0.125653f,  0.277049f,    -0.141329f,
4489       -0.00457003f, -0.161038f,    0.588462f,   0.323317f,    0.49762f,
4490       0.477561f,    0.901705f,     -0.264511f,  0.256557f,    0.076023f,
4491       -0.0460696f,  0.0830666f,    -0.0651269f, -0.881245f,   -0.285999f,
4492       0.53127f,     0.914533f,     0.0505795f,  -0.3054f,     -0.0988696f,
4493       -0.0658403f,  0.15979f,      -0.453316f,  -0.824834f,   -0.280222f,
4494       -0.686952f,   -0.0768344f,   -1.12235f,   -0.815408f,   0.0202134f,
4495       -0.111892f,   0.0847659f,    -0.18763f,   0.597782f,    0.364016f
4496     };
4497 
4498 static const float
4499     av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_32[] = {
4500       -1.541f,     -0.00935641f, -1.50754f, -0.638648f, -0.679403f,
4501       -0.0387804f, -0.714791f,   -1.69522f, 0.435677f,  -1.5846f,
4502       0.108788f,   0.614982f,    0.111048f, -0.465826f, -0.611358f,
4503       0.637197f,   0.929621f,    -1.20889f, 0.954558f,  0.716529f
4504     };
4505 
4506 static const float av1_fp_simple_motion_search_term_none_logits_kernel_32[] = {
4507   0.396195f,   -0.791364f,  -0.881893f, 1.0542069f, 0.772562f,
4508   0.60815647f, 1.117405f,   -1.272638f, 0.483183f,  -0.917147f,
4509   0.690799f,   -0.601466f,  -0.545536f, -0.416353f, -0.927874f,
4510   0.972198f,   -0.3770457f, 0.542694f,  -0.591889f, 0.464565f
4511 };
4512 
4513 static const float av1_fp_simple_motion_search_term_none_logits_bias_32[] = {
4514   -0.590318f
4515 };
4516 
4517 static const NN_CONFIG av1_fp_simple_motion_search_term_none_nn_config_32 = {
4518   NUM_FEATURES_32,
4519   NUM_LOGITS_32,
4520   NUM_HIDDEN_LAYERS_32,
4521   {
4522       NUM_LAYER_0_UNITS_32,
4523   },
4524   {
4525       av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_32,
4526       av1_fp_simple_motion_search_term_none_logits_kernel_32,
4527   },
4528   {
4529       av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_32,
4530       av1_fp_simple_motion_search_term_none_logits_bias_32,
4531   },
4532 };
4533 
4534 #undef NUM_HIDDEN_LAYERS_32
4535 #undef NUM_FEATURES_32
4536 #undef NUM_LAYER_0_UNITS_32
4537 #undef NUM_LOGITS_32
4538 
4539 #define NUM_HIDDEN_LAYERS_16 1
4540 #define NUM_FEATURES_16 20
4541 #define NUM_LAYER_0_UNITS_16 24
4542 #define NUM_LOGITS_16 1
4543 
4544 static const float
4545     av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_16[] = {
4546       -0.315922f,   0.74455f,     -0.0196939f,  0.238336f,    0.288554f,
4547       0.0845902f,   -0.0121831f,  0.455303f,    0.0235902f,   0.218997f,
4548       -0.0445164f,  0.0752211f,   0.0539915f,   -0.0439682f,  -0.397139f,
4549       -0.0030004f,  -0.106365f,   0.845384f,    0.684638f,    -0.965702f,
4550       0.307643f,    -0.0433377f,  -0.0644826f,  -0.214946f,   -0.44467f,
4551       0.142967f,    0.0109982f,   -0.344458f,   -0.42947f,    0.269175f,
4552       -0.88534f,    -0.28077f,    -1.36018f,    -0.33725f,    -0.0885953f,
4553       -0.123887f,   0.218107f,    -0.0759977f,  0.739124f,    0.684048f,
4554       0.577964f,    -0.328481f,   -0.247837f,   0.00546713f,  0.191895f,
4555       -0.145274f,   0.320121f,    -0.482379f,   0.534585f,    -0.1582f,
4556       0.944784f,    0.944665f,    0.0494451f,   -0.0399724f,  -0.170375f,
4557       -0.0869746f,  0.106216f,    -0.120556f,   -1.57849f,    -0.752895f,
4558       0.424454f,    -0.0269515f,  0.00398589f,  0.214165f,    -0.142986f,
4559       0.199223f,    0.049624f,    -0.116783f,   -0.648119f,   -0.311599f,
4560       0.122629f,    -0.0338422f,  0.345092f,    -0.408254f,   0.601037f,
4561       -0.00146985f, 0.00133926f,  0.0392668f,   -0.931156f,   0.31429f,
4562       -0.150243f,   0.0755763f,   -0.32177f,    0.258521f,    -0.104078f,
4563       -0.144506f,   0.0199566f,   -0.454723f,   -0.292959f,   -0.0953681f,
4564       -1.24843f,    0.446814f,    -0.311363f,   0.0590878f,   -0.0568717f,
4565       -0.421585f,   0.179852f,    0.668763f,    0.48914f,     0.290584f,
4566       -1.14053f,    -1.37576f,    0.420112f,    -0.158582f,   0.268231f,
4567       0.252999f,    0.276423f,    0.529033f,    0.141127f,    0.702762f,
4568       0.181407f,    -0.0279289f,  -0.0194757f,  0.0752152f,   -0.136963f,
4569       0.00902489f,  0.125334f,    0.0680212f,   -0.370449f,   0.438003f,
4570       -0.600869f,   0.154209f,    -0.36306f,    -0.484209f,   0.140093f,
4571       0.0743079f,   -0.143317f,   0.0442872f,   0.272089f,    0.601531f,
4572       1.20687f,     -0.280695f,   0.222235f,    -0.0106747f,  -0.017026f,
4573       0.204008f,    -0.0316111f,  -0.64679f,    -0.866749f,   -0.774231f,
4574       0.306231f,    -0.0940114f,  -0.56555f,    -0.34399f,    0.425142f,
4575       0.424064f,    -0.50189f,    -0.146558f,   0.544899f,    0.141728f,
4576       1.14592f,     -0.0124826f,  0.111613f,    -0.0862228f,  0.0211737f,
4577       0.0614017f,   0.0245077f,   -0.454523f,   -0.0766391f,  -0.436808f,
4578       0.251409f,    -0.13354f,    -0.242447f,   -0.311807f,   -0.844505f,
4579       -0.671486f,   0.0946297f,   0.241702f,    0.856521f,    0.529763f,
4580       -0.869772f,   -0.0016341f,  0.14511f,     0.0136254f,   -0.0359721f,
4581       -0.0454713f,  0.00664495f,  0.0373555f,   0.653991f,    -0.075867f,
4582       -0.102728f,   -0.947685f,   -0.119479f,   -0.145413f,   0.148364f,
4583       0.310885f,    -0.266837f,   0.354087f,    0.299469f,    0.603911f,
4584       0.257161f,    0.0190527f,   0.152862f,    -0.0987196f,  -0.293369f,
4585       0.139026f,    -0.128421f,   0.0505933f,   -0.703803f,   1.08628f,
4586       -0.562294f,   -0.818943f,   0.102178f,    0.727399f,    -0.228433f,
4587       0.484057f,    0.0595919f,   -0.0559087f,  -0.549447f,   0.176168f,
4588       1.41744f,     -0.126284f,   0.0987251f,   -0.00123073f, 0.00510827f,
4589       0.105209f,    0.0671775f,   -0.438525f,   0.211028f,    -0.782459f,
4590       0.286411f,    -0.459887f,   0.0633669f,   0.329958f,    -0.0736945f,
4591       0.45188f,     -0.2447f,     0.676601f,    0.600321f,    -0.0336198f,
4592       0.108531f,    0.0452834f,   -0.0848577f,  0.0731281f,   1.32381f,
4593       -0.118349f,   0.129497f,    -0.840938f,   -1.45444f,    -0.559047f,
4594       -0.248109f,   -0.491559f,   -0.139812f,   0.175964f,    0.168687f,
4595       0.123031f,    0.201625f,    0.422849f,    0.34436f,     0.0426694f,
4596       0.558045f,    -0.246772f,   0.679483f,    -0.0959578f,  -0.102879f,
4597       0.391029f,    0.280906f,    0.0867408f,   -1.10932f,    0.402526f,
4598       -0.227285f,   0.336087f,    -0.237765f,   0.185619f,    -0.309732f,
4599       0.0781132f,   -0.0234955f,  0.0828806f,   0.19966f,     -0.241288f,
4600       -0.224634f,   0.0638918f,   -0.143521f,   -0.0206692f,  -0.27131f,
4601       0.973051f,    1.12031f,     0.262846f,    0.471585f,    0.105231f,
4602       -0.386434f,   -0.355846f,   0.7359f,      0.567308f,    0.130768f,
4603       0.242369f,    -0.0272523f,  -0.118436f,   0.374145f,    0.24802f,
4604       -1.00186f,    -0.0241195f,  0.0140446f,   0.0202831f,   0.163197f,
4605       0.0399298f,   -0.00912791f, -0.280572f,   -0.309893f,   -0.644495f,
4606       0.243838f,    0.731391f,    0.0725078f,   0.350308f,    -0.136691f,
4607       0.208814f,    0.0218567f,   -0.0805393f,  -0.18681f,    -0.214638f,
4608       0.273354f,    -0.355047f,   0.242748f,    0.472951f,    -0.202705f,
4609       0.405247f,    0.161622f,    -0.284883f,   -1.31181f,    -0.661056f,
4610       -0.248219f,   -0.827307f,   0.289221f,    0.660529f,    0.48563f,
4611       0.407366f,    0.0327303f,   -0.0610309f,  -0.647064f,   0.0899991f,
4612       0.376267f,    1.27555f,     0.0264175f,   0.153931f,    1.07345f,
4613       0.0715052f,   0.174473f,    0.01322f,     -0.715723f,   0.113909f,
4614       0.100968f,    -0.457287f,   -0.672022f,   -0.20532f,    0.895176f,
4615       0.357034f,    0.5413f,      0.918393f,    -0.455f,      -0.499617f,
4616       -1.21799f,    0.0634338f,   0.144944f,    -0.106715f,   0.0227713f,
4617       -0.0203213f,  0.030851f,    -0.0726756f,  0.589192f,    -0.060841f,
4618       -0.198521f,   0.497179f,    -0.0591156f,  -0.135466f,   -0.132638f,
4619       -0.181333f,   -0.332358f,   0.0349959f,   0.212885f,    -0.536206f,
4620       -0.425009f,   -0.035525f,   0.0384449f,   0.0360549f,   -0.0383953f,
4621       -0.0263281f,  -0.0228435f,  1.11771f,     0.928061f,    -0.163923f,
4622       -0.327868f,   -0.894518f,   0.00448907f,  0.0805977f,   0.329559f,
4623       0.157429f,    0.292729f,    0.497688f,    0.188659f,    0.203724f,
4624       -1.26001f,    -0.0392533f,  -0.0566088f,  0.000859925f, 0.125254f,
4625       0.054261f,    0.0357295f,   -0.393813f,   -0.275944f,   0.299657f,
4626       -0.211421f,   0.038172f,    -0.439829f,   -0.913949f,   0.35642f,
4627       0.865473f,    -0.472033f,   -0.752376f,   0.995255f,    0.417965f,
4628       -0.680645f,   0.0622027f,   0.128878f,    -0.0357859f,  0.0793577f,
4629       0.203629f,    -0.0600867f,  0.0512268f,   0.528584f,    0.23889f,
4630       0.38255f,     -0.216407f,   -0.0338828f,  0.0328103f,   -0.885678f,
4631       -0.716634f,   0.438663f,    0.320841f,    -0.119656f,   0.626092f,
4632       0.8526f,      -0.0325005f,  -0.0275416f,  -0.171131f,   0.0260563f,
4633       -0.0162027f,  0.0879367f,   -0.340473f,   0.0220265f,   -0.1731f,
4634       0.512539f,    0.587822f,    -0.175619f,   0.177215f,    -0.35458f,
4635       -0.159059f,   -0.423754f,   0.0198413f,   -0.336208f,   -0.359052f,
4636       -1.50819f,    0.0628184f,   0.054506f,    0.0048834f,   0.361657f,
4637       0.00986886f,  -0.0721521f,  -0.256765f,   1.41173f,     0.376196f,
4638       -0.0783331f,  0.174803f,    -0.00240091f, -0.306571f,   -0.304654f,
4639       -0.0348377f,  0.115569f,    -0.20359f,    -0.162341f,   -0.0443526f,
4640       -0.848317f,   -0.228167f,   0.699534f,    0.482092f,    -0.0921484f,
4641       -0.172425f,   -0.0610094f,  -0.188327f,   0.836209f,    0.541725f
4642     };
4643 
4644 static const float
4645     av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_16[] = {
4646       -0.388147f, -0.0868767f, 0.702129f,  0.376659f, -0.709988f, 0.496603f,
4647       -0.238442f, -1.35761f,   -0.391887f, 0.235468f, -0.327982f, 0.731842f,
4648       1.0949f,    -0.789218f,  -0.881452f, 0.514341f, 0.727894f,  -0.494498f,
4649       -1.32304f,  -1.22643f,   -0.294287f, -1.3974f,  -0.128148f, -0.0956137f
4650     };
4651 
4652 static const float av1_fp_simple_motion_search_term_none_logits_kernel_16[] = {
4653   0.456147f,   0.248707f,  -0.5205241f, -0.1506567f, 0.388359f,   -0.6074409f,
4654   -0.4719775f, -0.733864f, 0.5588447f,  -0.4021345f, -1.140733f,  -0.73399f,
4655   -0.4299591f, 0.450688f,  0.817564f,   -0.265486f,  -0.3525806f, 0.55188314f,
4656   1.365457f,   1.180764f,  0.587772f,   -0.870683f,  0.818839f,   0.318488f
4657 };
4658 
4659 static const float av1_fp_simple_motion_search_term_none_logits_bias_16[] = {
4660   -0.1046478f
4661 };
4662 
4663 static const NN_CONFIG av1_fp_simple_motion_search_term_none_nn_config_16 = {
4664   NUM_FEATURES_16,
4665   NUM_LOGITS_16,
4666   NUM_HIDDEN_LAYERS_16,
4667   {
4668       NUM_LAYER_0_UNITS_16,
4669   },
4670   {
4671       av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_16,
4672       av1_fp_simple_motion_search_term_none_logits_kernel_16,
4673   },
4674   {
4675       av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_16,
4676       av1_fp_simple_motion_search_term_none_logits_bias_16,
4677   },
4678 };
4679 
4680 #undef NUM_HIDDEN_LAYERS_16
4681 #undef NUM_FEATURES_16
4682 #undef NUM_LAYER_0_UNITS_16
4683 #undef NUM_LOGITS_16
4684 
4685 #define NUM_HIDDEN_LAYERS_8 1
4686 #define NUM_FEATURES_8 20
4687 #define NUM_LAYER_0_UNITS_8 16
4688 #define NUM_LOGITS_8 1
4689 
4690 static const float
4691     av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_8[] = {
4692       -1.11024f,    -0.530449f,    -0.164768f,  0.675431f,   0.456155f,
4693       0.711099f,    -0.248095f,    0.112132f,   -0.131481f,  0.234457f,
4694       0.128073f,    0.306214f,     0.175471f,   0.220189f,   -0.270533f,
4695       0.293534f,    -0.0795547f,   0.234901f,   -0.191754f,  0.101171f,
4696       -0.108621f,   0.395477f,     -0.529459f,  -0.354854f,  -0.941334f,
4697       -0.237689f,   0.39357f,      0.527129f,   0.174333f,   -0.00520422f,
4698       1.22219f,     -0.21815f,     0.0866816f,  -0.29591f,   -0.212968f,
4699       0.00431436f,  -0.295382f,    -0.582317f,  -0.284654f,  0.486427f,
4700       -0.202448f,   -0.0421883f,   -0.116346f,  -0.345832f,  -0.0471637f,
4701       -0.149954f,   -0.0969526f,   -0.59491f,   0.594364f,   0.298285f,
4702       -1.33301f,    0.149562f,     0.097433f,   0.157641f,   -0.231132f,
4703       -0.0191656f,  0.149396f,     0.811553f,   1.07336f,    0.140674f,
4704       1.02134f,     0.455909f,     -0.0548795f, 0.0459996f,  -0.0589837f,
4705       -0.116328f,   -0.607502f,    -0.232595f,  -0.517977f,  -0.325901f,
4706       1.35047f,     -0.148698f,    0.0313182f,  0.181634f,   0.06539f,
4707       0.00820322f,  0.0522113f,    -1.06071f,   -0.817999f,  -0.527422f,
4708       -1.39175f,    -0.110088f,    0.0858626f,  -0.247541f,  0.29043f,
4709       1.13767f,     0.185834f,     0.390613f,   -0.501175f,  -0.214176f,
4710       -0.256376f,   0.496687f,     0.240471f,   0.218852f,   0.513543f,
4711       0.400559f,    -0.249168f,    -0.752987f,  0.430491f,   -0.72299f,
4712       0.339754f,    0.396623f,     -0.0638322f, 0.353122f,   0.355662f,
4713       -0.0704821f,  0.195448f,     0.179396f,   0.486533f,   0.0815535f,
4714       -0.503726f,   -0.000321223f, 0.501591f,   -0.117849f,  0.217667f,
4715       -0.123391f,   -0.4026f,      0.149756f,   -0.0359276f, -0.0990213f,
4716       -0.215278f,   -0.293649f,    0.301629f,   -0.11081f,   -0.206725f,
4717       -0.00147108f, 0.363644f,     -0.430092f,  0.169524f,   0.116091f,
4718       -0.583605f,   -0.0974948f,   0.253256f,   0.22648f,    0.136902f,
4719       -0.882541f,   -0.75078f,     -0.0629343f, 0.411035f,   0.265742f,
4720       -0.360904f,   -0.899324f,    0.605871f,   0.0318372f,  0.0735312f,
4721       -0.00960722f, 0.691249f,     0.127449f,   -0.133021f,  -0.0793589f,
4722       0.665591f,    -0.0682262f,   -0.0437626f, 0.0783621f,  2.25727f,
4723       0.126529f,    -0.0320763f,   -0.261759f,  -1.19987f,   0.216295f,
4724       -0.253886f,   -0.642908f,    0.1865f,     0.00299179f, 0.0246782f,
4725       -0.00750628f, 0.566367f,     0.99916f,    -0.0209625f, 0.273254f,
4726       1.09724f,     0.30026f,      0.21585f,    -0.0276715f, 0.338996f,
4727       0.129884f,    -0.00628438f,  0.0461783f,  -1.36378f,   -0.394756f,
4728       -0.395261f,   0.215928f,     0.252803f,   -0.207108f,  -0.0506214f,
4729       -0.0138889f,  0.124197f,     -0.0522996f, 0.533803f,   -0.25729f,
4730       -0.463514f,   0.128322f,     -1.04751f,   -0.605498f,  -0.107235f,
4731       -0.00813289f, 0.539742f,     -0.0524178f, 0.272101f,   0.151935f,
4732       0.607511f,    -0.0608427f,   0.36342f,    0.0999134f,  0.69712f,
4733       -0.152471f,   0.364244f,     0.410644f,   0.312606f,   0.405679f,
4734       -0.371656f,   -0.0492209f,   -0.148911f,  0.214996f,   -0.274749f,
4735       -0.0372888f,  0.079023f,     -0.429136f,  -1.30393f,   -0.833824f,
4736       -1.31373f,    -0.445343f,    0.526917f,   1.30569f,    -0.0626746f,
4737       0.282353f,    -0.28552f,     0.28084f,    -0.234934f,  0.227076f,
4738       1.09919f,     0.33248f,      -0.114933f,  0.40629f,    0.331031f,
4739       0.245334f,    -0.0318782f,   0.00735305f, -1.58715f,   0.126443f,
4740       -0.09472f,    -0.182152f,    0.311673f,   -0.186136f,  0.817743f,
4741       0.928961f,    0.117334f,     -0.373644f,  -0.0797864f, 0.205565f,
4742       0.0789797f,   0.0757131f,    -0.152409f,  0.30301f,    -0.0170824f,
4743       -0.194496f,   0.485547f,     0.370124f,   -0.802044f,  -0.789671f,
4744       0.669258f,    0.55082f,      -0.438853f,  0.0597597f,  -0.0148101f,
4745       -0.41603f,    0.0486339f,    -0.464523f,  -0.413725f,  0.00907629f,
4746       0.70351f,     -0.136422f,    -0.145957f,  -0.0626726f, -0.115773f,
4747       -0.333937f,   0.135474f,     -0.379598f,  -0.134422f,  0.227595f,
4748       0.908927f,    0.759504f,     -0.0088258f, -0.349333f,  0.122667f,
4749       -0.682175f,   0.2201f,       -0.332003f,  -0.44433f,   -0.620308f,
4750       -1.36716f,    -0.0167907f,   -0.538969f,  0.256824f,   -0.0706724f,
4751       -0.0392471f,  -0.156312f,    0.153699f,   1.41967f,    0.0434739f,
4752       0.428178f,    -0.0714879f,   0.0912104f,  0.00687985f, 0.341789f,
4753       0.217381f,    0.128288f,     0.0286751f,  0.527344f,   -0.428139f,
4754       0.60908f,     1.02074f,      -0.0977894f, 0.158067f,   0.28958f,
4755       -0.065152f,   0.120616f,     -0.882976f,  -1.10413f,   -1.37497f
4756     };
4757 
4758 static const float
4759     av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_8[] = {
4760       1.37086f,  -1.61858f, -1.32395f,  0.276031f, -0.124696f, -1.71489f,
4761       -1.68429f, 1.79103f,  -0.335306f, -1.81523f, 0.841083f,  -0.542628f,
4762       -1.82168f, 0.459829f, 0.0949306f, 0.918486f
4763     };
4764 
4765 static const float av1_fp_simple_motion_search_term_none_logits_kernel_8[] = {
4766   -0.283418f, -0.444453f, 0.4977782f, -0.4138758f, 0.41890771f, 0.22149438f,
4767   0.545079f,  -0.729164f, 0.619389f,  0.5169534f,  -0.4236282f, 0.7304213f,
4768   0.531938f,  -0.14828f,  0.75119f,   -0.464074f
4769 };
4770 
4771 static const float av1_fp_simple_motion_search_term_none_logits_bias_8[] = {
4772   -2.22338f
4773 };
4774 
4775 static const NN_CONFIG av1_fp_simple_motion_search_term_none_nn_config_8 = {
4776   NUM_FEATURES_8,
4777   NUM_LOGITS_8,
4778   NUM_HIDDEN_LAYERS_8,
4779   {
4780       NUM_LAYER_0_UNITS_8,
4781   },
4782   {
4783       av1_fp_simple_motion_search_term_none_hiddenlayer_0_kernel_8,
4784       av1_fp_simple_motion_search_term_none_logits_kernel_8,
4785   },
4786   {
4787       av1_fp_simple_motion_search_term_none_hiddenlayer_0_bias_8,
4788       av1_fp_simple_motion_search_term_none_logits_bias_8,
4789   },
4790 };
4791 
4792 #undef NUM_HIDDEN_LAYERS_8
4793 #undef NUM_FEATURES_8
4794 #undef NUM_LAYER_0_UNITS_8
4795 #undef NUM_LOGITS_8
4796 
4797 static const float av1_fp_simple_motion_search_term_none_thresh_32 =
4798     -2.2884985045792563f;
4799 static const float av1_fp_simple_motion_search_term_none_thresh_16 =
4800     -1.6656874577527165f;
4801 static const float av1_fp_simple_motion_search_term_none_thresh_8 =
4802     -3.608804354309157f;
4803 
4804 #ifdef __cplusplus
4805 }  // extern "C"
4806 #endif
4807 
4808 #endif  // AOM_AV1_ENCODER_PARTITION_MODEL_WEIGHTS_H_
4809