Home
last modified time | relevance | path

Searched refs:concatInputsTensorInfos (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp560 std::vector<arm_compute::TensorInfo> concatInputsTensorInfos; in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
564 concatInputsTensorInfos.reserve(maxTime); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
581 concatInputsTensorInfos.emplace_back(aclConcatTensorInfo); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
583 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
808 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in ClUnidirectionalSequenceLstmFloatWorkloadValidate()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp560 std::vector<arm_compute::TensorInfo> concatInputsTensorInfos; in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
564 concatInputsTensorInfos.reserve(maxTime); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
581 concatInputsTensorInfos.emplace_back(aclConcatTensorInfo); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
583 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
811 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp541 std::vector<arm_compute::TensorInfo> concatInputsTensorInfos; in NeonUnidirectionalSequenceLstmWorkloadValidate() local
545 concatInputsTensorInfos.reserve(maxTime); in NeonUnidirectionalSequenceLstmWorkloadValidate()
562 concatInputsTensorInfos.emplace_back(aclConcatTensorInfo); in NeonUnidirectionalSequenceLstmWorkloadValidate()
564 concatInputsTensorInfosPtr.emplace_back(&concatInputsTensorInfos[i]); in NeonUnidirectionalSequenceLstmWorkloadValidate()
784 concatInputsTensorInfos[i].set_tensor_shape(acl_shape_expand); in NeonUnidirectionalSequenceLstmWorkloadValidate()