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

Method IsConvolution3dSupported

src/backends/cl/ClLayerSupport.cpp:937–968  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

935}
936
937bool ClLayerSupport::IsConvolution3dSupported(const TensorInfo& input,
938 const TensorInfo& output,
939 const Convolution3dDescriptor& descriptor,
940 const TensorInfo& weights,
941 const Optional<TensorInfo>& biases,
942 Optional<std::string&> reasonIfUnsupported) const
943{
944 bool isFastMathEnabled = false;
945#if defined(ARMCOMPUTECL_ENABLED)
946 if (m_ModelContextPtr)
947{
948 if (m_ModelContextPtr.get() != nullptr)
949 {
950 auto modelOptions = dynamic_cast<ClBackendModelContext*>(m_ModelContextPtr.get());
951 if (modelOptions)
952 {
953 isFastMathEnabled = modelOptions->IsFastMathEnabled();
954 }
955 }
956}
957#endif
958
959 FORWARD_WORKLOAD_VALIDATE_FUNC(ClConvolution3dWorkloadValidate,
960 reasonIfUnsupported,
961 input,
962 output,
963 descriptor,
964 weights,
965 biases,
966 isFastMathEnabled,
967 nullptr);
968}
969
970bool ClLayerSupport::IsDequantizeSupported(const TensorInfo& input,
971 const TensorInfo& output,

Callers 1

Calls 2

getMethod · 0.45
IsFastMathEnabledMethod · 0.45

Tested by

no test coverage detected