22 #include <llvm/IR/LLVMContext.h>
23 #include <llvm/IR/Type.h>
31 llvm::Function* llvm_function,
33 const std::unordered_map<const Function*, llvm::Function*>&
f);
void translate_function(const Function *function, llvm::Function *llvm_function, const ReductionCode &reduction_code, const std::unordered_map< const Function *, llvm::Function * > &f)
llvm::Type * llvm_type(const Type type, llvm::LLVMContext &ctx)
torch::Tensor f(torch::Tensor x, torch::Tensor W_target, torch::Tensor b_target)