Home
last modified time | relevance | path

Searched defs:layer_weights (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/libaom/av1/encoder/
H A Dml.c42 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_c() local
61 const float *layer_weights = nn_config->weights[num_layers]; in av1_nn_predict_c() local
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dml_avx2.c149 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_avx2() local
H A Dml_sse3.c162 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_sse3() local
/aosp_15_r20/external/libaom/av1/encoder/arm/
H A Dml_neon.c275 const float *layer_weights = nn_config->weights[layer]; in av1_nn_predict_neon() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp484 auto layer_weights = weights[index]; in mkldnn_rnn() local