21 const char* name =
nullptr)
23 , m_Descriptor(convolution2dDescriptor)
34 const char* name =
nullptr)
override 58 const char* name =
nullptr)
60 , m_Descriptor(descriptor)
71 const char* name =
nullptr)
override 93 const char* name =
nullptr)
95 , m_Descriptor(descriptor)
102 const char* name =
nullptr)
override 123 const char* name =
nullptr)
125 , m_Descriptor(descriptor)
127 , m_Variance(variance)
138 const char* name =
nullptr)
override 164 const char* name =
nullptr)
171 const char* name =
nullptr)
187 const char* name =
nullptr)
189 , m_Descriptor(descriptor)
190 , m_InputParams(params)
196 const char* name =
nullptr)
201 CheckInputParameters(params);
207 void CheckConstTensorPtrs(
const std::string& name,
const ConstTensor* expected,
const ConstTensor* actual);
219 const char* name =
nullptr)
221 , m_Descriptor(descriptor)
222 , m_InputParams(params)
228 const char* name =
nullptr)
233 CheckInputParameters(params);
239 void CheckConstTensorPtrs(
const std::string& name,
const ConstTensor* expected,
const ConstTensor* actual);
251 const char* name =
nullptr)
253 , m_InputParams(params)
258 const char* name =
nullptr)
262 CheckInputParameters(params);
267 void CheckConstTensorPtrs(
const std::string& name,
const ConstTensor* expected,
const ConstTensor* actual);
void CheckLayerName(const char *name)
void VisitQLstmLayer(const IConnectableLayer *layer, const QLstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr)
Function a QLstm layer should call back to when its Accept(ILayerVisitor&) function is invoked...
void VisitQuantizedLstmLayer(const IConnectableLayer *layer, const QuantizedLstmInputParams ¶ms, const char *name=nullptr)
Function a QuantizedLstm layer should call back to when its Accept(ILayerVisitor&) function is invoke...
Interface for a layer that is connectable to other layers via InputSlots and OutputSlots.
A Convolution2dDescriptor for the Convolution2dLayer.
void CheckLayerPointer(const IConnectableLayer *layer)
Copyright (c) 2021 ARM Limited and Contributors.
TestDepthwiseConvolution2dLayerVisitor(const DepthwiseConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
void CheckConstTensors(const ConstTensor &expected, const ConstTensor &actual)
TestBatchNormalizationLayerVisitor(const BatchNormalizationDescriptor &descriptor, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr)
An LstmDescriptor for the LstmLayer.
TestFullyConnectedLayerVistor(const FullyConnectedDescriptor &descriptor, const char *name=nullptr)
A FullyConnectedDescriptor for the FullyConnectedLayer.
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
TestConvolution2dLayerVisitor(const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
void VisitFullyConnectedLayer(const IConnectableLayer *layer, const FullyConnectedDescriptor &fullyConnectedDescriptor, const char *name=nullptr) override
Function that a fully connected layer should call back to when its Accept(ILayerVisitor&) function is...
TestConstantLayerVisitor(const ConstTensor &input, const char *name=nullptr)
void VisitBatchNormalizationLayer(const IConnectableLayer *layer, const BatchNormalizationDescriptor &descriptor, const ConstTensor &mean, const ConstTensor &variance, const ConstTensor &beta, const ConstTensor &gamma, const char *name=nullptr) override
Function that a batch normalization layer should call back to when its Accept(ILayerVisitor&) functio...
void CheckDescriptor(const Convolution2dDescriptor &convolution2dDescriptor)
A QLstmDescriptor for the QLstmLayer.
void VisitDepthwiseConvolution2dLayer(const IConnectableLayer *layer, const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
Function that a 2D depthwise convolution layer with biases should call back to when its Accept(ILayer...
virtual ~TestDepthwiseConvolution2dLayerVisitor()
void VisitLstmLayer(const IConnectableLayer *layer, const LstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr)
Function an Lstm layer should call back to when its Accept(ILayerVisitor&) function is invoked...
virtual ~TestFullyConnectedLayerVistor()
TestLstmLayerVisitor(const LstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr)
void VisitConvolution2dLayer(const IConnectableLayer *layer, const Convolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
Function that a 2D convolution layer should call back to when its Accept(ILayerVisitor&) function is ...
virtual ~TestConvolution2dLayerVisitor()
void CheckOptionalConstTensors(const Optional< ConstTensor > &expected, const Optional< ConstTensor > &actual)
TestQLstmLayerVisitor(const QLstmDescriptor &descriptor, const LstmInputParams ¶ms, const char *name=nullptr)
TestQuantizedLstmLayerVisitor(const QuantizedLstmInputParams ¶ms, const char *name=nullptr)
void VisitConstantLayer(const IConnectableLayer *layer, const ConstTensor &input, const char *name=nullptr)
Function a layer with no inputs and a single output, which always corresponds to the passed in consta...
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
A BatchNormalizationDescriptor for the BatchNormalizationLayer.