| /aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/ |
| H A D | armnn_descriptors.i | 1064 ") StackDescriptor;
1065 struct StackDescriptor struct
1067 StackDescriptor();
1068 StackDescriptor(uint32_t axis, uint32_t numInputs, const armnn::TensorShape& inputShape);
1074 bool operator ==(const StackDescriptor& rhs) const; argument
|
| /aosp_15_r20/external/armnn/src/armnn/test/ |
| H A D | InferOutputTests.hpp | 281 void StackInferOutputShapeImpl(const armnn::StackDescriptor descriptor, in StackInferOutputShapeImpl()
294 armnn::StackDescriptor descriptor; in StackInferOutputShapeFromInputsMatchTest()
324 armnn::StackDescriptor descriptor; in StackInferOutputShapeFromInputsNoMatchTest()
352 const armnn::StackDescriptor& descriptor, in CreateStackLayerHelper()
386 armnn::StackDescriptor descriptor; in StackValidateTensorShapesFromInputsMatchTest()
412 armnn::StackDescriptor descriptor; in StackValidateTensorShapesFromInputsNoMatchTest()
|
| H A D | TestNameAndDescriptorLayerVisitor.cpp | 251 armnn::StackDescriptor GetDescriptor<armnn::StackDescriptor>() in GetDescriptor()
253 return armnn::StackDescriptor(1, 2, { 2, 2 }); in GetDescriptor()
|
| /aosp_15_r20/external/armnn/src/armnn/layers/ |
| H A D | StackLayer.hpp | 13 class StackLayer : public LayerWithParameters<StackDescriptor>
44 StackLayer(const StackDescriptor& param, const char* name);
|
| H A D | StackLayer.cpp | 17 StackLayer::StackLayer(const StackDescriptor& param, const char* name) in StackLayer()
|
| /aosp_15_r20/external/armnn/src/armnn/ |
| H A D | SerializeLayerParameters.hpp | 172 template <> struct StringifyLayerParameters<StackDescriptor>
174 static void Serialize(ParameterStringifyFunction& fn, const StackDescriptor& desc);
|
| H A D | SerializeLayerParameters.cpp | 487 void StringifyLayerParameters<StackDescriptor>::Serialize(ParameterStringifyFunction& fn, const Sta… in Serialize()
|
| H A D | Network.hpp | 207 IConnectableLayer* AddStackLayer(const StackDescriptor& stackDescriptor,
|
| H A D | ILayerSupport.cpp | 388 *(PolymorphicDowncast<const StackDescriptor*>(&descriptor)), in IsLayerSupported()
1153 const StackDescriptor& descriptor, in IsStackSupported()
|
| /aosp_15_r20/external/armnn/include/armnn/ |
| H A D | Descriptors.hpp | 1218 struct StackDescriptor : BaseDescriptor struct
1220 StackDescriptor() in StackDescriptor() argument
1226 StackDescriptor(uint32_t axis, uint32_t numInputs, const TensorShape& inputShape) in StackDescriptor() function
1232 bool operator ==(const StackDescriptor& rhs) const in operator ==() argument
|
| H A D | DescriptorsFwd.hpp | 49 struct StackDescriptor;
|
| H A D | LayerSupport.hpp | 363 const StackDescriptor& descriptor,
|
| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonStackWorkload.hpp | 16 const StackDescriptor& descriptor);
|
| H A D | NeonStackWorkload.cpp | 29 const StackDescriptor& descriptor) in NeonStackWorkloadValidate()
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClStackWorkload.hpp | 16 const StackDescriptor& descriptor);
|
| H A D | ClStackWorkload.cpp | 31 const StackDescriptor& descriptor) in ClStackWorkloadValidate()
|
| /aosp_15_r20/external/armnn/delegate/classic/src/ |
| H A D | Pack.hpp | 65 armnn::StackDescriptor desc; in VisitPackOperator()
|
| /aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/ |
| H A D | __init__.py | 104 StackDescriptor, StridedSliceDescriptor, SoftmaxDescriptor, TransposeConvolution2dDescriptor, \
|
| /aosp_15_r20/external/armnn/src/armnnSerializer/ |
| H A D | ArmnnSchema.fbs | 982 descriptor:StackDescriptor;
985 table StackDescriptor {
|
| H A D | Serializer.hpp | 324 const armnn::StackDescriptor& stackDescriptor,
|
| /aosp_15_r20/external/armnn/src/backends/cl/ |
| H A D | ClLayerSupport.cpp | 608 *(PolymorphicDowncast<const StackDescriptor*>(&descriptor)), in IsLayerSupported()
1484 const StackDescriptor& descriptor, in IsStackSupported()
|
| H A D | ClLayerSupport.hpp | 325 const StackDescriptor& descriptor,
|
| /aosp_15_r20/external/armnn/src/backends/neon/ |
| H A D | NeonLayerSupport.cpp | 603 … *(PolymorphicDowncast<const StackDescriptor*>(&descriptor)), in IsLayerTypeSupported()
1535 const StackDescriptor& descriptor, in IsStackSupported()
|
| H A D | NeonLayerSupport.hpp | 320 const StackDescriptor& descriptor,
|
| /aosp_15_r20/external/armnn/include/armnn/backends/ |
| H A D | WorkloadData.hpp | 152 struct StackQueueDescriptor : QueueDescriptorWithParameters<StackDescriptor>
|