Searched defs:pooled_plane (Results 1 – 1 of 1) sorted by relevance
32 const dtype* __restrict__ output_data, const int pooled_plane, in MaxPoolGradBackwardNoMaskNCDHW()83 const dtype* __restrict__ output_data, const int pooled_plane, in MaxPoolGradBackwardNoMaskNDHWC()139 const int batch, const int pooled_plane, const int pooled_height, in operator ()()