Searched refs:pool_output_sizes (Results 1 – 3 of 3) sorted by relevance
226 const std::vector<int64_t> output_sizes_ceil = pool_output_sizes( in _mkldnn_pooling()238 output_sizes = pool_output_sizes( in _mkldnn_pooling()256 output_sizes = pool_output_sizes( in _mkldnn_pooling()314 const std::vector<int64_t> output_sizes_ceil = pool_output_sizes( in _mkldnn_pooling_backward()327 output_sizes = pool_output_sizes( in _mkldnn_pooling_backward()
24 std::vector<int64_t> pool_output_sizes(
8 std::vector<int64_t> pool_output_sizes( in pool_output_sizes() function