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

Method IsConvolution2dSupported

src/backends/cl/ClLayerSupport.cpp:904–935  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

902}
903
904bool ClLayerSupport::IsConvolution2dSupported(const TensorInfo& input,
905 const TensorInfo& output,
906 const Convolution2dDescriptor& descriptor,
907 const TensorInfo& weights,
908 const Optional<TensorInfo>& biases,
909 Optional<std::string&> reasonIfUnsupported) const
910{
911 bool isFastMathEnabled = false;
912#if defined(ARMCOMPUTECL_ENABLED)
913 if (m_ModelContextPtr)
914 {
915 if (m_ModelContextPtr.get() != nullptr)
916 {
917 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
918 if (modelOptions)
919 {
920 isFastMathEnabled = modelOptions->IsFastMathEnabled();
921 }
922 }
923 }
924#endif
925
926 FORWARD_WORKLOAD_VALIDATE_FUNC(ClConvolution2dWorkloadValidate,
927 reasonIfUnsupported,
928 input,
929 output,
930 descriptor,
931 weights,
932 biases,
933 isFastMathEnabled,
934 nullptr);
935}
936
937bool ClLayerSupport::IsConvolution3dSupported(const TensorInfo& input,
938 const TensorInfo& output,

Callers 1

Calls 2

getMethod · 0.45
IsFastMathEnabledMethod · 0.45

Tested by

no test coverage detected