Reduce binary size
This commit is contained in:
parent
19c2da736e
commit
4f69f98357
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user