Searched refs:adaptive_max_pool2d_with_indices (Results 1 – 6 of 6) sorted by relevance
425 inline std::tuple<Tensor, Tensor> adaptive_max_pool2d_with_indices( in adaptive_max_pool2d_with_indices() function444 inline std::tuple<Tensor, Tensor> adaptive_max_pool2d_with_indices( in adaptive_max_pool2d_with_indices() function447 return detail::adaptive_max_pool2d_with_indices(input, options.output_size()); in adaptive_max_pool2d_with_indices()455 return std::get<0>(adaptive_max_pool2d_with_indices(input, output_size)); in adaptive_max_pool2d()
164 return F::detail::adaptive_max_pool2d_with_indices( in forward_with_indices()
398 adaptive_max_pool2d_with_indices
1241 def adaptive_max_pool2d_with_indices( function1260 adaptive_max_pool2d_with_indices,1283 return adaptive_max_pool2d_with_indices(input, output_size)[0]1290 if_true=adaptive_max_pool2d_with_indices,
145 def adaptive_max_pool2d_with_indices(
830 …torch.nn.functional.adaptive_max_pool2d_with_indices: lambda input, output_size, return_indices=Fa…