Searched defs:dilation_vec (Results 1 – 2 of 2) sorted by relevance
217 auto dilation_vec = expand_param_if_needed(dilation, "dilation", dims); in _mkldnn_pooling() local308 auto dilation_vec = expand_param_if_needed(dilation, "dilation", dims); in _mkldnn_pooling_backward() local
101 auto dilation_vec = dilation_.vec(); in apply_impl_helper() local