Searched defs:output_size_vec (Results 1 – 1 of 1) sorted by relevance
495 auto output_size_vec = in mkldnn_adaptive_avg_pool2d() local647 auto output_size_vec = grad_output.sizes(); in mkldnn_adaptive_avg_pool2d_backward() local