| /external/tensorflow/tensorflow/lite/kernels/ |
| D | unidirectional_sequence_lstm_test.cc | 138 UnidirectionalLSTMOpModel* lstm, float tolerance = 1e-5, in VerifyGoldens()
255 UnidirectionalLSTMOpModel lstm( in TEST_F() local
317 UnidirectionalLSTMOpModel lstm( in TEST_F() local
384 HybridUnidirectionalLSTMOpModel lstm( in TEST_P() local
447 HybridUnidirectionalLSTMOpModel lstm( in TEST_P() local
558 UnidirectionalLSTMOpModel lstm( in TEST_F() local
620 HybridUnidirectionalLSTMOpModel lstm( in TEST_P() local
682 HybridUnidirectionalLSTMOpModel lstm( in TEST_P() local
1344 UnidirectionalLSTMOpModel lstm( in TEST_F() local
1413 HybridUnidirectionalLSTMOpModel lstm( in TEST_P() local
[all …]
|
| D | bidirectional_sequence_lstm_test.cc | 440 BidirectionalLSTMOpModel lstm( in TEST_P() local
606 BidirectionalLSTMOpModel lstm( in TEST_P() local
769 BidirectionalLSTMOpModel lstm( in TEST() local
930 BidirectionalLSTMOpModel lstm( in TEST() local
1081 BidirectionalLSTMOpModel lstm( in TEST() local
1232 BidirectionalLSTMOpModel lstm( in TEST() local
1935 BidirectionalLSTMOpModel lstm( in TEST() local
2640 BidirectionalLSTMOpModel lstm( in TEST_P() local
2814 BidirectionalLSTMOpModel lstm( in TEST_P() local
|
| D | quant_basic_lstm_test.cc | 168 QuantizedLSTMOpModel* lstm) { in VerifyGoldens()
210 QuantizedLSTMOpModel lstm( in TEST_P() local
|
| D | lstm_test.cc | 341 void VerifyGoldens(LSTMOpModel* lstm, float tolerance) { in VerifyGoldens()
383 void SetAllWeightsAndBiases(LSTMOpModel* lstm) { in SetAllWeightsAndBiases()
475 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_P() local
546 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_P() local
1128 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_P() local
1224 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_P() local
1298 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_P() local
1692 LSTMIntegerOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
1860 LSTMIntegerOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
2030 LSTMIntegerOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
|
| D | optional_tensor_test.cc | 235 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST() local
|
| D | lstm.cc | 42 namespace lstm { namespace
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | lstm_full_test.cc | 228 void VerifyGoldens(LSTMOpModel* lstm, float tolerance) { in VerifyGoldens()
318 LSTMOpModel lstm( in TEST_P() local
386 LSTMOpModel lstm( in TEST_P() local
949 LSTMOpModel lstm( in TEST_P() local
1035 LSTMOpModel lstm( in TEST_F() local
1098 LSTMOpModel lstm( in TEST_F() local
|
| /external/pytorch/test/cpp/jit/ |
| D | test_utils.cpp | 239 std::pair<at::Tensor, at::Tensor> lstm( in lstm() function
|
| /external/gemmlowp/test/ |
| D | benchmark_meta_gemm.cc | 292 std::vector<Shape> lstm; in main() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 163 FunctionType lstm; in compute_target() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils.cc | 727 auto lstm = builder->create<mlir::TFL::UnidirectionalSequenceLSTMOp>( in ConvertKerasLSTMLayer() local
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 3452 LSTMOpModel* lstm, float tolerance = 1e-5) { in VerifyGoldens()
3538 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F() local
3602 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F() local
3714 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F() local
4372 LSTMOpModel lstm(n_batch, n_input, n_cell, n_output, in TEST_F() local
|
| /external/pytorch/aten/src/ATen/native/ |
| D | RNN.cpp | 1430 std::tuple<Tensor, Tensor, Tensor> lstm( in lstm() function
1491 std::tuple<Tensor, Tensor, Tensor> lstm( in lstm() function
|
| /external/pytorch/torch/onnx/ |
| D | symbolic_opset9.py | 4556 def lstm(g: jit_utils.GraphContext, *args): function
|
| /external/pytorch/test/cpp/api/ |
| D | modules.cpp | 4434 auto lstm = LSTMCell(1, 2); in TEST_F() local
|
| /external/pytorch/test/ |
| D | test_jit.py | 2843 def lstm(x, hx, cx, w_ih, w_hh, b_ih, b_hh): function
11145 def lstm(x, hx, cx, w_ih, w_hh, b_ih, b_hh): function
|