Searched refs:adaptive_max_pool3d_with_indices (Results 1 – 7 of 7) sorted by relevance
481 inline std::tuple<Tensor, Tensor> adaptive_max_pool3d_with_indices( in adaptive_max_pool3d_with_indices() function500 inline std::tuple<Tensor, Tensor> adaptive_max_pool3d_with_indices( in adaptive_max_pool3d_with_indices() function503 return detail::adaptive_max_pool3d_with_indices(input, options.output_size()); in adaptive_max_pool3d_with_indices()511 return std::get<0>(adaptive_max_pool3d_with_indices(input, output_size)); in adaptive_max_pool3d()
174 return F::detail::adaptive_max_pool3d_with_indices( in forward_with_indices()
399 adaptive_max_pool3d_with_indices
1297 def adaptive_max_pool3d_with_indices( function1317 adaptive_max_pool3d_with_indices,1340 return adaptive_max_pool3d_with_indices(input, output_size)[0]1347 if_true=adaptive_max_pool3d_with_indices,
150 def adaptive_max_pool3d_with_indices(
234 return torch.nn.functional.adaptive_max_pool3d_with_indices(
832 …torch.nn.functional.adaptive_max_pool3d_with_indices: lambda input, output_size, return_indices=Fa…