Searched defs:need_acc (Results 1 – 3 of 3) sorted by relevance
152 inline void create_acc_buffer(Tensor& buffer, const Tensor& self, bool need_acc) { in create_acc_buffer()170 bool need_acc = isReducedFloatingType(self.scalar_type()); in operator ()() local267 bool need_acc = isReducedFloatingType(self.scalar_type()); in operator ()() local363 bool need_acc = isReducedFloatingType(self.scalar_type()); in operator ()() local458 bool need_acc = isReducedFloatingType(self.scalar_type()); in operator ()() local554 bool need_acc = isReducedFloatingType(self.scalar_type()); in operator ()() local768 static constexpr bool need_acc = is_reduced_floating_point_v<scalar_t>; in cpu_scatter_reduce_expanded_index() local
95 static constexpr bool need_acc = is_reduced_floating_point_v<scalar_t>; in spmm_reduce_kernel_impl() local193 static constexpr bool need_acc = is_reduced_floating_point_v<scalar_t>; in spmm_reduce_arg_kernel_impl() local
418 constexpr bool need_acc = !std::is_same_v<scalar_t, acc_t>; variable