C10，來自于Caffe Tensor Library的縮寫
ATen，來自于 A TENsor library for C++11的縮寫
c10_utils.h

#pragma once#include <ATen/core/dispatch/Dispatcher.h>template <class... Inputs> inline std::vector<c10::IValue> makeStack(Inputs&&... inputs) {return {std::forward<Inputs>(inputs)...}; }template <class... Args> inline std::vector<c10::IValue> callOp(const c10::OperatorHandle& op,Args... args) {auto stack = makeStack(std::forward<Args>(args)...);c10::Dispatcher::singleton().callBoxed(op, &stack);return stack; }template <class... Args> inline std::vector<c10::IValue> callOp(const char* func_name,const char* overload_name,Args... args) {const c10::optional<c10::OperatorHandle> op_handle =c10::Dispatcher::singleton().findSchema({func_name, overload_name});assert(op_handle.has_value());return callOp(op_handle.value(), args...); }

Macros.h

#pragma once #include <c10/macros/Macros.h>

Array.h

#pragma once// A fixed-size array type usable from both host and // device code.#include <c10/macros/Macros.h>namespace at { namespace detail {template <typename T, int size> struct alignas(16) Array {T data[size];C10_HOST_DEVICE T operator[](int i) const {return data[i];}C10_HOST_DEVICE T& operator[](int i) {return data[i];} #ifdef __HIP_PLATFORM_HCC__C10_HOST_DEVICE Array() = default;C10_HOST_DEVICE Array(const Array&) = default;C10_HOST_DEVICE Array& operator=(const Array&) = default; #elseArray() = default;Array(const Array&) = default;Array& operator=(const Array&) = default; #endif// Fill the array with x.C10_HOST_DEVICE Array(T x) {for (int i = 0; i < size; i++) {data[i] = x;}} };}}

總結(jié)

以上是生活随笔為你收集整理的ATen(A TENsor library for C++11)剖析(1)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。