Searched defs:nonlinearity (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/ |
| H A D | init.cpp | 44 NonlinearityType nonlinearity) { in calculate_kaiming_std()
59 double calculate_gain(NonlinearityType nonlinearity, double param) { in calculate_gain()
184 NonlinearityType nonlinearity) { in kaiming_uniform_()
196 NonlinearityType nonlinearity) { in kaiming_normal_()
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | init.py | 72 def calculate_gain(nonlinearity, param=None): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | rnn.cpp | 402 compute_rnn_options_base_mode(RNNOptions::nonlinearity_t nonlinearity) { in compute_rnn_options_base_mode()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | rnn.py | 1216 self, input_size, hidden_size, bias=True, nonlinearity="tanh", dtype=torch.qint8 argument
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 3044 def rnn_cell(nonlinearity): argument
3051 def rnn_cell_data(nonlinearity): argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 10104 nonlinearity, argument
10112 def __init__(self, layers, nonlinearity, bidirect, dropout, batch_first): argument
10130 def __init__(self, layers, nonlinearity, bidirect, dropout, batch_first): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 9230 def test(nonlinearity, *args, **kwargs): argument
|