This commit applies `lws_hint()` in the case of below operators.
TensorFlowMax, TensorFlowSum, EXP, GATHER, CAST, PAD
Signed-off-by: Jiyoung Yun <jy910.yun@samsung.com>
unsigned int idx = 0;
add_3D_tensor_argument(idx, _input, slice);
add_3D_tensor_argument(idx, _output, slice);
- enqueue(queue, *this, slice);
+ enqueue(queue, *this, slice, lws_hint());
} while (collapsed.slide_window_slice_3D(slice));
}
unsigned int idx = 0;
add_3D_tensor_argument(idx, _input, slice);
add_3D_tensor_argument(idx, _output, slice);
- enqueue(queue, *this, slice);
+ enqueue(queue, *this, slice, lws_hint());
} while (collapsed.slide_window_slice_3D(slice));
}
add_4D_tensor_argument(idx, _input, window_collapsed);
add_3D_tensor_argument(idx, _indices, window_collapsed);
add_4D_tensor_argument(idx, _output, window_collapsed);
- enqueue(queue, *this, window_collapsed);
+ enqueue(queue, *this, window_collapsed, lws_hint());
}
add_4D_tensor_argument(idx, _input, slice_in);
add_4D_tensor_argument(idx, _output, slice_out);
_kernel.setArg<cl_int4>(idx++, paddingValues);
- enqueue(queue, *this, slice_out);
+ enqueue(queue, *this, slice_out, lws_hint());
} while (window.slide_window_slice_4D(slice_out) && window.slide_window_slice_4D(slice_in));
}
idx = 0;
add_4D_tensor_argument(idx, _input, slice_in);
add_4D_tensor_argument(idx, _output, slice_out);
- enqueue(queue, *this, slice_out);
+ enqueue(queue, *this, slice_out, lws_hint());
// Recover output's shape of output tensor
_output->info()->set_tensor_shape(shape_out);