Skip to content
Snippets Groups Projects
Commit 7e5f09c4 authored by Olivier BICHLER's avatar Olivier BICHLER Committed by Maxence Naud
Browse files

Hotfix for more generic registry

parent 43f93816
No related branches found
No related tags found
1 merge request!132[UPD] version 0.4.1 -> 0.5.0
Hide whitespace changes
Inline Side-by-side
include/aidge/backend/cpu/operator/ResizeImpl_kernels.hpp
+ 19
− 18
View file @ 7e5f09c4
......@@ -99,30 +99,31 @@ void ResizeImpl_cpu_forward_kernel(
}
return;
}
// Kernels registration to implementation entry point
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int16},
{DataType::Float32},
{DataType::Float32},
{DataType::UInt64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::Int16}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int16_t>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int32},
{DataType::Float32},
{DataType::Float32},
{DataType::UInt64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::Int32}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int32_t>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Int64},
{DataType::Float32},
{DataType::Float32},
{DataType::Int64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::UInt64}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<int64_t>,
......@@ -130,27 +131,27 @@ REGISTRAR(ResizeImpl_cpu,
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float16},
{DataType::Float32},
{DataType::Float32},
{DataType::UInt64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::Float16}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<half_float::half>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float32},
{DataType::Float32},
{DataType::Float32},
{DataType::UInt64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::Float32}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<float>,
nullptr});
REGISTRAR(ResizeImpl_cpu,
{{{DataType::Float64},
{DataType::Float32},
{DataType::Float32},
{DataType::UInt64}},
{DataType::Any},
{DataType::Any},
{DataType::Any}},
{DataType::Float64}},
{ProdConso::inPlaceModel,
ResizeImpl_cpu_forward_kernel<double>,
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

Copyright © Eclipse Foundation, Inc. All Rights Reserved.     Privacy Policy | Terms of Use | Copyright Agent