Searched defs:inputShapeShrink (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 280 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmWorkload() local
299 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmWorkload() local
715 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
734 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 294 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkload() local
313 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkload() local
733 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
752 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 294 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkload() local
310 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkload() local
734 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
750 TensorShape inputShapeShrink({inputShape[1], inputShape[2]}); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local
|