Searched refs:compare_kernel_with_scalars (Results 1 – 1 of 1) sorted by relevance
75 C10_NOINLINE void compare_kernel_with_scalars(TensorIteratorBase &iter, OpType op) { in compare_kernel_with_scalars() function83 compare_kernel_with_scalars(iter, OpType::GE); in ge_kernel_cuda()87 compare_kernel_with_scalars(iter, OpType::GT); in gt_kernel_cuda()91 compare_kernel_with_scalars(iter, OpType::LE); in le_kernel_cuda()95 compare_kernel_with_scalars(iter, OpType::LT); in lt_kernel_cuda()