@@ -193,9 +193,8 @@ inline void dtype_specialized_elementwise_fn_impl(
193193 });
194194}
195195
196- template <typename CTYPE_COMPUTE, typename Op, typename ... Args>
196+ template <typename CTYPE_COMPUTE, typename ... Args>
197197inline bool validate_elementwise_fn_inputs (
198- const Op& compute_fun,
199198 KernelRuntimeContext& ctx,
200199 const Tensor& out,
201200 SupportedTensorDtypes out_dtypes,
@@ -288,7 +287,7 @@ inline void apply_elementwise_fn_runtime_out_dtypes(
288287 SupportedTensorDtypes out_dtypes,
289288 Args... inputs) {
290289 const bool inputs_valid = validate_elementwise_fn_inputs<CTYPE_COMPUTE>(
291- compute_fun, ctx, out, out_dtypes, inputs...);
290+ ctx, out, out_dtypes, inputs...);
292291 if (!inputs_valid) {
293292 return ;
294293 }
@@ -313,7 +312,7 @@ inline void apply_elementwise_fn(
313312 const Tensor& out,
314313 Args... inputs) {
315314 const bool inputs_valid = validate_elementwise_fn_inputs<CTYPE_COMPUTE>(
316- compute_fun, ctx, out, out_dtypes, inputs...);
315+ ctx, out, out_dtypes, inputs...);
317316 if (!inputs_valid) {
318317 return ;
319318 }
0 commit comments