• Home
  • Raw
  • Download

Lines Matching refs:data_layout

41 …sor<T> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout)  in pooling_layer_internal()  argument
100 if(data_layout == DataLayout::NCHW) in pooling_layer_internal()
176 …loat> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
178 …half> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
180 …half> &src, const PoolingLayerInfo &info, SimpleTensor<uint32_t> *indices, DataLayout data_layout);
183 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer() argument
186 return pooling_layer_internal<T, T>(src, info, indices, data_layout); in pooling_layer()
191 DataLayout data_layout) in pooling_layer() argument
194 …SimpleTensor<float> dst_tmp = pooling_layer_internal<float>(src_tmp, info, indices, data_layout); in pooling_layer()
200 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer() argument
203 … SimpleTensor<float> dst_tmp = pooling_layer_internal<float>(src_tmp, info, indices, data_layout); in pooling_layer()
209 …nfo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout) in pooling_layer() argument
214 return pooling_layer_internal<half, float>(src, info, indices, data_layout); in pooling_layer()
217 return pooling_layer_internal<half>(src, info, indices, data_layout); in pooling_layer()
220 …fo, const QuantizationInfo &output_qinfo, SimpleTensor<uint32_t> *indices, DataLayout data_layout);