Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
torch / include / ATen / native / cuda / ReduceOps.h
Size: Mime:

namespace at {
struct TensorIterator;
}

namespace c10 {
class Scalar;
}

namespace at { namespace native {

void norm_launch_kernel(TensorIterator &iter, double val);
void min_launch_kernel(TensorIterator &iter);
void max_launch_kernel(TensorIterator &iter);
void aminmax_launch_kernel(TensorIterator &iter);
void min_all_launch_kernel(TensorIterator &iter);
void max_all_launch_kernel(TensorIterator &iter);
void aminmax_allreduce_launch_kernel(TensorIterator &iter);

}}  // namespace at::native