Searched refs:input_to_output_w (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | LSTMLayerFixture.h | 90 … TensorType input_to_output_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() 173 ARM_COMPUTE_ASSERT(input_to_output_w.info()->is_resizable()); in compute_target() 191 input_to_output_w.allocator()->allocate(); in compute_target() 208 ARM_COMPUTE_ASSERT(!input_to_output_w.info()->is_resizable()); in compute_target() 226 fill(AccessorType(input_to_output_w), 3); in compute_target() 345 SimpleTensor<T> input_to_output_w{ input_weights_shape, data_type }; in compute_reference() 372 fill(input_to_output_w, 3); in compute_reference() 486 …T> fully_connected_output = reference::fully_connected_layer(input, input_to_output_w, output_gate… in compute_reference()
|