Home
last modified time | relevance | path

Searched refs:recurrent_weights_vector (Results 1 – 2 of 2) sorted by relevance

/external/ComputeLibrary/src/runtime/CL/functions/
DCLLSTMLayerQuantized.cpp121 std::vector<const ICLTensor *> recurrent_weights_vector; in configure() local
122 recurrent_weights_vector.emplace_back(recurrent_to_input_weights); in configure()
123 recurrent_weights_vector.emplace_back(recurrent_to_forget_weights); in configure()
124 recurrent_weights_vector.emplace_back(recurrent_to_cell_weights); in configure()
125 recurrent_weights_vector.emplace_back(recurrent_to_output_weights); in configure()
131 …_concat_recurrent_weights.configure(compile_context, recurrent_weights_vector, &_recurrent_weights… in configure()
340 std::vector<const ITensorInfo *> recurrent_weights_vector; in validate() local
341 recurrent_weights_vector.emplace_back(recurrent_to_input_weights); in validate()
342 recurrent_weights_vector.emplace_back(recurrent_to_forget_weights); in validate()
343 recurrent_weights_vector.emplace_back(recurrent_to_cell_weights); in validate()
[all …]
/external/ComputeLibrary/src/runtime/NEON/functions/
DNELSTMLayerQuantized.cpp104 …std::vector<const ITensor *> recurrent_weights_vector{ recurrent_to_input_weights, recurrent_to_fo… in configure() local
110 … _concat_recurrent_weights.configure(recurrent_weights_vector, &_recurrent_weights, Window::DimY); in configure()
307 std::vector<const ITensorInfo *> recurrent_weights_vector; in validate() local
308 recurrent_weights_vector.emplace_back(recurrent_to_input_weights); in validate()
309 recurrent_weights_vector.emplace_back(recurrent_to_forget_weights); in validate()
310 recurrent_weights_vector.emplace_back(recurrent_to_cell_weights); in validate()
311 recurrent_weights_vector.emplace_back(recurrent_to_output_weights); in validate()
313 …ARM_COMPUTE_RETURN_ON_ERROR(NEConcatenateLayer::validate(recurrent_weights_vector, &recurrent_weig… in validate()