saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-01 23:30:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Hide Col2Im registration behind DML_TARGET_VERSION 6300 (#18829)

Browse source 
Hide Col2Im registration behind DML_TARGET_VERSION 6300

Co-authored-by: Sheil Kumar <sheilk@microsoft.com>
This commit is contained in:
Sheil Kumar 2023-12-14 14:20:55 -08:00 committed by Jeff Bloomfield
parent d2f7a5b128
commit b2f81c8725
2 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/DmlOperatorCol2Im.cpp
View file

@ -6,6 +6,8 @@
namespace Dml
{
#if DML_TARGET_VERSION >= 0x6300
class DmlOperatorCol2Im : public DmlOperator, public Col2ImHelper
{
public:
@ -56,4 +58,6 @@ public:
DML_OP_DEFINE_CREATION_FUNCTION(Col2Im, DmlOperatorCol2Im);
#endif // DML_TARGET_VERSION >= 0x6300
} // namespace Dml

7
onnxruntime/core/providers/dml/DmlExecutionProvider/src/Operators/OperatorRegistration.cpp
View file

@ -503,7 +503,11 @@ DML_OP_EXTERN_CREATION_FUNCTION(DynamicQuantizeLinear);
DML_OP_EXTERN_CREATION_FUNCTION(MatMulInteger);
DML_OP_EXTERN_CREATION_FUNCTION(ConvInteger);
DML_OP_EXTERN_CREATION_FUNCTION(Trilu);
#if DML_TARGET_VERSION >= 0x6300
DML_OP_EXTERN_CREATION_FUNCTION(Col2Im);
#endif
DML_OP_EXTERN_CREATION_FUNCTION(Shape);
DML_OP_EXTERN_CREATION_FUNCTION(Size);
DML_OP_EXTERN_CREATION_FUNCTION(Attention);
@ -771,7 +775,10 @@ constexpr static OperatorRegistrationInformation operatorRegistrationInformation
{REG_INFO( 16, ScatterND, typeNameListScatterGatherND, supportedTypeListScatterGatherND, DmlGraphSupport::Supported, requiredConstantCpuInputs(), std::nullopt, QueryScatter)},
{REG_INFO( 9, EyeLike, typeNameListEyeLike, supportedTypeListEyeLike, DmlGraphSupport::Supported)},
{REG_INFO( 14, Trilu, typeNameListDefault, supportedTypeListAllScalars, DmlGraphSupport::Supported, requiredConstantCpuInputs(1))},
#if DML_TARGET_VERSION >= 0x6300
{REG_INFO( 18, Col2Im, typeNameListDefault, supportedTypeListAllScalars, DmlGraphSupport::Supported, requiredConstantCpuInputs(1, 2))},
#endif
// Data reorganization that merely changes the dimensions while keeping the data identical.
{REG_INFO_COPY( 7, Identity, typeNameListDefault, supportedTypeListAllScalars, DmlGraphSupport::Supported)},