Tensor Comprehensions
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
tc::ATenCompilationUnit Member List

This is the complete list of members for tc::ATenCompilationUnit, including all inherited members.

ATenCompilationUnit()tc::ATenCompilationUnitexplicit
compile(const std::string &name, const std::vector< at::Tensor > &inputs, const MappingOptions &options)tc::ATenCompilationUnit
define(const std::string &language)tc::ATenCompilationUnit
executionEngine_tc::ATenCompilationUnitprivate
inferOutputTensorInfo(const std::string &name, const std::vector< at::Tensor > &inputs)tc::ATenCompilationUnit
run(const std::string &name, const std::vector< at::Tensor > &inputs, std::vector< at::Tensor > &outputs, size_t handle, bool profile=false)tc::ATenCompilationUnit
uncheckedRun(const std::vector< at::Tensor > &inputs, std::vector< at::Tensor > &outputs, size_t handle)tc::ATenCompilationUnit