Home
last modified time | relevance | path

Searched defs:forward_with_indices (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dpooling.cpp89 std::tuple<Tensor, Tensor> MaxPool1dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::MaxPool1dImpl
110 std::tuple<Tensor, Tensor> MaxPool2dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::MaxPool2dImpl
131 std::tuple<Tensor, Tensor> MaxPool3dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::MaxPool3dImpl
152 std::tuple<Tensor, Tensor> AdaptiveMaxPool1dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::AdaptiveMaxPool1dImpl
162 std::tuple<Tensor, Tensor> AdaptiveMaxPool2dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::AdaptiveMaxPool2dImpl
172 std::tuple<Tensor, Tensor> AdaptiveMaxPool3dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::AdaptiveMaxPool3dImpl
319 std::tuple<Tensor, Tensor> FractionalMaxPool2dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::FractionalMaxPool2dImpl
378 std::tuple<Tensor, Tensor> FractionalMaxPool3dImpl::forward_with_indices( in forward_with_indices() function in torch::nn::FractionalMaxPool3dImpl