Searched refs:input_to_forget_w (Results 1 – 1 of 1) sorted by relevance
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | LSTMLayerFixture.h | 88 … TensorType input_to_forget_w = create_tensor<TensorType>(input_weights_shape, data_type); in compute_target() local
164 …lstm.configure(&input, &input_to_forget_w, &input_to_cell_w, &input_to_output_w, &recurrent_to_for… in compute_target()
171 ARM_COMPUTE_ASSERT(input_to_forget_w.info()->is_resizable()); in compute_target()
189 input_to_forget_w.allocator()->allocate(); in compute_target()
206 ARM_COMPUTE_ASSERT(!input_to_forget_w.info()->is_resizable()); in compute_target()
224 fill(AccessorType(input_to_forget_w), 1); in compute_target()
343 SimpleTensor<T> input_to_forget_w{ input_weights_shape, data_type }; in compute_reference()
370 fill(input_to_forget_w, 1); in compute_reference()
409 …T> fully_connected_forget = reference::fully_connected_layer(input, input_to_forget_w, forget_gate… in compute_reference()
|