Tweak the ML partition aggressiveness STATS_CHANGED Change-Id: Ic857b36c94d405b99cd2b063af3dc52231afead7
diff --git a/av1/encoder/partition_model_weights.h b/av1/encoder/partition_model_weights.h index 493d893..576b981 100644 --- a/av1/encoder/partition_model_weights.h +++ b/av1/encoder/partition_model_weights.h
@@ -2622,61 +2622,63 @@ #define FEATURE_SIZE 4 // Mean and std static const float av1_hd_partition_breakout_nn_mean_128[FEATURE_SIZE] = { - 10.730792598027795f, - 376.6180794798389f, - 699.9787168227473f, - 2069.426780349039f, + 1.7569518370024242f, + 5.5324198861333107f, + 5.5260479508123526f, + 7.1913091289363038f, }; static const float av1_hd_partition_breakout_nn_std_128[FEATURE_SIZE] = { - 29.023640354060735f, 659.3967891826734f, 940.2882285903872f, - 2566.2070264789245f + 1.3286579037156365f, 1.4302357105525316f, 1.947275803679912f, + 1.4856880036033451f }; static const float av1_hd_partition_breakout_nn_mean_64[FEATURE_SIZE] = { - 22.220171631578605f, - 360.49969866754554f, - 642.6605866566223f, - 1022.2158389850164f, + 2.5083185600034166f, + 5.5459875545856132f, + 5.4419578738009537f, + 6.3575563591302293f, }; static const float av1_hd_partition_breakout_nn_std_64[FEATURE_SIZE] = { - 69.0770174031301f, 758.5544028348198f, 891.8245485291498f, 1901.3873859786233f + 1.5327874427371651f, 1.4413133143150851f, 1.852577618179309f, + 1.7441221691208793f }; static const float av1_hd_partition_breakout_nn_mean_32[FEATURE_SIZE] = { - 56.23900643356807f, - 411.708483646884f, - 552.7719521312714f, - 671.8117635469166f, + 3.2925380941946631f, + 5.6592503897046313f, + 5.2656875189877344f, + 5.88623228906336f, }; static const float av1_hd_partition_breakout_nn_std_32[FEATURE_SIZE] = { - 198.97960359590587f, 959.3829390828911f, 830.9980036115545f, - 1498.4982976375336f + 1.6894083970930263f, 1.4685304613224999f, 1.8598497961630693f, + 1.7648307635114531f }; static const float av1_hd_partition_breakout_nn_mean_16[FEATURE_SIZE] = { - 136.9843430416749f, - 532.9528184025893f, - 504.7980827006724f, - 445.7892729738405f, + 4.334899946924148f, + 5.8802354638427667f, + 5.2098555971571239f, + 5.5808807097219715f, }; static const float av1_hd_partition_breakout_nn_std_16[FEATURE_SIZE] = { - 478.52767648003527f, 1308.8412547684493f, 795.1318845991449f, - 1176.5665198192673f + 1.7269713123782717f, 1.5227226730026808f, 1.9255508380289716f, + 1.6959046024771494f }; static const float av1_hd_partition_breakout_nn_mean_8[FEATURE_SIZE] = { - 275.0553416698255f, - 566.7785552508784f, - 406.48781378761095f, - 247.1067473273997f, + 5.17688933331104f, + 6.0068390263641165f, + 4.9884361306237377f, + 5.0973575838899237f, }; static const float av1_hd_partition_breakout_nn_std_8[FEATURE_SIZE] = { - 935.5925922140688f, 1435.299322400821f, 678.9806893808023f, 791.3637162723987f + 1.693000978888298f, 1.4741850818874491f, 1.8398258859567682f, + 1.6434668808089992f }; static const float *const av1_hd_partition_breakout_nn_mean[5] = { @@ -2693,418 +2695,397 @@ static const float av1_hd_partition_breakout_nn_weights_128_layer0[FEATURE_SIZE * 32] = { - 0.4625445008277893f, 0.046699944883584976f, - 0.01686757430434227f, -0.20717839896678925f, - 3.7204904556274414f, -0.09939002245664597f, - -0.03257349878549576f, -0.02249263785779476f, - 4.27296257019043f, -0.2201482355594635f, - -0.02630791626870632f, -0.8290131092071533f, - 4.106664657592773f, -0.03872422128915787f, - -0.0008374569588340819f, 0.008610394783318043f, - 2.8612468242645264f, 0.029475420713424683f, - 0.01878979615867138f, -0.07678258419036865f, - 2.642153024673462f, -0.7023047208786011f, - -0.16276267170906067f, -1.7577937841415405f, - 3.9672751426696777f, -0.019311338663101196f, - -0.00475256098434329f, 7.173077028710395e-05f, - -0.06531551480293274f, -0.07583651691675186f, - -0.02518392540514469f, 0.005303085781633854f, - 4.6276326179504395f, -0.4294697642326355f, - -0.06581428647041321f, -0.4343421459197998f, - -3.5704429149627686f, -0.3216317892074585f, - 0.2882029414176941f, 0.6410080194473267f, - -3.4585084915161133f, 0.24044956266880035f, - 0.2667864263057709f, 0.35314950346946716f, - -0.9072761535644531f, 0.4799148142337799f, - -1.9291492700576782f, -0.14138458669185638f, - -4.956376075744629f, -0.4101237654685974f, - -0.00801194179803133f, 0.016341974958777428f, - 10.001052856445312f, -0.10566182434558868f, - -0.008026782423257828f, -0.04508136212825775f, - -0.11174708604812622f, -0.1031208261847496f, - 0.004450094420462847f, -0.0714421346783638f, - -3.7611849308013916f, -0.42005446553230286f, - -0.39350515604019165f, 0.47153064608573914f, - -0.36057817935943604f, -2.4768900871276855f, - -0.0051008667796850204f, 0.588866114616394f, - 3.6189825534820557f, -0.018487676978111267f, - 0.033942464739084244f, 0.3527982831001282f, - 2.350560426712036f, -1.840644359588623f, - 0.1856888085603714f, -1.4500601291656494f, - -2.9771881103515625f, 0.15055160224437714f, - -0.7389538884162903f, -0.02373156137764454f, - 3.244478464126587f, 0.09777998179197311f, - 0.03143433853983879f, -0.3365783095359802f, - -3.579131841659546f, -0.3401176333427429f, - 0.3683990240097046f, 0.6233412027359009f, - -5.737433433532715f, -1.0643895864486694f, - -0.0862712413072586f, -0.8468044400215149f, - 4.366209983825684f, -0.03896212950348854f, - -0.005277127958834171f, -0.054806072264909744f, - 4.20292854309082f, -0.33671531081199646f, - -0.04025593772530556f, -0.4954434335231781f, - 1.4843676090240479f, -0.05833450332283974f, - -1.2679489850997925f, 0.526577353477478f, - 4.186404705047607f, -0.019188573583960533f, - 0.03789565712213516f, 0.35230082273483276f, - 3.3375051021575928f, -0.01609140820801258f, - 0.027185490354895592f, 0.25255802273750305f, - 4.186381816864014f, -0.10051476955413818f, - -0.03126750886440277f, -0.02391117438673973f, - 3.763073205947876f, -0.04791776463389397f, - -2.3893446268630214e-05f, 0.010260583832859993f, - -2.701782464981079f, -0.3117542862892151f, - -0.08752211928367615f, 0.4682832360267639f, - 2.9983580112457275f, -0.47988685965538025f, - 0.2418350726366043f, 0.451855331659317f + -0.24421754479408264f, 0.5826895833015442f, -0.468494713306427f, + 0.05104721337556839f, -0.3030756115913391f, -0.47995227575302124f, + 0.30220189690589905f, 0.6468079686164856f, -0.6581971049308777f, + -0.08561690151691437f, -0.25164133310317993f, -0.7251309752464294f, + 0.28622835874557495f, -0.3805813491344452f, -1.2038640975952148f, + -0.03289572894573212f, -0.37501949071884155f, -0.6697373390197754f, + 0.10718680173158646f, 0.39980190992355347f, -0.6406639218330383f, + 0.3059859871864319f, -0.04687154293060303f, -0.0283386018127203f, + 0.2230420708656311f, -0.024931879714131355f, 0.029338961467146873f, + -0.39070868492126465f, 0.12055488675832748f, 0.44709470868110657f, + -0.2240329086780548f, 0.3899381160736084f, 0.7343757748603821f, + 0.3804762363433838f, -0.6964248418807983f, 0.031393926590681076f, + 1.1329405307769775f, 0.2145451456308365f, 0.15080654621124268f, + -0.5583563446998596f, 0.7352061867713928f, 0.03883780539035797f, + -0.003559485776349902f, -0.6891534328460693f, -0.2841552793979645f, + -0.033437542617321014f, -0.05212170630693436f, 0.46065035462379456f, + 0.49060872197151184f, -1.0210745334625244f, -0.20526815950870514f, + 0.3765658140182495f, -0.050877176225185394f, -0.4010245203971863f, + 0.2002274990081787f, -0.45646876096725464f, 0.028925752267241478f, + 0.44919347763061523f, -0.02976205199956894f, -0.14882135391235352f, + -0.23057465255260468f, -0.1957021802663803f, -0.02549680881202221f, + -0.8748094439506531f, -0.48586851358413696f, -0.4607398211956024f, + -0.19540899991989136f, 0.857029139995575f, -0.37086406350135803f, + -0.15118056535720825f, 0.18724219501018524f, 0.9177047610282898f, + -0.04173488914966583f, -0.3510979413986206f, 0.17673315107822418f, + -0.4069298505783081f, 0.020244762301445007f, -0.6126715540885925f, + 0.6862805485725403f, 0.1272497922182083f, -0.4310610890388489f, + -0.5232776403427124f, -0.26814699172973633f, 0.8925990462303162f, + -0.12301289290189743f, -0.5783265829086304f, 0.6845996379852295f, + 0.22650295495986938f, -0.11178003996610641f, -0.2689964771270752f, + -1.2071754932403564f, 0.8008237481117249f, -0.07050541788339615f, + -0.572569727897644f, 0.27917924523353577f, -0.6794210076332092f, + 0.37107107043266296f, 0.04119817912578583f, 0.08820059895515442f, + -0.29868319630622864f, -0.8464934229850769f, -0.6945635080337524f, + -0.6215496063232422f, 0.08221497386693954f, -0.6280669569969177f, + -0.44894012808799744f, 0.5003432631492615f, 0.4358958899974823f, + -0.03372211381793022f, 0.523205041885376f, -0.03965454921126366f, + -0.13546960055828094f, -0.3377343714237213f, -0.3924517333507538f, + -0.1991700977087021f, 0.6752629280090332f, -1.0135334730148315f, + 0.5485906600952148f, -0.020843572914600372f, -0.1461324244737625f, + -0.7540525794029236f, 0.21933406591415405f, -0.5565986633300781f, + 0.1918906718492508f, -0.40834516286849976f, 0.9896414279937744f, + 0.11563636362552643f, -0.017263229936361313f }; static const float av1_hd_partition_breakout_nn_bias_128_layer0[32] = { - -0.09511213004589081f, 0.9938786625862122f, 1.4839892387390137f, - 1.0876423120498657f, 0.8019428253173828f, 0.2738432288169861f, - 1.071292519569397f, -0.09138987958431244f, 1.2080857753753662f, - -0.6339017748832703f, -0.4984428584575653f, -1.2238473892211914f, - -1.3475229740142822f, 2.4443564414978027f, -0.16558663547039032f, - 0.033367302268743515f, -0.7447078227996826f, 1.0401278734207153f, - -0.4657363295555115f, -1.0536733865737915f, 0.8565444946289062f, - -0.5986987948417664f, -0.789206862449646f, 1.1764090061187744f, - 1.3952962160110474f, -0.17746582627296448f, 1.187064290046692f, - 0.9370515942573547f, 1.1067490577697754f, 1.003738522529602f, - -2.1517810821533203f, 0.8461123704910278f + 0.2676561176776886f, -0.29510897397994995f, -1.1071470975875854f, + 0.583503007888794f, -0.472633957862854f, -0.44106626510620117f, + 0.6721175909042358f, -0.42975232005119324f, -0.5145384073257446f, + 0.15512774884700775f, 1.0278737545013428f, -0.6500952839851379f, + 1.0237534046173096f, 0.5816664099693298f, 0.6633380055427551f, + -1.3322787284851074f, -0.4271796643733978f, 0.06402774155139923f, + 0.5194029808044434f, 0.03623528033494949f, -0.34109270572662354f, + -0.15831837058067322f, -0.07824128866195679f, 0.8782597184181213f, + -0.014673121273517609f, -0.20494380593299866f, -0.3248092532157898f, + 0.6707151532173157f, -0.2768913507461548f, -0.31312206387519836f, + -0.6461104154586792f, 0.16326004266738892f }; static const float av1_hd_partition_breakout_nn_weights_128_layer1[32] = { - -0.00859279278665781f, -2.9574432373046875f, -4.146053314208984f, - -3.348489761352539f, -0.05143212527036667f, -2.0400538444519043f, - -4.622243404388428f, -0.05732434615492821f, -5.16105318069458f, - 1.1596481800079346f, 0.8911784291267395f, 8.114648818969727f, - 19.255064010620117f, 11.491721153259277f, -0.2828827202320099f, - 1.5688998699188232f, 0.7430379986763f, -1.0839532613754272f, - -1.3526417016983032f, 5.244595527648926f, -4.02890682220459f, - 0.8462737798690796f, 5.2530364990234375f, -4.688608646392822f, - -4.285271644592285f, 0.3642231523990631f, -2.4636125564575195f, - -0.8168565034866333f, -6.247287273406982f, -1.9640626907348633f, - -2.2665669918060303f, -0.7841953039169312f + -0.31587469577789307f, 0.2444663941860199f, 1.1686184406280518f, + -1.136441707611084f, 0.462795227766037f, 0.16067592799663544f, + -0.08983717858791351f, -0.1526145339012146f, 1.2807552814483643f, + -1.8411308526992798f, -2.5061304569244385f, -0.33663204312324524f, + -1.3827414512634277f, -0.2779307961463928f, -0.47907790541648865f, + 1.9458348751068115f, 0.3252658247947693f, 0.6043151617050171f, + -0.2127833366394043f, 0.3176756501197815f, 0.9920678734779358f, + 0.49299460649490356f, 0.8988507390022278f, -0.7701797485351562f, + 0.29247382283210754f, 0.7251673340797424f, 0.460658997297287f, + -0.3683358430862427f, 0.6006630659103394f, 0.7680628299713135f, + 0.8999216556549072f, -0.529272735118866f }; static const float av1_hd_partition_breakout_nn_bias_128_layer1[1] = { - 0.1112925186753273f + -0.22755585610866547f }; static const float av1_hd_partition_breakout_nn_weights_64_layer0[FEATURE_SIZE * 32] = { - -2.9600589275360107f, 0.042996086180210114f, -0.5422487258911133f, - -0.16088658571243286f, 2.665553331375122f, 0.17293912172317505f, - -0.011884346604347229f, -1.3571131229400635f, -4.151463508605957f, - -0.09992470592260361f, -0.03328946977853775f, -0.19198642671108246f, - -4.485296726226807f, 0.0004126991261728108f, 0.00528523325920105f, - 0.0014443921390920877f, 3.302469253540039f, -0.054684557020664215f, - 0.0016338867135345936f, -2.7690939903259277f, 2.5339231491088867f, - 0.5221245884895325f, -0.17638295888900757f, -0.31949788331985474f, - 2.323598623275757f, 0.2059188336133957f, -0.01752222701907158f, - -1.1489571332931519f, 2.984546422958374f, -0.19020183384418488f, - 0.010538210161030293f, -1.4752967357635498f, 1.8535363674163818f, - 0.8166176080703735f, -0.5628044009208679f, -1.9997116327285767f, - -2.098487377166748f, 0.3857003152370453f, -1.727634310722351f, - -0.15402595698833466f, 3.4408957958221436f, -0.2683435082435608f, - 0.023594189435243607f, 0.31177008152008057f, 3.039790391921997f, - 0.4344753623008728f, 0.00961180217564106f, -0.45155903697013855f, - -2.201700210571289f, -2.646042585372925f, 0.01605065166950226f, - 0.13390818238258362f, -3.6872377395629883f, 0.7149689793586731f, - -0.04385269060730934f, -0.10101474821567535f, 2.27207088470459f, - 0.34321531653404236f, 0.04720620810985565f, -0.7443615198135376f, - -5.059787750244141f, 0.02228701300919056f, -0.0019030789844691753f, - 0.0005637781578116119f, 3.129986524581909f, 0.09729813039302826f, - -0.024177810177206993f, -1.8108325004577637f, 3.3083548545837402f, - -0.7547611594200134f, -0.44618573784828186f, -0.21931181848049164f, - -2.2004897594451904f, -2.6458916664123535f, 0.09130378812551498f, - 0.2665710747241974f, -0.5643004775047302f, -0.33926868438720703f, - -0.6195669174194336f, 0.7023015022277832f, -5.160871982574463f, - -0.0034803124144673347f, -0.011646432802081108f, 0.012036978267133236f, - -3.1459012031555176f, -1.4449714422225952f, -0.0737914890050888f, - 0.6170333623886108f, 2.4652867317199707f, -0.05448657274246216f, - -0.16370250284671783f, -0.6195785999298096f, -4.647854328155518f, - -0.00691827479749918f, 0.038473278284072876f, 0.032320134341716766f, - -4.32373571395874f, 0.004058001562952995f, -0.004842042922973633f, - -0.011608919128775597f, 2.67899489402771f, -0.21639853715896606f, - -0.004101857542991638f, -2.9161124229431152f, 4.8421549797058105f, - 0.04978768154978752f, -0.04711421951651573f, -0.06902165710926056f, - -0.16757740080356598f, 0.021471548825502396f, -0.09624961763620377f, - -0.05160382017493248f, -4.769743919372559f, 0.1121208518743515f, - -0.024247704073786736f, -0.20753318071365356f, 0.47400805354118347f, - 0.46464693546295166f, -2.335646390914917f, -0.00663331151008606f, - -1.8362538814544678f, -1.3952279090881348f, -0.0224999338388443f, - 0.9369595050811768f, -0.32786062359809875f, -1.2076143026351929f, - -0.4175379276275635f, 0.920979917049408f + 0.4476310610771179f, -0.8712111711502075f, 0.1288183629512787f, + 0.792915940284729f, -0.37857282161712646f, -0.5999379754066467f, + -0.09074056148529053f, 0.7428027987480164f, -0.806922197341919f, + -0.06513206660747528f, 0.0015326066641137004f, 0.03463340178132057f, + -0.45006656646728516f, 0.8719941973686218f, -0.3037683367729187f, + -0.4873082637786865f, 0.3746185898780823f, -0.35810524225234985f, + 0.16044893860816956f, -0.5600897073745728f, 0.7878753542900085f, + -0.057914119213819504f, -0.05587376281619072f, -0.32516810297966003f, + -0.054352328181266785f, 0.7050285339355469f, 0.4860455393791199f, + -0.26461395621299744f, 0.4708733856678009f, 0.6615039110183716f, + -0.6717810034751892f, 0.25766628980636597f, -0.20877176523208618f, + 0.4233146607875824f, -0.8261363506317139f, 0.6218927502632141f, + -0.7742594480514526f, -0.09597370028495789f, -0.016565624624490738f, + 0.3243916630744934f, 0.730843722820282f, 0.3434459865093231f, + -0.6115328073501587f, 0.05172838270664215f, -0.8085055947303772f, + -0.23082391917705536f, 0.25912967324256897f, 0.48390132188796997f, + -0.24071218073368073f, -0.21271266043186188f, 0.15473750233650208f, + 1.1897283792495728f, -0.605708122253418f, 0.8830835223197937f, + -0.8744847774505615f, -0.10007809847593307f, 0.14527341723442078f, + -0.3217686712741852f, -0.6968852281570435f, -0.12154795229434967f, + -0.8508146405220032f, -0.23319056630134583f, 0.27581697702407837f, + 0.49674510955810547f, 0.2984752655029297f, -0.24128034710884094f, + -0.6872956156730652f, 0.6757507920265198f, -0.9962611198425293f, + -0.24671325087547302f, 0.04458463191986084f, 0.5092823505401611f, + -0.755521833896637f, 0.1556125432252884f, -0.19932909309864044f, + 0.8233707547187805f, -0.33803892135620117f, 0.1641087830066681f, + -0.1640886515378952f, 0.2807005047798157f, 0.11740092933177948f, + 0.9441748261451721f, -0.19026336073875427f, 0.1905575692653656f, + -0.7156860828399658f, -0.30425164103507996f, -0.6542183756828308f, + -0.048906467854976654f, 0.6393331289291382f, -0.5186219811439514f, + -0.15262047946453094f, 0.729802668094635f, 0.4545252323150635f, + -0.5133960843086243f, 0.26521825790405273f, -0.4991665482521057f, + -0.4927463233470917f, -0.07089114189147949f, 0.33401384949684143f, + 0.3023838400840759f, 0.7638855576515198f, -0.4981137812137604f, + -0.5541877150535583f, -0.11165370792150497f, 0.3648005425930023f, + 0.5878415107727051f, 0.11315485090017319f, -1.0285885334014893f, + -0.45433279871940613f, -0.7583989500999451f, -0.14650657773017883f, + 0.24613532423973083f, -0.2773364782333374f, -0.9591728448867798f, + 0.5280625224113464f, 0.5242233276367188f, -0.39398059248924255f, + -0.8652032613754272f, 0.5243239998817444f, 0.5806688666343689f, + 0.42181527614593506f, -0.09341983497142792f, 0.12693725526332855f, + -0.39904987812042236f, 0.15719516575336456f, 0.16289645433425903f, + 0.20234838128089905f, -0.25414201617240906f }; static const float av1_hd_partition_breakout_nn_bias_64_layer0[32] = { - -0.9125985503196716f, 0.9354565739631653f, -0.9121711850166321f, - -1.1538583040237427f, 0.08968774974346161f, 1.1123112440109253f, - 1.4973655939102173f, 0.608467698097229f, 0.18589675426483154f, - -1.441516399383545f, 0.9556304812431335f, 0.7926673889160156f, - -1.4284138679504395f, -0.7179064750671387f, -0.047664791345596313f, - -1.2288035154342651f, 0.6012410521507263f, 0.18902179598808289f, - -0.89959716796875f, 0.23802867531776428f, -1.300124168395996f, - -0.08421888202428818f, 0.28658464550971985f, -0.27196547389030457f, - -1.0962705612182617f, -0.350778728723526f, 0.8084302544593811f, - -0.11797107756137848f, -0.9147244691848755f, -0.9423892498016357f, - -0.46693262457847595f, -1.179872989654541f + -0.34709370136260986f, -0.009843609295785427f, -1.6221565008163452f, + -0.668915867805481f, 0.9886232018470764f, 0.6522868275642395f, + -0.37716832756996155f, -0.07447462528944016f, 0.3766818642616272f, + -1.1512787342071533f, 0.40794867277145386f, -0.14982998371124268f, + -0.19331441819667816f, -0.9201817512512207f, 0.48975181579589844f, + -0.137226402759552f, -0.5480761528015137f, -0.14310558140277863f, + -0.035678427666425705f, -0.724799633026123f, 0.6203687787055969f, + -0.26913538575172424f, -0.45061108469963074f, 1.1059123277664185f, + -0.9563385844230652f, 0.40938442945480347f, 0.4411890506744385f, + 0.1871682107448578f, 0.004259683191776276f, -0.050544340163469315f, + 0.789713442325592f, 0.26867425441741943f }; static const float av1_hd_partition_breakout_nn_weights_64_layer1[32] = { - 1.888397216796875f, -1.6998655796051025f, 2.5516481399536133f, - 31.333364486694336f, -7.420342922210693f, -1.175126552581787f, - -2.0703518390655518f, -1.0818297863006592f, -1.0498319864273071f, - 12.22414779663086f, -0.6940112113952637f, -0.3731234669685364f, - 11.562948226928711f, 0.6934157013893127f, 1.0499767065048218f, - 15.071033477783203f, -2.1736562252044678f, 0.47170618176460266f, - 1.8899933099746704f, 0.2540336847305298f, 27.493698120117188f, - 0.8192774057388306f, -0.011624461971223354f, 1.5983129739761353f, - 34.282840728759766f, -15.804471969604492f, 1.1542948484420776f, - -0.09561372548341751f, 4.60494327545166f, 1.2926712036132812f, - 0.4446795880794525f, -0.7026389241218567f + 0.7450170516967773f, 0.1674197018146515f, 8.603666305541992f, + -1.4572426080703735f, -0.6806990504264832f, -0.36511117219924927f, + -0.21973447501659393f, -0.33904269337654114f, 0.494196355342865f, + 2.428866386413574f, -0.33053985238075256f, 0.5259426236152649f, + 0.7364671230316162f, 1.8298554420471191f, -0.20149026811122894f, + 0.4657254219055176f, 0.9128966331481934f, 0.41023728251457214f, + 0.4559532403945923f, -0.49007904529571533f, -0.6186080574989319f, + 0.6019283533096313f, 0.41913628578186035f, -0.9652310609817505f, + -1.266809105873108f, -0.6605119109153748f, -1.0554698705673218f, + 0.28501376509666443f, 0.40640202164649963f, 0.4747793972492218f, + -0.3028396964073181f, -0.020081687718629837f }; static const float av1_hd_partition_breakout_nn_bias_64_layer1[1] = { - -0.15468640625476837f + -0.35164690017700195f }; static const float av1_hd_partition_breakout_nn_weights_32_layer0[FEATURE_SIZE * 32] = { - -0.18351112306118011f, -0.15651831030845642f, 0.0013713851803913713f, - 0.01054045557975769f, -3.780832529067993f, -0.15531419217586517f, - 0.03428387641906738f, -0.027534646913409233f, -4.079058647155762f, - 0.05176529288291931f, -0.0065496210008859634f, -0.08574667572975159f, - -1.9617663621902466f, 0.3167760670185089f, -0.33277764916419983f, - 0.7234531044960022f, 2.2539572715759277f, -0.02695995196700096f, - 0.005501582287251949f, -3.854844093322754f, -2.5093464851379395f, - -1.8817286491394043f, -0.13509777188301086f, 0.8389045000076294f, - -0.8123356103897095f, -2.210233449935913f, 0.04169511795043945f, - 1.3030999898910522f, 1.0332109928131104f, 0.15076060593128204f, - -0.02417255938053131f, -3.10256290435791f, -3.5063133239746094f, - 0.1668155938386917f, -0.11349064111709595f, -0.08673816174268723f, - 0.2023429572582245f, 0.2143218219280243f, -2.2571842670440674f, - 0.1890597939491272f, -2.0899853706359863f, -0.5377482771873474f, - -0.24231456220149994f, 0.7975266575813293f, -1.104709267616272f, - 0.42163029313087463f, -1.588227391242981f, -0.5404914021492004f, - 0.39026254415512085f, 0.3967314660549164f, -2.0958690643310547f, - -0.1487407088279724f, -1.2271692752838135f, -2.4342153072357178f, - 0.0073483348824083805f, 0.4332350790500641f, 1.8399561643600464f, - 0.20915934443473816f, 0.014152542687952518f, -2.279658079147339f, - 2.1412830352783203f, -1.2423213720321655f, 0.10377844423055649f, - 0.0026721248868852854f, -1.515641212463379f, 0.011470284312963486f, - 0.433901309967041f, 0.03397373855113983f, 1.9424909353256226f, - 0.007673368323594332f, -0.019963057711720467f, -3.2635231018066406f, - -3.814363956451416f, -0.008298925124108791f, -0.0006046147318556905f, - -0.03078455477952957f, -2.5249812602996826f, -1.5319974422454834f, - 0.03523765504360199f, 0.47840240597724915f, -2.459254503250122f, - -0.060729049146175385f, 0.3428601920604706f, 1.0676825046539307f, - -3.196352481842041f, 0.00556157436221838f, -0.014416366815567017f, - 0.00013636458606924862f, 0.5342910289764404f, 0.08967946469783783f, - -1.079426884651184f, 0.0459836907684803f, 1.169906735420227f, - 0.23663409054279327f, 0.15936784446239471f, 0.12904685735702515f, - -1.941832423210144f, -2.3517515659332275f, -0.4566086530685425f, - 1.001920223236084f, -2.479144334793091f, -1.4872301816940308f, - 0.44298961758613586f, 0.0910615548491478f, 0.34078770875930786f, - -1.0173975229263306f, 0.2069491446018219f, 0.8345462679862976f, - 1.5491070747375488f, 0.20149222016334534f, 0.2451339066028595f, - -1.176571011543274f, -2.3429925441741943f, 0.2398013025522232f, - -0.9587415456771851f, -0.05250968039035797f, -3.1348395347595215f, - -0.55987948179245f, -0.040084246546030045f, 0.29037174582481384f, - 1.663692593574524f, 0.6011162400245667f, 0.014114673249423504f, - -0.9607003331184387f, -1.0632480382919312f, -0.9293855428695679f, - -0.7134267091751099f, 0.9321397542953491f + -0.8952129483222961f, -0.23612700402736664f, -0.022868705913424492f, + 0.504967451095581f, 0.3275434970855713f, 1.1059952974319458f, + -0.34120088815689087f, -0.07344819605350494f, 0.15775921940803528f, + -0.04879460483789444f, 0.1439337134361267f, 1.2064639329910278f, + 0.006475068163126707f, -0.4494408369064331f, -0.10751494765281677f, + 1.1100929975509644f, 0.12578976154327393f, 0.4349801242351532f, + 0.5385650396347046f, -0.4146132171154022f, -0.6666572690010071f, + 0.1782642900943756f, -0.43637481331825256f, -0.522347629070282f, + -0.7048850655555725f, -0.21371017396450043f, 0.1338515281677246f, + 0.9378389716148376f, -0.9088003635406494f, 0.3172213137149811f, + -0.6936320066452026f, 0.6413770318031311f, -0.6204543113708496f, + -0.20075663924217224f, 0.011177961714565754f, 0.3956983685493469f, + -0.23885823786258698f, 0.7718412280082703f, 0.013668450526893139f, + -0.11212749779224396f, -0.3808520436286926f, -0.019013067707419395f, + -0.5678153038024902f, 1.038125991821289f, -0.0636964738368988f, + -0.9305693507194519f, 0.5443329811096191f, 0.5650880932807922f, + -0.2772569954395294f, -0.7757272720336914f, 0.02383701130747795f, + 0.5687315464019775f, -0.4549035429954529f, -0.16180022060871124f, + 0.033420149236917496f, 0.5288522839546204f, -0.3871588110923767f, + 0.712287962436676f, 0.008839298039674759f, -0.4081243574619293f, + -0.44496551156044006f, -0.49737873673439026f, 0.5079053044319153f, + 0.4881437122821808f, 0.6699126362800598f, 0.28043776750564575f, + -0.5355777740478516f, 0.2881544232368469f, 0.33460190892219543f, + -0.26208120584487915f, -0.44822946190834045f, -0.733630895614624f, + -0.35417506098747253f, 0.6161319613456726f, -0.0993569940328598f, + -0.7897841334342957f, -0.551094651222229f, 0.4497978985309601f, + -0.31630903482437134f, 0.46718651056289673f, -0.19858941435813904f, + 1.0053157806396484f, -0.3194795548915863f, 0.31316739320755005f, + -0.035107459872961044f, -0.7919036746025085f, 0.04070507362484932f, + -0.325952410697937f, -0.8687034845352173f, -0.30160313844680786f, + 0.16643138229846954f, 1.1927493810653687f, -0.5958532094955444f, + -0.38102859258651733f, -0.06056114658713341f, 1.063319206237793f, + -0.028595328330993652f, 0.7656055688858032f, 0.5335075855255127f, + -0.6592297554016113f, -0.3134363293647766f, -0.8477844595909119f, + 0.14632871747016907f, 0.5351374745368958f, -0.007659543305635452f, + -0.8012065291404724f, 0.47477757930755615f, 0.4271889626979828f, + -0.6191681623458862f, -0.14256015419960022f, 0.038999997079372406f, + 0.4897593557834625f, -0.713534414768219f, 0.07782091945409775f, + -0.28364527225494385f, -0.17344607412815094f, -0.24871549010276794f, + 0.6229367852210999f, 0.35328975319862366f, -0.48995962738990784f, + 0.9903460741043091f, -0.1367904245853424f, -0.14807336032390594f, + -0.28647252917289734f, 0.6976669430732727f, -0.5536572933197021f, + 0.3724263310432434f, -0.7798140645027161f }; static const float av1_hd_partition_breakout_nn_bias_32_layer0[32] = { - -0.11753689497709274f, -0.7140882015228271f, -0.9722395539283752f, - -0.6984077095985413f, -0.726676881313324f, -0.36886507272720337f, - -0.17676463723182678f, -0.10097962617874146f, -0.29938116669654846f, - -1.727449893951416f, 0.5804333090782166f, -1.2764358520507812f, - -1.108852505683899f, -0.8272318840026855f, 0.6785213947296143f, - 0.11738266795873642f, -2.057903528213501f, -0.37823784351348877f, - -0.9595664143562317f, -1.0248923301696777f, -0.23907040059566498f, - -0.8479264974594116f, -0.13894951343536377f, -1.7720012664794922f, - -1.6527719497680664f, -0.20360442996025085f, -1.7053756713867188f, - 0.8720151782035828f, -1.0570132732391357f, -0.7993026971817017f, - 0.46637022495269775f, -0.8374565243721008f + 0.5906000733375549f, 0.8543934226036072f, 0.43217557668685913f, + -0.643294632434845f, 0.30037060379981995f, -0.7660125494003296f, + 0.1731051206588745f, -0.7480033040046692f, -0.6353603005409241f, + 0.4871520698070526f, 0.7321553230285645f, 0.2810375988483429f, + -0.43400588631629944f, -0.17055644094944f, -0.8143026232719421f, + 0.13951803743839264f, -1.2939090728759766f, -0.5045616626739502f, + -1.295732021331787f, -0.9267586469650269f, -0.13748344779014587f, + -0.8858642578125f, 0.24190591275691986f, 0.34041425585746765f, + 0.05495613068342209f, 0.25397151708602905f, 0.24138814210891724f, + 0.12242670357227325f, -0.5556198358535767f, -0.46377164125442505f, + 0.6218398213386536f, 0.7764495015144348f }; static const float av1_hd_partition_breakout_nn_weights_32_layer1[32] = { - -0.329619824886322f, 4.491082191467285f, 11.859853744506836f, - -0.2909659743309021f, -20.884370803833008f, 1.2395148277282715f, - 0.7091589570045471f, -2.458725690841675f, 1.714850664138794f, - 2.3455934524536133f, 0.7239453792572021f, 11.537474632263184f, - 3.062530755996704f, 2.7946059703826904f, -1.4265711307525635f, - 0.44554367661476135f, -1.534838080406189f, -4.844552993774414f, - 18.15884780883789f, 5.826704978942871f, 0.5580787062644958f, - 11.919114112854004f, 0.24757426977157593f, 0.30521970987319946f, - -1.6404640674591064f, 0.848618745803833f, -0.8315955996513367f, - -0.8414434194564819f, 6.57717227935791f, 2.829108476638794f, - -0.25260433554649353f, -0.4767799377441406f + 0.6254786252975464f, -0.724627673625946f, 0.5428725481033325f, + 0.8609197735786438f, -0.33933258056640625f, 0.6795636415481567f, + 0.5508725643157959f, 1.4960711002349854f, 0.8083735108375549f, + -0.3456595838069916f, 0.5514245629310608f, 0.28833192586898804f, + 0.7953433394432068f, -0.009960951283574104f, -1.7862937450408936f, + 0.1605624407529831f, 1.6885508298873901f, -0.49641919136047363f, + -4.162553310394287f, -2.510878086090088f, 0.3701621890068054f, + -0.5248422622680664f, 0.5657362937927246f, 0.48716917634010315f, + -0.23934587836265564f, 0.45801791548728943f, 0.3940700590610504f, + 0.07297442853450775f, 0.287155419588089f, -0.0904238149523735f, + -0.608720064163208f, -0.7513124942779541f }; static const float av1_hd_partition_breakout_nn_bias_32_layer1[1] = { - 0.04493425786495209f + 0.1117519736289978f }; static const float av1_hd_partition_breakout_nn_weights_16_layer0[FEATURE_SIZE * 32] = { - -0.03770807012915611f, 0.1042923852801323f, 0.3589613735675812f, - 0.8915500044822693f, -0.025765806436538696f, -0.8080071806907654f, - 0.11935761570930481f, 1.081723690032959f, -1.0217446088790894f, - -0.05894668400287628f, -0.0417763888835907f, 1.9492236375808716f, - -1.6240522861480713f, -1.0665417909622192f, -0.4230113625526428f, - 2.6151089668273926f, -1.0250322818756104f, -1.013023853302002f, - -0.3417854607105255f, 1.2021280527114868f, -2.932366132736206f, - -1.5600024461746216f, 0.013506319373846054f, 1.827286958694458f, - 0.5162537097930908f, 0.37939924001693726f, -0.12543700635433197f, - -3.9469857215881348f, -0.5308187007904053f, 0.055751170963048935f, - -2.1356542110443115f, 0.9524611830711365f, -1.2815498113632202f, - -0.2583577334880829f, 0.5526014566421509f, -0.991321861743927f, - 0.4657832682132721f, -0.32342609763145447f, 0.012717160396277905f, - -4.607941150665283f, -0.7786141633987427f, -0.7440438270568848f, - -0.10956363379955292f, 1.9957225322723389f, -3.557837724685669f, - -0.06580297648906708f, -0.019815972074866295f, 0.48607689142227173f, - 0.9143388271331787f, 0.7489029765129089f, 0.23694343864917755f, - -2.3731346130371094f, -2.455751895904541f, 0.02998301573097706f, - -1.352641224861145f, 1.4107884168624878f, -2.7167775630950928f, - -1.3195676803588867f, -0.0024603719357401133f, 1.568859338760376f, - -1.368638277053833f, -0.6318024396896362f, -0.8460041284561157f, - 0.7431659698486328f, -0.17625141143798828f, -0.25151070952415466f, - 0.19983969628810883f, 2.0860755443573f, -3.904813289642334f, - -0.38988929986953735f, -0.002895161509513855f, 0.714531660079956f, - -3.0156004428863525f, -0.6497556567192078f, 0.32678696513175964f, - 0.5911594033241272f, -1.1386737823486328f, -0.6372389793395996f, - -0.029834266752004623f, 2.0517866611480713f, -1.0608230829238892f, - -0.955141544342041f, -0.10072463750839233f, 1.5374855995178223f, - -1.2410826683044434f, -2.169691562652588f, 0.003649030579254031f, - 2.0521841049194336f, -0.9012426733970642f, 0.6359571218490601f, - 0.10054442286491394f, 0.9355012774467468f, -2.433013439178467f, - -1.3059970140457153f, -0.062273263931274414f, 1.2142263650894165f, - 0.07180827856063843f, -0.02799983322620392f, 0.012213055975735188f, - -0.15036505460739136f, -1.5302343368530273f, -0.9248730540275574f, - -1.2439353466033936f, 1.6203926801681519f, -0.26789119839668274f, - 0.29344648122787476f, -0.12080803513526917f, 2.095102548599243f, - -3.6576807498931885f, 0.32156985998153687f, -0.00601181061938405f, - 0.3735950291156769f, -1.2174468040466309f, -0.21403838694095612f, - -2.081902503967285f, 1.0142937898635864f, -1.2942562103271484f, - -2.4739151000976562f, 0.11531462520360947f, 1.7982516288757324f, - -2.203162670135498f, 0.1353112757205963f, -1.742370843887329f, - 0.44314906001091003f, -1.1483821868896484f, -0.6968573331832886f, - -0.059433888643980026f, 0.9399102330207825f + 0.3427221179008484f, 0.06073892489075661f, -0.06882427632808685f, + 1.1336252689361572f, 0.097809799015522f, -0.6385823488235474f, + 0.43692731857299805f, -0.6029260754585266f, -0.13368098437786102f, + -0.6389026045799255f, 0.2193300873041153f, 0.6348177790641785f, + 0.22468054294586182f, 0.13958360254764557f, -0.05150606855750084f, + -0.28101637959480286f, -0.4977607727050781f, -0.3215360641479492f, + -0.16947278380393982f, 1.0890949964523315f, -1.02256441116333f, + 0.34789103269577026f, 0.34546756744384766f, -0.08185210078954697f, + 0.24066677689552307f, -1.2535117864608765f, 0.04398220032453537f, + 0.8025804162025452f, -0.2541569769382477f, 0.3501926362514496f, + -0.3649175763130188f, -0.2356642484664917f, 0.8068639039993286f, + -0.23699556291103363f, -0.2195892035961151f, -0.1547853797674179f, + -0.15287409722805023f, 0.4898677468299866f, -0.3860497772693634f, + 0.32813069224357605f, -0.7604209184646606f, -0.6611303091049194f, + 0.006376428063958883f, 1.0097582340240479f, 0.18538667261600494f, + 0.5587815642356873f, 0.30065974593162537f, -0.8898136615753174f, + 0.63029944896698f, 0.839385986328125f, -0.2562950849533081f, + -0.5478677749633789f, -0.7537129521369934f, 0.019417811185121536f, + 0.23349952697753906f, 0.07770441472530365f, 0.0019170257728546858f, + 0.003552661743015051f, 0.012630019336938858f, 0.03414732217788696f, + -0.04959012567996979f, -0.21843530237674713f, -0.5894723534584045f, + 0.21037738025188446f, -0.4222605526447296f, -0.5836210250854492f, + 0.011957126669585705f, 0.0031051053665578365f, -0.5127609968185425f, + -0.7208832502365112f, 0.05614267289638519f, 1.1000267267227173f, + -0.7031041383743286f, -0.20904500782489777f, 0.2827422320842743f, + 0.5169914960861206f, -0.03972400724887848f, -0.24878446757793427f, + -0.6129308938980103f, 0.3691819906234741f, -0.5902302861213684f, + -0.44936394691467285f, -0.20858828723430634f, -0.1926502138376236f, + 0.015391144901514053f, -0.05831986665725708f, 0.0012774848146364093f, + 0.9318875670433044f, -0.07232701778411865f, -0.07282842695713043f, + -0.3941495716571808f, 0.6986988186836243f, -0.6321771144866943f, + -0.1466921716928482f, -0.18214966356754303f, -0.2929930090904236f, + -0.048453543335199356f, 0.5302963852882385f, 0.3090183138847351f, + -0.7714007496833801f, -0.03153527155518532f, -0.28210026025772095f, + -0.0035391938872635365f, 1.1437654495239258f, 0.3694542348384857f, + -0.8283377885818481f, -0.06223354861140251f, 0.7067912817001343f, + -0.5538484454154968f, -0.3097091019153595f, 0.24007144570350647f, + 0.5142718553543091f, 0.5205507278442383f, -0.7828086018562317f, + 0.02196805737912655f, -0.5540323853492737f, -0.5541141629219055f, + -0.5407395958900452f, 0.83812016248703f, 0.5495181679725647f, + -0.36925947666168213f, -0.5186708569526672f, 0.035871513187885284f, + 0.9290533661842346f, 0.7525129914283752f, 0.16506646573543549f, + 0.015168293379247189f, -0.22260211408138275f }; static const float av1_hd_partition_breakout_nn_bias_16_layer0[32] = { - -2.500817060470581f, -1.1564264297485352f, 0.2977291941642761f, - -0.255332350730896f, -0.44481080770492554f, -0.6241381168365479f, - -0.4780181348323822f, -0.6444894075393677f, -1.2649517059326172f, - -1.3098241090774536f, 0.10575855523347855f, -0.6572186946868896f, - -0.13408327102661133f, -2.2591185569763184f, -0.6009128093719482f, - -0.9144248962402344f, 0.5553908944129944f, -0.40132245421409607f, - 0.3925309181213379f, 0.9891119003295898f, 0.7784745097160339f, - -0.47138360142707825f, -1.2775310277938843f, -0.3515026867389679f, - -0.4055931568145752f, -0.4989534318447113f, 0.7892708778381348f, - -0.5805770754814148f, -1.247983694076538f, -0.46460986137390137f, - -1.3504952192306519f, -0.29709282517433167f + 0.3051907420158386f, -0.7940329313278198f, 0.37281763553619385f, + -0.47349175810813904f, 0.9813013076782227f, -0.6225157976150513f, + -0.3041200339794159f, 0.18893377482891083f, -0.04121162369847298f, + 0.0799027606844902f, -0.06257568299770355f, 0.004709158092737198f, + 0.4165017306804657f, 0.40799248218536377f, -0.1605808585882187f, + 0.4690406322479248f, 0.2635003328323364f, 0.37693503499031067f, + 0.509822428226471f, 0.6932690739631653f, 0.3800438344478607f, + -0.9304105639457703f, 0.688030481338501f, -1.478393316268921f, + -0.08685281872749329f, -0.7200548648834229f, -0.9290581345558167f, + 0.47607457637786865f, -0.2131124883890152f, 0.4063669741153717f, + 0.6758500337600708f, 0.4055664539337158f }; static const float av1_hd_partition_breakout_nn_weights_16_layer1[32] = { - 0.37168681621551514f, 1.6819430589675903f, -0.24184460937976837f, - -0.5973770022392273f, -0.9343476295471191f, 9.948596000671387f, - -2.4088406562805176f, 0.7863813638687134f, -0.4910288453102112f, - -8.380901336669922f, -0.41362130641937256f, 5.82503604888916f, - -0.4502200186252594f, -1.826108694076538f, 4.183648109436035f, - 2.714763879776001f, -0.26116943359375f, 2.803555727005005f, - 0.8531390428543091f, 1.0828324556350708f, 0.8189658522605896f, - -2.4576377868652344f, -0.3627878427505493f, 2.138772487640381f, - -0.016609186306595802f, 0.7119042873382568f, 0.5726228356361389f, - 1.508818507194519f, 7.78598690032959f, 2.531076669692993f, - 6.370460510253906f, -0.04699355363845825f + 0.4050608277320862f, -0.41333451867103577f, 0.17236362397670746f, + 0.024612678214907646f, 0.9329894781112671f, -0.838448166847229f, + -1.4745935201644897f, 0.09989786148071289f, -0.10163675248622894f, + 0.17781522870063782f, 2.2620022296905518f, -0.3684936761856079f, + -0.6526914834976196f, 0.2942826449871063f, -0.35902100801467896f, + 0.10697637498378754f, 0.20386190712451935f, 1.2234491109848022f, + 0.30403849482536316f, 0.35678544640541077f, 0.29488465189933777f, + 1.736344814300537f, 0.5220925807952881f, -1.7380571365356445f, + -0.2642151713371277f, 0.7513654828071594f, -1.9644155502319336f, + 0.1906810700893402f, -0.35255730152130127f, 0.5716033577919006f, + 0.653854250907898f, -0.26972928643226624f }; static const float av1_hd_partition_breakout_nn_bias_16_layer1[1] = { - 0.42323046922683716f + 0.35929590463638306f }; static const float av1_hd_partition_breakout_nn_weights_8_layer0[FEATURE_SIZE * 32] = { - -0.9726589918136597f, -0.5411639213562012f, 0.7087171673774719f, - 1.7313759326934814f, -2.402184009552002f, -0.047024089843034744f, - -0.027457909658551216f, 1.6469149589538574f, 0.20576165616512299f, - -0.7170919179916382f, -0.10603219270706177f, 2.669275999069214f, - -0.2676953077316284f, -0.16688808798789978f, -0.03148927167057991f, - 1.4055346250534058f, -0.7267357110977173f, -1.3847031593322754f, - 0.21190257370471954f, -1.012730598449707f, -1.2875311374664307f, - -0.9335927963256836f, 0.9185367226600647f, -0.262455552816391f, - -1.8908239603042603f, -0.7330380082130432f, 0.580338716506958f, - 1.0240660905838013f, -0.741085946559906f, -0.4254375696182251f, - 1.0939886569976807f, 0.8454716205596924f, 0.028208844363689423f, - -1.108168363571167f, -0.5589184165000916f, 2.6788110733032227f, - -2.6662967205047607f, -0.7947397232055664f, 0.0377255417406559f, - 2.2486352920532227f, -0.6704809069633484f, 0.7823348641395569f, - 0.18206557631492615f, 1.8088916540145874f, -0.2850566804409027f, - -0.2228451520204544f, -0.0077590541914105415f, 0.267416387796402f, - -0.11488392949104309f, -0.3818961977958679f, 0.15274959802627563f, - 1.7191359996795654f, -0.1928352564573288f, 0.1540122777223587f, - -1.7266273498535156f, 0.16492529213428497f, -0.26917898654937744f, - 0.0453917570412159f, 0.006803665775805712f, -3.431412696838379f, - -0.8123732209205627f, 0.20750433206558228f, 0.043349288403987885f, - 1.5046347379684448f, -0.43123146891593933f, 0.496737003326416f, - 0.13186758756637573f, 1.348464012145996f, -1.0381258726119995f, - -0.14341726899147034f, -0.12307563424110413f, -0.47205984592437744f, - -0.26413795351982117f, -0.4825311005115509f, 0.01020186860114336f, - 1.8038893938064575f, 0.2135389894247055f, -0.5266213417053223f, - -0.34630000591278076f, 2.1957969665527344f, -2.6997740268707275f, - 0.22156886756420135f, 0.01723811775445938f, 0.9266343116760254f, - -0.29436543583869934f, -0.21920204162597656f, 0.050585292279720306f, - 1.8021506071090698f, 0.28632524609565735f, -1.0584778785705566f, - -0.004406697116792202f, 1.6804509162902832f, 0.1515306681394577f, - -0.5138083696365356f, -1.493039846420288f, 1.1335597038269043f, - -2.868360757827759f, -0.7500830888748169f, -0.018952427431941032f, - 1.7541661262512207f, -1.2600187063217163f, -0.19970472157001495f, - -0.027286890894174576f, -2.7354300022125244f, -1.349392056465149f, - -2.1341166496276855f, 0.1937991976737976f, -0.12994812428951263f, - -0.8666549921035767f, 0.3629917502403259f, -0.2922564446926117f, - 1.5711613893508911f, -1.5161727666854858f, -0.07508653402328491f, - -0.015387722291052341f, 1.1435129642486572f, 0.08286841958761215f, - -0.4614250361919403f, -0.26704689860343933f, 2.349350929260254f, - -0.16499075293540955f, 0.025024881586432457f, 0.001498897559940815f, - -0.04137025028467178f, -0.8304779529571533f, -0.20447227358818054f, - 0.009074501693248749f, 0.8690717816352844f + -1.1174798011779785f, -0.02540789172053337f, -0.024517325684428215f, + 0.9283695220947266f, 0.514511227607727f, 0.008708865381777287f, + -0.7743955254554749f, 0.03391582891345024f, -0.4907812476158142f, + 0.06405021250247955f, 0.009607836604118347f, -0.06413879245519638f, + -0.3100389838218689f, 0.027276739478111267f, 0.02754838392138481f, + 0.2741911709308624f, -0.33674532175064087f, -0.12841999530792236f, + -0.07583238184452057f, 1.2658021450042725f, -0.07618967443704605f, + 0.018406564369797707f, 0.36800676584243774f, 1.1263890266418457f, + 0.29222336411476135f, -0.1937945932149887f, -0.781943678855896f, + 0.08040471374988556f, 0.06932314485311508f, 0.27592363953590393f, + 0.6615497469902039f, -0.3753996193408966f, 0.04914139211177826f, + -0.18584918975830078f, 0.3851499855518341f, -0.14877784252166748f, + 0.8386304974555969f, -0.9132646322250366f, 0.09750116616487503f, + -0.38142505288124084f, -0.27932387590408325f, -0.3158399760723114f, + 0.3219115734100342f, -0.11211645603179932f, -0.1601802557706833f, + 0.199840247631073f, -0.1438094526529312f, 1.4131516218185425f, + -0.6507886648178101f, -0.35280969738960266f, -0.07980803400278091f, + 0.9108555912971497f, -0.49617353081703186f, -0.2692374289035797f, + 0.019175099208950996f, 0.5736552476882935f, -0.48202967643737793f, + -0.26134830713272095f, 0.018342891708016396f, 0.6154197454452515f, + -0.18230408430099487f, -0.8209198713302612f, 0.6161503195762634f, + 0.4757719039916992f, 0.5258967876434326f, -0.9027984142303467f, + -0.47966063022613525f, 0.6204502582550049f, -1.342071294784546f, + 0.12434060871601105f, 0.14531831443309784f, -0.15101315081119537f, + -0.35774746537208557f, 0.08184701204299927f, -0.04071677848696709f, + 1.2944822311401367f, -0.06492389738559723f, 0.34913739562034607f, + -0.08534345775842667f, 1.3283460140228271f, -0.7555822134017944f, + 0.3319704234600067f, -0.17644117772579193f, 0.5248327255249023f, + -0.46480831503868103f, -0.23956067860126495f, 0.0013285222230479121f, + 0.5432708859443665f, 0.30651840567588806f, 0.37756049633026123f, + -0.03282826766371727f, 0.06274658441543579f, -0.2130001038312912f, + 0.2907213270664215f, 0.09125600755214691f, -0.133547842502594f, + -0.05953352898359299f, 0.032438505440950394f, 0.05902266874909401f, + -0.043947484344244f, -1.100358486175537f, 0.2626442313194275f, + 0.056724611669778824f, -0.19415289163589478f, 0.010464449413120747f, + -0.385423481464386f, -0.29392293095588684f, 0.5806636214256287f, + -0.28265658020973206f, 1.0120196342468262f, 0.17220932245254517f, + -0.9805957078933716f, -0.368004709482193f, -0.42056915163993835f, + -0.1005094051361084f, -0.740735650062561f, 0.8810465335845947f, + -0.4035398066043854f, 0.06286423653364182f, -0.26255688071250916f, + -0.24484986066818237f, -0.1917300969362259f, 0.23190940916538239f, + -0.5081801414489746f, -0.909449577331543f, 0.6952412128448486f, + -0.41929101943969727f, -0.28803330659866333f }; static const float av1_hd_partition_breakout_nn_bias_8_layer0[32] = { - 0.7538588047027588f, -0.0891113132238388f, 0.6207568049430847f, - 0.1754276007413864f, -0.9016708731651306f, -0.5710541009902954f, - 0.3960225582122803f, -0.5215008854866028f, 0.1935194879770279f, - 0.1463070660829544f, -0.5508337020874023f, -0.25621265172958374f, - -0.21759319305419922f, -0.6702320575714111f, -0.510668158531189f, - -0.11305645108222961f, 0.9119726419448853f, 0.932460367679596f, - 0.5708784461021423f, 0.9256153702735901f, 0.4665280282497406f, - 0.2577301561832428f, 0.4606616199016571f, 0.4183802306652069f, - -0.11034739017486572f, -1.2057222127914429f, -1.5402069091796875f, - 0.36954033374786377f, -0.07251504063606262f, 0.4745430052280426f, - -0.4834096133708954f, -0.08527179062366486f + 0.9311190843582153f, -0.04600553959608078f, 0.4736453592777252f, + -0.21461233496665955f, -0.6837071776390076f, 0.41968896985054016f, + 0.5667760372161865f, -0.295064777135849f, 0.6580685973167419f, + -0.12489219009876251f, 0.6441367864608765f, 0.017892301082611084f, + 0.3467213213443756f, 0.24851299822330475f, 0.27420252561569214f, + 0.6503739356994629f, -0.9539271593093872f, -0.7639755606651306f, + 0.3438030183315277f, -0.65008944272995f, 0.1381210833787918f, + 0.17897674441337585f, 0.6611759066581726f, -0.6359773278236389f, + 0.753626823425293f, -0.8921093940734863f, 0.7966955304145813f, + -0.20982062816619873f, 0.2511371076107025f, -0.23097757995128632f, + 1.1271780729293823f, 0.26889708638191223f }; static const float av1_hd_partition_breakout_nn_weights_8_layer1[32] = { - 1.1129406690597534f, -1.5291675329208374f, -0.8876321911811829f, - 0.30224356055259705f, -0.40715256333351135f, -0.5691804885864258f, - 0.8208304643630981f, -0.6657942533493042f, -1.1190624237060547f, - 3.5144896507263184f, -0.4905086159706116f, -0.04449653998017311f, - 0.8738682270050049f, 0.38138872385025024f, -2.4828171730041504f, - -0.31892910599708557f, 0.4230459928512573f, 0.6047843098640442f, - 0.5003713369369507f, 1.1065287590026855f, 0.97183758020401f, - 0.2900541126728058f, 0.5317407846450806f, 0.6475094556808472f, - 4.615662574768066f, -7.686334133148193f, -0.9662672877311707f, - 0.6237889528274536f, -0.5734347701072693f, -0.4399879276752472f, - -0.016914604231715202f, -0.6294510960578918f + 2.6572957038879395f, 0.17708347737789154f, 0.09978588670492172f, + -0.15976227819919586f, 2.394906997680664f, 0.2525644302368164f, + 0.4016534388065338f, -0.18184596300125122f, 0.20857582986354828f, + -0.6041266918182373f, 0.1736220121383667f, 0.5656698346138f, + 1.886559247970581f, 1.0891855955123901f, 0.3225744664669037f, + 0.5683131217956543f, -1.198813796043396f, -1.226926326751709f, + 0.42729490995407104f, -1.0998389720916748f, -1.6383575201034546f, + 0.47224658727645874f, 0.3989439904689789f, 0.003893804969266057f, + 0.2592338025569916f, -0.9896919131278992f, 0.23605197668075562f, + -1.2522234916687012f, 0.44355013966560364f, -0.37252721190452576f, + 0.4963497519493103f, -0.596142053604126f }; static const float av1_hd_partition_breakout_nn_bias_8_layer1[1] = { - 1.0166552066802979f + 0.8022159934043884f }; static const float @@ -3950,19 +3931,19 @@ { // lowres { - 2.312415f, // p = 0.9099 - 1.848067f, // p = 0.8639 - 2.708050f, // p = 0.9375 - 1.567311f, // p = 0.8274 - 3.167687f, // p = 0.9596 + 1.587045f, // p = 0.830200 + 1.982054f, // p = 0.878900 + 2.150334f, // p = 0.895700 + 2.093809f, // p = 0.890300 + 2.428837f, // p = 0.919000 }, // midres { - 2.312415f, // p = 0.9099 - 1.848067f, // p = 0.8639 - 2.708050f, // p = 0.9375 - 1.567311f, // p = 0.8274 - 3.167687f, // p = 0.9596 + 2.044592f, // p = 0.885400 + 1.961544f, // p = 0.876700 + 2.940236f, // p = 0.949800 + 2.301486f, // p = 0.909000 + 2.265745f, // p = 0.906000 }, // hdres { @@ -3977,19 +3958,19 @@ { // lowres { - 2.312415f, // p = 0.9099 - 1.848067f, // p = 0.8639 - 2.708050f, // p = 0.9375 - 1.567311f, // p = 0.8274 - 3.167687f, // p = 0.9596 + 1.390049f, // p = 0.800600 + 1.784893f, // p = 0.856300 + 3.199090f, // p = 0.960800 + 2.119666f, // p = 0.892800 + 2.026000f, // p = 0.883500 }, // midres { - 2.312415f, // p = 0.9099 - 1.848067f, // p = 0.8639 - 2.708050f, // p = 0.9375 - 1.567311f, // p = 0.8274 - 3.167687f, // p = 0.9596 + 1.587045f, // p = 0.830200 + 1.982054f, // p = 0.878900 + 2.150334f, // p = 0.895700 + 2.093809f, // p = 0.890300 + 2.428837f, // p = 0.919000 }, // hdres { @@ -4101,19 +4082,19 @@ { // lowres { - -2.197225f, // p = 0.1 - -5.289093f, // p = 0.005021 - -2.522730f, // p = 0.07428 - -7.174085f, // p = 0.0007656 - -6.316499f, // p = 0.001803 + -2.758650f, // p = 0.059600 + -4.385780f, // p = 0.012300 + -4.230259f, // p = 0.014340 + -6.455747f, // p = 0.001569 + -8.541803f, // p = 0.000195 }, // midres { - -2.197225f, // p = 0.1 - -5.289093f, // p = 0.005021 - -2.522730f, // p = 0.07428 - -7.174085f, // p = 0.0007656 - -6.316499f, // p = 0.001803 + -4.881569f, // p = 0.007528 + -4.656635f, // p = 0.009409 + -3.037505f, // p = 0.045760 + -6.679364f, // p = 0.001255 + -5.944256f, // p = 0.002614 }, // hdres { @@ -4128,19 +4109,19 @@ { // lowres { - -2.197225f, // p = 0.1 - -5.289093f, // p = 0.005021 - -2.522730f, // p = 0.07428 - -7.174085f, // p = 0.0007656 - -6.316499f, // p = 0.001803 + -4.381672f, // p = 0.012350 + -4.034317f, // p = 0.017390 + -3.986288f, // p = 0.018230 + -7.325495f, // p = 0.000658 + -7.908147f, // p = 0.000368 }, // midres { - -2.197225f, // p = 0.1 - -5.289093f, // p = 0.005021 - -2.522730f, // p = 0.07428 - -7.174085f, // p = 0.0007656 - -6.316499f, // p = 0.001803 + -2.758650f, // p = 0.059600 + -4.385780f, // p = 0.012300 + -4.230259f, // p = 0.014340 + -6.455747f, // p = 0.001569 + -8.541803f, // p = 0.000195 }, // hdres {
diff --git a/av1/encoder/partition_strategy.c b/av1/encoder/partition_strategy.c index a4a1f82..1c0f1bd 100644 --- a/av1/encoder/partition_strategy.c +++ b/av1/encoder/partition_strategy.c
@@ -1588,16 +1588,29 @@ float rate_f = (float)AOMMIN(rd_stats->rate, INT_MAX); rate_f = ((float)x->rdmult / 128.0f / 512.0f / (float)(1 << num_pels_log2)) * rate_f; - features[feature_index++] = rate_f; + features[feature_index++] = + cpi->sf.part_sf.ml_partition_search_breakout_model_index + ? log1pf((float)rate_f) + : rate_f; const float dist_f = (float)(AOMMIN(rd_stats->dist, INT_MAX) >> num_pels_log2); - features[feature_index++] = dist_f; + features[feature_index++] = + cpi->sf.part_sf.ml_partition_search_breakout_model_index + ? log1pf((float)dist_f) + : dist_f; - features[feature_index++] = (float)pb_source_variance; + features[feature_index++] = + cpi->sf.part_sf.ml_partition_search_breakout_model_index + ? log1pf((float)pb_source_variance) + : (float)pb_source_variance; const int dc_q = (int)x->plane[0].dequant_QTX[0] >> (bit_depth - 8); - features[feature_index++] = (float)(dc_q * dc_q) / 256.0f; + features[feature_index++] = + cpi->sf.part_sf.ml_partition_search_breakout_model_index + ? log1pf((float)(dc_q * dc_q) / 256.0f) + : (float)(dc_q * dc_q) / 256.0f; + assert(feature_index == FEATURES); if (cpi->sf.part_sf.ml_partition_search_breakout_model_index) {
diff --git a/av1/encoder/speed_features.c b/av1/encoder/speed_features.c index 56832c6..8cc56a2 100644 --- a/av1/encoder/speed_features.c +++ b/av1/encoder/speed_features.c
@@ -233,12 +233,11 @@ } if (is_720p_or_larger) { - sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9618367258814811f; - sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9990705139233304f; - sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9648891196441841f; - + sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f; + sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f; + sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f; sf->part_sf.ml_partition_search_breakout_model_index = 1; } else { sf->part_sf.ml_partition_search_breakout_thresh[0] = -1.0f; @@ -259,11 +258,11 @@ } if (is_720p_or_larger) { - sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9583713938680828f; - sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9634239069901543f; - sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9000000000000001f; - sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9196596355880025f; + sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f; + sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f; + sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f; sf->part_sf.ml_partition_search_breakout_model_index = 1; } @@ -785,11 +784,11 @@ } if (is_720p_or_larger) { - sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9618367258814811f; - sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9990705139233304f; - sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9648891196441841f; + sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f; + sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f; + sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f; sf->part_sf.ml_partition_search_breakout_model_index = 1; } else { sf->part_sf.ml_partition_search_breakout_thresh[0] = -1.0f; @@ -812,11 +811,11 @@ } if (is_720p_or_larger) { - sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.9583713938680828f; - sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.9999999f; - sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.9634239069901543f; - sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.9000000000000001f; - sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.9196596355880025f; + sf->part_sf.ml_partition_search_breakout_thresh[0] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[1] = 0.5042595622791082f; + sf->part_sf.ml_partition_search_breakout_thresh[2] = 0.5f; + sf->part_sf.ml_partition_search_breakout_thresh[3] = 0.8378425823517456f; + sf->part_sf.ml_partition_search_breakout_thresh[4] = 0.8047585616503903f; sf->part_sf.ml_partition_search_breakout_model_index = 1; }