record_function_with_scope Class — pytorch Architecture

Source Code

aten/src/ATen/record_function.h lines 552–566

template <typename Inputs, typename... Args>
void record_function_with_scope(
    RecordFunction& guard,
    RecordFunction::FunctionDescriptor fn,
    const Inputs& inputs,
    Args&&... args) {
  if (guard.needsInputs()) {
    guard.before(
        fn,
        c10::ArrayRef<const c10::IValue>(inputs.data(), inputs.size()),
        std::forward<Args>(args)...);
  } else {
    guard.before(fn, std::forward<Args>(args)...);
  }
}