Lines Matching refs:SimpleTensor

41 SimpleTensor<T> pooling_layer_internal(const SimpleTensor<T> &src, const PoolingLayerInfo &info, Si…  in pooling_layer_internal()
45 …SimpleTensor<T> dst{ compute_pool_shape(TensorInfo(src.shape(), 1, src.data_type()), info), src.da…  in pooling_layer_internal()
49         *indices = SimpleTensor<uint32_t> { pooled_shape, DataType::U32, 1 };  in pooling_layer_internal()
177 template SimpleTensor<float> pooling_layer_internal<float>(const SimpleTensor<float> &src, const Po…
179 template SimpleTensor<half> pooling_layer_internal<half>(const SimpleTensor<half> &src, const Pooli…
181 template SimpleTensor<half> pooling_layer_internal<half, float>(const SimpleTensor<half> &src, cons…
184 SimpleTensor<T> pooling_layer(const SimpleTensor<T> &src, const PoolingLayerInfo &info, const Quant…  in pooling_layer()
191 SimpleTensor<uint8_t> pooling_layer<uint8_t>(const SimpleTensor<uint8_t> &src, const PoolingLayerIn…  in pooling_layer()
194     SimpleTensor<float>   src_tmp = convert_from_asymmetric(src);  in pooling_layer()
195 …SimpleTensor<float>   dst_tmp = pooling_layer_internal<float>(src_tmp, info, indices, data_layout);  in pooling_layer()
196     SimpleTensor<uint8_t> dst     = convert_to_asymmetric<uint8_t>(dst_tmp, output_qinfo);  in pooling_layer()
201 SimpleTensor<int8_t> pooling_layer<int8_t>(const SimpleTensor<int8_t> &src, const PoolingLayerInfo …  in pooling_layer()
203     SimpleTensor<float>  src_tmp = convert_from_asymmetric(src);  in pooling_layer()
204 … SimpleTensor<float>  dst_tmp = pooling_layer_internal<float>(src_tmp, info, indices, data_layout);  in pooling_layer()
205     SimpleTensor<int8_t> dst     = convert_to_asymmetric<int8_t>(dst_tmp, output_qinfo);  in pooling_layer()
210 SimpleTensor<half> pooling_layer(const SimpleTensor<half> &src, const PoolingLayerInfo &info, const…  in pooling_layer()
221 …te SimpleTensor<float> pooling_layer(const SimpleTensor<float> &src, const PoolingLayerInfo &info,…