Lines Matching refs:SimpleTensor
36 …st SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, SimpleTensor<T> &w… in fuse_batch_normalization_dwc_layer()
37 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon) in fuse_batch_normalization_dwc_layer()
71 …n_conv_layer(const SimpleTensor<T> &w, const SimpleTensor<T> &mean, const SimpleTensor<T> &var, Si… in fuse_batch_normalization_conv_layer()
72 const SimpleTensor<T> &b, in fuse_batch_normalization_conv_layer()
73 … const SimpleTensor<T> &beta, const SimpleTensor<T> &gamma, float epsilon) in fuse_batch_normalization_conv_layer()
104 …alization_dwc_layer(const SimpleTensor<float> &w, const SimpleTensor<float> &mean, const SimpleTen…
105 …SimpleTensor<float> &b_fused, const SimpleTensor<float> &b, const SimpleTensor<float> &beta, const…
106 …_layer(const SimpleTensor<half> &w, const SimpleTensor<half> &mean, const SimpleTensor<half> &var,…
107 …const SimpleTensor<half> &b, const SimpleTensor<half> &beta, const SimpleTensor<half> &gamma, floa…
108 …lization_conv_layer(const SimpleTensor<float> &w, const SimpleTensor<float> &mean, const SimpleTen…
109 …SimpleTensor<float> &b_fused, const SimpleTensor<float> &b, const SimpleTensor<float> &beta, const…
110 …_layer(const SimpleTensor<half> &w, const SimpleTensor<half> &mean, const SimpleTensor<half> &var,…
111 …const SimpleTensor<half> &b, const SimpleTensor<half> &beta, const SimpleTensor<half> &gamma, floa…