Home / Class/ mtl_setArg Class — pytorch Architecture

mtl_setArg Class — pytorch Architecture

Architecture documentation for the mtl_setArg class in OperationUtils.h from the pytorch codebase.

Class c

Entity Profile

Source Code

aten/src/ATen/native/mps/OperationUtils.h lines 521–524

template <typename T>
inline void mtl_setArg(id<MTLComputeCommandEncoder> encoder, const T& val, unsigned idx) {
  mtl_setBytes(encoder, val, idx);
}

Source

Analyze Your Own Codebase

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

Try Supermodel Free