Searched refs:softplus_backward_kernel (Results 1 – 2 of 2) sorted by relevance
44 void softplus_backward_kernel( in softplus_backward_kernel() function72 REGISTER_DISPATCH(softplus_backward_stub, &softplus_backward_kernel);
1082 void softplus_backward_kernel(TensorIteratorBase& iter, const Scalar& beta_, const Scalar& threshol… in softplus_backward_kernel() function1427 ALSO_REGISTER_AVX512_DISPATCH(softplus_backward_stub, &softplus_backward_kernel);