Searched defs:shapeExpandTimeMajor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 395 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in NeonUnidirectionalSequenceLstmWorkload() local
772 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 414 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in NeonUnidirectionalSequenceLstmFloatWorkload() local
799 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 411 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in ClUnidirectionalSequenceLstmFloatWorkload() local
796 TensorShape shapeExpandTimeMajor({1, shape[0], shape[1]}); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
|