Skip to content

Commit 9eddfbb

Browse files
committed
Change binary op to use raw_buffer
PiperOrigin-RevId: 838600926
1 parent f13e1b9 commit 9eddfbb

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

ynnpack/subgraph/elementwise.cc

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -180,10 +180,8 @@ auto make_unary_elementwise_impl(unary_kernel_fn kernel) {
180180
// data, this would need to change.
181181
auto make_binary_elementwise_impl(binary_kernel_fn kernel,
182182
const binary_params& params) {
183-
return [kernel, params](
184-
slinky::buffer<const void, YNN_MAX_TENSOR_RANK> a,
185-
slinky::buffer<const void, YNN_MAX_TENSOR_RANK> b,
186-
slinky::buffer<void, YNN_MAX_TENSOR_RANK> x) -> slinky::index_t {
183+
return [kernel, params](slinky::raw_buffer a, slinky::raw_buffer b,
184+
slinky::raw_buffer x) -> slinky::index_t {
187185
slinky::dim a_dims[2], b_dims[2], x_dims[2];
188186

189187
for (int i = 0; i < 2; ++i) {

0 commit comments

Comments
 (0)