unary_op_impl_float Class — pytorch Architecture

Architecture documentation for the unary_op_impl_float class in UnaryOps.cpp from the pytorch codebase.

Class cpp

Entity Profile

Source Code

aten/src/ATen/native/UnaryOps.cpp lines 418–424

template <typename Stub, typename ...Args>
static inline Tensor unary_op_impl_float(const Tensor& self, Stub& stub, Args... args) {
  Tensor result;
  auto iter = TensorIterator::unary_float_op(result, self);
  stub(iter.device_type(), iter, args...);
  return iter.output();
}

Source

View on GitHub

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free