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 95 const char* name =
nullptr)
97 , m_Descriptor(descriptor)
108 const char* name =
nullptr)
override 133 const char* name =
nullptr)
135 , m_Descriptor(descriptor)
137 , m_Variance(variance)
148 const char* name =
nullptr)
override 174 const char* name =
nullptr)
181 const char* name =
nullptr)
197 const char* name =
nullptr)
199 , m_Descriptor(descriptor)
200 , m_InputParams(params)
206 const char* name =
nullptr)
211 CheckInputParameters(params);
217 void CheckConstTensorPtrs(
const std::string& name,
const ConstTensor* expected,
const ConstTensor* actual);
229 const char* name =
nullptr)
231 , m_InputParams(params)
236 const char* name =
nullptr)
240 CheckInputParameters(params);
245 void CheckConstTensorPtrs(
const std::string& name,
const ConstTensor* expected,
const ConstTensor* actual);
void CheckLayerName(const char *name)
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) 2020 ARM Limited.
TestDepthwiseConvolution2dLayerVisitor(const DepthwiseConvolution2dDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)
void CheckConstTensors(const ConstTensor &expected, const ConstTensor &actual)
TestFullyConnectedLayerVistor(const FullyConnectedDescriptor &descriptor, const ConstTensor &weights, const Optional< ConstTensor > biases, const char *name=nullptr)
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.
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)
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)
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()
void VisitFullyConnectedLayer(const IConnectableLayer *layer, const FullyConnectedDescriptor &fullyConnectedDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr) override
Function that a fully connected layer should call back to when its Accept(ILayerVisitor&) function is...
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)
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.