Searched refs:input_weights_shape (Results 1 – 3 of 3) sorted by relevance
| /external/ComputeLibrary/tests/validation/CL/ |
| D | LSTMLayerQuantized.cpp | 88 TensorShape input_weights_shape{ input_size, output_size }; in TEST_SUITE() local
93 …auto input_to_input_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE()
94 …auto input_to_forget_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE()
95 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE()
96 …auto input_to_output_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_SUITE()
227 TensorShape input_weights_shape{ input_size, output_size }; in TEST_CASE() local
232 …auto input_to_input_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE()
233 …auto input_to_forget_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE()
234 …auto input_to_cell_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE()
235 …auto input_to_output_weights = create_tensor<CLTensor>(input_weights_shape, DataType::QASYMM8,… in TEST_CASE()
[all …]
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | LSTMLayerQuantized.cpp | 95 TensorShape input_weights_shape{ input_size, output_size }; in TEST_SUITE() local
100 …auto input_to_input_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE()
101 …auto input_to_forget_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE()
102 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE()
103 …auto input_to_output_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_SUITE()
234 TensorShape input_weights_shape{ input_size, output_size }; in TEST_CASE() local
239 …auto input_to_input_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE()
240 …auto input_to_forget_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE()
241 …auto input_to_cell_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE()
242 …auto input_to_output_weights = create_tensor<Tensor>(input_weights_shape, DataType::QASYMM8, 1… in TEST_CASE()
[all …]
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 50 …void setup(TensorShape input_shape, TensorShape input_weights_shape, TensorShape recurrent_weights… in setup() argument
54 …_target = compute_target(input_shape, input_weights_shape, recurrent_weights_shape, cell_bias_shap… in setup()
56 …_reference = compute_reference(input_shape, input_weights_shape, recurrent_weights_shape, cell_bia… in setup()
79 … compute_target(const TensorShape &input_shape, const TensorShape &input_weights_shape, const Tens… in compute_target() argument
83 const unsigned int num_cells = input_weights_shape.y(); in compute_target()
88 … TensorType input_to_forget_w = create_tensor<TensorType>(input_weights_shape, data_type); in compute_target()
89 … TensorType input_to_cell_w = create_tensor<TensorType>(input_weights_shape, data_type); in compute_target()
90 … TensorType input_to_output_w = create_tensor<TensorType>(input_weights_shape, data_type); in compute_target()
122 input_to_input_w = create_tensor<TensorType>(input_weights_shape, data_type); in compute_target()
324 …mpute_reference(const TensorShape &input_shape, const TensorShape &input_weights_shape, const Tens… in compute_reference() argument
[all …]
|