Searched refs:_mkldnn_pooling (Results 1 – 1 of 1) sorted by relevance
201 static Tensor _mkldnn_pooling( in _mkldnn_pooling() function380 return _mkldnn_pooling( in mkldnn_max_pool2d()404 return _mkldnn_pooling( in mkldnn_max_pool3d()429 return _mkldnn_pooling( in mkldnn_avg_pool2d()465 return _mkldnn_pooling( in mkldnn_avg_pool3d()507 return _mkldnn_pooling( in mkldnn_adaptive_avg_pool2d()