Searched refs:leaky_relu_backward_kernel (Results 1 – 2 of 2) sorted by relevance
38 void leaky_relu_backward_kernel( in leaky_relu_backward_kernel() function60 REGISTER_DISPATCH(leaky_relu_backward_stub, &leaky_relu_backward_kernel);
999 static void leaky_relu_backward_kernel(TensorIteratorBase& iter, const Scalar& negval_) { in leaky_relu_backward_kernel() function1407 REGISTER_DISPATCH(leaky_relu_backward_stub, &leaky_relu_backward_kernel);