Searched defs:adaptive_max_pool2d (Results 1 – 4 of 4) sorted by relevance
15 TORCH_META_FUNC(adaptive_max_pool2d) (const Tensor& input, IntArrayRef output_size) { in TORCH_META_FUNC() argument
452 inline Tensor adaptive_max_pool2d( in adaptive_max_pool2d() function473 inline Tensor adaptive_max_pool2d( in adaptive_max_pool2d() function
1286 adaptive_max_pool2d = boolean_dispatch( variable
4289 def adaptive_max_pool2d(x, output_size): function