Home / Class/ setArg Class — pytorch Architecture

setArg Class — pytorch Architecture

Architecture documentation for the setArg class in MetalShaderLibrary.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/mps/MetalShaderLibrary.h lines 78–86

  template <
      typename Container,
      typename = std::enable_if_t<detail::has_size_type_v<Container>>>
  inline void setArg(unsigned idx, const Container& values) {
    setArg(
        idx,
        values.data(),
        values.size() * sizeof(typename Container::value_type));
  }

Source

Analyze Your Own Codebase

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

Try Supermodel Free