MCPcopy Create free account
hub / github.com/ARM-software/armnn / VisitConvolutionOperator

Function VisitConvolutionOperator

delegate/opaque/src/Convolution.hpp:967–986  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

965}
966
967TfLiteStatus VisitConvolutionOperator(DelegateData& delegateData,
968 TfLiteOpaqueContext* tfLiteContext,
969 TfLiteOpaqueNode* tfLiteNode,
970 int nodeIndex,
971 int32_t operatorCode)
972{
973 switch(operatorCode)
974 {
975 case kTfLiteBuiltinConv2d:
976 return VisitConv2dOperator(delegateData, tfLiteContext, tfLiteNode, nodeIndex, operatorCode);
977 case kTfLiteBuiltinConv3d:
978 return VisitConv3dOperator(delegateData, tfLiteContext, tfLiteNode, nodeIndex, operatorCode);
979 case kTfLiteBuiltinDepthwiseConv2d:
980 return VisitDepthwiseConv2dOperator(delegateData, tfLiteContext, tfLiteNode, nodeIndex, operatorCode);
981 case kTfLiteBuiltinTransposeConv:
982 return VisitTransposeConv2dOperator(delegateData, tfLiteContext, tfLiteNode, nodeIndex, operatorCode);
983 default:
984 return kTfLiteError;
985 }
986}
987
988}

Callers 1

VisitNodeMethod · 0.70

Calls 4

VisitConv2dOperatorFunction · 0.70
VisitConv3dOperatorFunction · 0.70

Tested by

no test coverage detected