Reduce binary size

This commit is contained in:
Tomasz Jankowski 2023-11-23 11:05:19 +01:00
parent 19c2da736e
commit 4f69f98357

View File

@ -11,6 +11,13 @@
namespace ov {
namespace reference {
namespace func {
template <class T>
T select(char s, T x, T y) {
return s ? x : y;
}
} // namespace func
template <typename T>
void select(const char* arg0,
const T* arg1,
@ -34,17 +41,7 @@ void select(const char* arg0,
const Shape& arg1_shape,
const Shape& arg2_shape,
const op::AutoBroadcastSpec& broadcast_spec) {
autobroadcast_select(arg0,
arg1,
arg2,
out,
arg0_shape,
arg1_shape,
arg2_shape,
broadcast_spec,
[](char s, T x, T y) -> T {
return static_cast<T>(s ? x : y);
});
autobroadcast_select(arg0, arg1, arg2, out, arg0_shape, arg1_shape, arg2_shape, broadcast_spec, func::select<T>);
}
} // namespace reference
} // namespace ov