| 2072 | } |
| 2073 | |
| 2074 | void SerializerStrategy::ExecuteStrategy(const armnn::IConnectableLayer* layer, |
| 2075 | const BaseDescriptor& descriptor, |
| 2076 | const std::vector<armnn::ConstTensor>& constants, |
| 2077 | const char* name, |
| 2078 | const armnn::LayerBindingId id) |
| 2079 | { |
| 2080 | IgnoreUnused(constants); |
| 2081 | |
| 2082 | switch (layer->GetType()) |
| 2083 | { |
| 2084 | case armnn::LayerType::Activation : |
| 2085 | { |
| 2086 | const armnn::ActivationDescriptor& layerDescriptor = |
| 2087 | static_cast<const armnn::ActivationDescriptor&>(descriptor); |
| 2088 | SerializeActivationLayer(layer, layerDescriptor, name); |
| 2089 | break; |
| 2090 | } |
| 2091 | case armnn::LayerType::Addition : |
| 2092 | { |
| 2093 | SerializeAdditionLayer(layer, name); |
| 2094 | break; |
| 2095 | } |
| 2096 | case armnn::LayerType::ArgMinMax : |
| 2097 | { |
| 2098 | const armnn::ArgMinMaxDescriptor& layerDescriptor = |
| 2099 | static_cast<const armnn::ArgMinMaxDescriptor&>(descriptor); |
| 2100 | SerializeArgMinMaxLayer(layer, layerDescriptor, name); |
| 2101 | break; |
| 2102 | } |
| 2103 | case armnn::LayerType::BatchMatMul: |
| 2104 | { |
| 2105 | const armnn::BatchMatMulDescriptor& layerDescriptor = |
| 2106 | static_cast<const armnn::BatchMatMulDescriptor&>(descriptor); |
| 2107 | SerializeBatchMatMulLayer(layer, |
| 2108 | layerDescriptor, |
| 2109 | name); |
| 2110 | break; |
| 2111 | } |
| 2112 | case armnn::LayerType::BatchNormalization : |
| 2113 | { |
| 2114 | const armnn::BatchNormalizationDescriptor& layerDescriptor = |
| 2115 | static_cast<const armnn::BatchNormalizationDescriptor&>(descriptor); |
| 2116 | SerializeBatchNormalizationLayer(layer, |
| 2117 | layerDescriptor, |
| 2118 | constants, |
| 2119 | name); |
| 2120 | break; |
| 2121 | } |
| 2122 | case armnn::LayerType::BatchToSpaceNd : |
| 2123 | { |
| 2124 | const armnn::BatchToSpaceNdDescriptor& layerDescriptor = |
| 2125 | static_cast<const armnn::BatchToSpaceNdDescriptor&>(descriptor); |
| 2126 | SerializeBatchToSpaceNdLayer(layer, |
| 2127 | layerDescriptor, |
| 2128 | name); |
| 2129 | break; |
| 2130 | } |
| 2131 | case armnn::LayerType::Cast : |
no test coverage detected