Home
last modified time | relevance | path

Searched refs:max_pool1d_with_indices (Results 1 – 11 of 11) sorted by relevance

/external/pytorch/torch/csrc/api/include/torch/nn/functional/
Dpooling.h188 inline std::tuple<Tensor, Tensor> max_pool1d_with_indices( in max_pool1d_with_indices() function
195 return torch::max_pool1d_with_indices( in max_pool1d_with_indices()
209 inline std::tuple<Tensor, Tensor> max_pool1d_with_indices( in max_pool1d_with_indices() function
212 return detail::max_pool1d_with_indices( in max_pool1d_with_indices()
/external/pytorch/aten/src/ATen/native/
DMaxPooling.cpp91 return std::get<0>(at::max_pool1d_with_indices( in max_pool1d()
DPooling.cpp78 std::tuple<Tensor, Tensor> max_pool1d_with_indices( in max_pool1d_with_indices() function
/external/pytorch/torch/csrc/api/src/nn/modules/
Dpooling.cpp91 return F::detail::max_pool1d_with_indices( in forward_with_indices()
/external/pytorch/functorch/op_analysis/
Dpublic_api427 max_pool1d_with_indices
/external/pytorch/aten/src/ATen/functorch/
DBatchRulesDecompositions.cpp199 OP_DECOMPOSE(max_pool1d_with_indices); in TORCH_LIBRARY_IMPL()
/external/pytorch/torch/nn/
Dfunctional.py664 def max_pool1d_with_indices( function
700 max_pool1d_with_indices,
712 return torch.max_pool1d_with_indices(
747 if_true=max_pool1d_with_indices,
Dfunctional.pyi.in68 def max_pool1d_with_indices(
/external/pytorch/torch/
Doverrides.py775 torch.max_pool1d_with_indices: (
920 torch.nn.functional.max_pool1d_with_indices: (
/external/pytorch/test/
Dtest_namedtensor.py299 … check_tuple_return(F.max_pool1d_with_indices, [named_tensor_1d, 2], named_tensor_1d.names)
/external/pytorch/torch/onnx/
Dsymbolic_opset9.py1491 max_pool1d_with_indices = _onnx_symbolic("aten::max_pool1d_with_indices")( variable
1627 max_pool1d_with_indices,