Searched defs:recurrent_weights_shape (Results 1 – 4 of 4) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | RNNLayerFixture.h | 46 … setup(TensorShape input_shape, TensorShape weights_shape, TensorShape recurrent_weights_shape, Te… in setup()
64 …rShape &input_shape, const TensorShape &weights_shape, const TensorShape &recurrent_weights_shape,… in compute_target()
114 …rShape &input_shape, const TensorShape &weights_shape, const TensorShape &recurrent_weights_shape,… in compute_reference()
|
| D | LSTMLayerFixture.h | 50 …(TensorShape input_shape, TensorShape input_weights_shape, TensorShape recurrent_weights_shape, Te… in setup()
79 … &input_shape, const TensorShape &input_weights_shape, const TensorShape &recurrent_weights_shape,… in compute_target()
324 … &input_shape, const TensorShape &input_weights_shape, const TensorShape &recurrent_weights_shape,… in compute_reference()
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | LSTMLayerQuantized.cpp | 89 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_SUITE() local
228 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_CASE() local
468 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_SUITE() local
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | LSTMLayerQuantized.cpp | 96 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_SUITE() local
235 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_CASE() local
475 TensorShape recurrent_weights_shape{ output_size, output_size }; in TEST_SUITE() local
|