Searched refs:LSTMModel (Results 1 – 5 of 5) sorted by relevance
| /external/executorch/examples/models/lstm/ |
| D | __init__.py | 8 from .model import LSTMModel
11 LSTMModel,
|
| D | model.py | 17 class LSTMModel(EagerModelBase): class
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | selective_build_test.cc | 71 TEST(SelectiveBuiltTest, LSTMModel) { in TEST() argument
|
| /external/pytorch/test/onnx/ |
| D | test_operators.py | 1065 class LSTMModel(torch.nn.Module): class
1077 LSTMModel(),
|
| D | test_pytorch_onnx_onnxruntime.py | 4655 class LSTMModel(torch.nn.Module): class
4664 self.run_test(LSTMModel(), (input,))
4668 class LSTMModel(torch.nn.Module): class
4678 self.run_test(LSTMModel(), (input,))
4682 class LSTMModel(torch.nn.Module): class
4695 self.run_test(LSTMModel(), (input, h0, c0))
4717 class LSTMModel(torch.nn.Module): class
4728 self.run_test(LSTMModel(), input)
4732 class LSTMModel(torch.nn.Module): class
4750 LSTMModel(), input, fixed_batch_size=True, additional_test_inputs=[input2]
[all …]
|