Searched defs:shapeExpandBatchMajor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 396 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in NeonUnidirectionalSequenceLstmWorkload() local
773 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 415 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in NeonUnidirectionalSequenceLstmFloatWorkload() local
800 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 412 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in ClUnidirectionalSequenceLstmFloatWorkload() local
797 TensorShape shapeExpandBatchMajor({shape[0], 1, shape[1]}); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
|