Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Drnn.cpp428 std::tuple<Tensor, Tensor> RNNImpl::forward_helper( in forward_helper() function in torch::nn::RNNImpl
606 std::tuple<Tensor, std::tuple<Tensor, Tensor>> LSTMImpl::forward_helper( in forward_helper() function in torch::nn::LSTMImpl
715 std::tuple<Tensor, Tensor> GRUImpl::forward_helper( in forward_helper() function in torch::nn::GRUImpl
/aosp_15_r20/external/pytorch/test/quantization/core/experimental/
H A Dtest_fake_quantize.py9 forward_helper = fake_quantize_function.forward variable
/aosp_15_r20/external/pytorch/torch/nn/utils/_expanded_weights/
H A Dexpanded_weights_utils.py38 def forward_helper(func, expanded_args, expanded_kwargs): function