1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
8 #include "nnet_base_matcher.hpp"
9 #include "quantization/quantization.h"
11 class ConvoluionLayerMatcher : public ::testing::MatcherInterface<const intel_nnet_type_t*> {
15 ConvoluionLayerMatcher(bool matchInserted, int matchQuantity) : matchInserted(matchInserted), matchQuantity(matchQuantity) {}
16 bool MatchAndExplain(const intel_nnet_type_t *foo, ::testing::MatchResultListener *listener) const override {
19 for(int i = 0; i < foo->nLayers; i++) {
20 if (foo->pLayers[i].nLayerKind != INTEL_CONVOLUTIONAL) continue;
22 auto conv = (intel_convolutional_layer_t*)foo->pLayers[i].pLayerStruct;
26 return !matchInserted;
28 void DescribeTo(::std::ostream *os) const override {
29 *os << "should "<< (matchInserted ? "" : "not ") << "have Convolution primitive as part of nnet structure";