Reduce binary size
This commit is contained in:
parent
19c2da736e
commit
4f69f98357
@ -11,6 +11,13 @@
|
|||||||
|
|
||||||
namespace ov {
|
namespace ov {
|
||||||
namespace reference {
|
namespace reference {
|
||||||
|
namespace func {
|
||||||
|
template <class T>
|
||||||
|
T select(char s, T x, T y) {
|
||||||
|
return s ? x : y;
|
||||||
|
}
|
||||||
|
} // namespace func
|
||||||
|
|
||||||
template <typename T>
|
template <typename T>
|
||||||
void select(const char* arg0,
|
void select(const char* arg0,
|
||||||
const T* arg1,
|
const T* arg1,
|
||||||
@ -34,17 +41,7 @@ void select(const char* arg0,
|
|||||||
const Shape& arg1_shape,
|
const Shape& arg1_shape,
|
||||||
const Shape& arg2_shape,
|
const Shape& arg2_shape,
|
||||||
const op::AutoBroadcastSpec& broadcast_spec) {
|
const op::AutoBroadcastSpec& broadcast_spec) {
|
||||||
autobroadcast_select(arg0,
|
autobroadcast_select(arg0, arg1, arg2, out, arg0_shape, arg1_shape, arg2_shape, broadcast_spec, func::select<T>);
|
||||||
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);
|
|
||||||
});
|
|
||||||
}
|
}
|
||||||
} // namespace reference
|
} // namespace reference
|
||||||
} // namespace ov
|
} // namespace ov
|
||||||
|
Loading…
Reference in New Issue
Block a user