| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | rnn_cell_impl.py | 42 LSTMCell = rnn_cell_impl.LSTMCell variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | lstm.py | 45 def LSTMCell(weights, m_prev, c_prev, x, pad): function
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantizable/modules/ |
| H A D | rnn.py | 19 class LSTMCell(torch.nn.Module): class
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | RNN.cpp | 722 struct LSTMCell : Cell<std::tuple<Tensor, Tensor>, cell_params> { struct
723 using hidden_type = std::tuple<Tensor, Tensor>;
725 hidden_type operator()( in operator ()()
|
| /aosp_15_r20/external/pytorch/benchmarks/fastrnns/ |
| H A D | custom_lstms.py | 122 class LSTMCell(jit.ScriptModule): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| H A D | rnn.py | 245 class LSTMCell(RNNCellBase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/legacy_rnn/ |
| H A D | rnn_cell_impl.py | 828 class LSTMCell(LayerRNNCell): class
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | rnn.py | 1609 class LSTMCell(RNNCellBase): class
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTM.cpp | 54 LSTMCell::LSTMCell(const Operation& operation, RunTimeOperandInfo* operands) { in LSTMCell() function in android::nn::LSTMCell
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/ |
| H A D | recurrent_v2.py | 841 class LSTMCell(recurrent.LSTMCell): class
|
| H A D | recurrent.py | 2252 class LSTMCell(DropoutRNNCellMixin, Layer): class
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/dynamic/modules/ |
| H A D | rnn.py | 1261 class LSTMCell(RNNCellBase): class
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | function_test.py | 1515 def LSTMCell(cls, x, mprev, cprev, weights): member in UnrollLSTMTest
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | modules.cpp | 4432 TEST_F(ModulesTest, LSTMCell) { in TEST_F() argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 207 def LSTMCell(input, hidden, w_ih, w_hh, b_ih=None, b_hh=None): function
|