Home
last modified time | relevance | path

Searched refs:conv_weights (Results 1 – 5 of 5) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/generic/
Dimpl.cpp31 void fused_batch_normalization_conv(const ITensor *conv_weights, const ITensor *conv_bias, ITensor … in fused_batch_normalization_conv() argument
35 const int size = 16 / conv_weights->info()->element_size(); in fused_batch_normalization_conv()
38 const bool run_in_place_weights = (fused_weights == nullptr) || (fused_weights == conv_weights); in fused_batch_normalization_conv()
49 Iterator conv_w_in(conv_weights, win); in fused_batch_normalization_conv()
50 Iterator conv_w_out(run_in_place_weights ? conv_weights : fused_weights, win); in fused_batch_normalization_conv()
126 template void fused_batch_normalization_conv<float32_t>(const ITensor *conv_weights, const ITensor …
130 template void fused_batch_normalization_conv<float16_t>(const ITensor *conv_weights, const ITensor …
Dfp32.cpp31 void fused_batch_normalization_conv_f32(const ITensor *conv_weights, const ITensor *conv_bias, ITen… in fused_batch_normalization_conv_f32() argument
34 …return fused_batch_normalization_conv<float32_t>(conv_weights, conv_bias, fused_weights, fused_bia… in fused_batch_normalization_conv_f32()
Dfp16.cpp32 void fused_batch_normalization_conv_f16(const ITensor *conv_weights, const ITensor *conv_bias, ITen… in fused_batch_normalization_conv_f16() argument
35 …return fused_batch_normalization_conv<float16_t>(conv_weights, conv_bias, fused_weights, fused_bia… in fused_batch_normalization_conv_f16()
Dimpl.h35 void fused_batch_normalization_conv(const ITensor *conv_weights, const ITensor *conv_bias, ITensor …
/external/ComputeLibrary/src/cpu/kernels/fuse_batch_normalization/
Dlist.h32 …void func_name(const ITensor *conv_weights, const ITensor *conv_bias, ITensor *fused_weights, ITen…