onnxruntime/onnxruntime
George Nash b4e8e9b004
Add DnnlOpManager (#7521)
* Add  DnnlOpManager

The DnnlOpManager is able to more accurately check if a node is
supported by the DNNLExecutionProvider.

The DNNLExecutionProvider::GetCapability function has been updated
to use the DnnlOpManager.

This commit adds the ability to check if data type, attributes,
and tensor dimensions of the node are supported.

The IsDimensionSupported function is no longer needed since the checks
it was doing have been moved into the individual implementations of
the virtual class DnnlNodeCapability.

Signed-off-by: George Nash <george.nash@intel.com>

* Fix AveragePool entry in the DnnlOpManager

Added check for ceil_mode attribute in the PoolNodeCapability
check.  DnnlExecutionProvider does not support ceil_mode other
than the default value.

Signed-off-by: George Nash <george.nash@intel.com>
2021-05-12 22:04:26 -05:00
..
contrib_ops Implement NCHWc Upsample linear mode (#7623) 2021-05-10 12:16:16 -07:00
core Add DnnlOpManager (#7521) 2021-05-12 22:04:26 -05:00
featurizers_ops/cpu Consolidate MLTypeCallDispatcher classes (#6651) 2021-02-12 13:26:56 -08:00
gsl
python Add ATenOp and call aten::embedding and its Backward Op from ORT (#7590) 2021-05-13 09:24:27 +08:00
test Add FromProviderOptions()/ToProviderOptions() for TensorRT EP (#7654) 2021-05-12 15:26:27 -07:00
tool/etw
wasm [js/web] support multi-thread for wasm backend (#7601) 2021-05-07 12:12:37 -07:00
.style.yapf
__init__.py Add robust dependency check for Python package (#6436) 2021-02-21 15:11:28 -08:00
ReformatSource.ps1
ReformatSourcePython.bat
VSCodeCoverage.runsettings