mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-30 20:18:08 +00:00
Modify the kernel declaration for Shrink op (#1554)
* Add capability for the input and output of Shrink op to share a commong buffer * Cosmetic change
This commit is contained in:
parent
6c271c63ac
commit
ceb8f1c1a2
2 changed files with 4 additions and 1 deletions
|
|
@ -11,7 +11,9 @@ namespace onnxruntime {
|
|||
ONNX_CPU_OPERATOR_KERNEL(
|
||||
Shrink,
|
||||
9,
|
||||
KernelDefBuilder().TypeConstraint("T", DataTypeImpl::AllNumericTensorTypes()),
|
||||
KernelDefBuilder()
|
||||
.MayInplace(0, 0)
|
||||
.TypeConstraint("T", DataTypeImpl::AllNumericTensorTypes()),
|
||||
Shrink);
|
||||
|
||||
namespace shrink_internal {
|
||||
|
|
|
|||
|
|
@ -17,6 +17,7 @@ namespace cuda {
|
|||
T, \
|
||||
kCudaExecutionProvider, \
|
||||
KernelDefBuilder() \
|
||||
.MayInplace(0, 0) \
|
||||
.TypeConstraint("T", DataTypeImpl::GetTensorType<T>()), \
|
||||
Shrink<T>);
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue