Searched defs:weight_IOHW_sizes (Results 1 – 1 of 1) sorted by relevance
603 std::vector<int64_t> weight_IOHW_sizes(dim); in _original_deconv_weight_size() local645 …std::vector<int64_t> weight_IOHW_sizes = weight_t.is_mkldnn() ? _original_deconv_weight_size(weigh… in _mkldnn_convolution_transpose() local758 std::vector<int64_t> weight_IOHW_sizes = _original_deconv_weight_size(weight_t, groups); in mkldnn_convolution_transpose_pointwise_meta() local