Searched refs:abs_kernel (Results 1 – 4 of 4) sorted by relevance
24 template <typename T> T abs_kernel(T x) { return std::abs(x); }); in abs_kernel_cuda()
19 gpu.launch_func @abs_kernel::@abs_kernel25 // CHECK-LABEL: gpu.module @abs_kernel26 gpu.module @abs_kernel {27 // CHECK-LABEL: llvm.func @abs_kernel32 …llvm.func @abs_kernel(%arg0: !llvm.ptr<f32>, %arg1: !llvm.ptr<f32>, %arg2: i64, %arg3: i64, %arg4:…
51 void abs_kernel(at::TensorIteratorBase& iter) { in abs_kernel() function139 REGISTER_PRIVATEUSE1_DISPATCH(abs_stub, &abs_kernel);
193 static void abs_kernel(TensorIteratorBase& iter) { in abs_kernel() function815 REGISTER_DISPATCH(abs_stub, &CPU_CAPABILITY::abs_kernel);