Tensor Comprehensions
|
This is the complete list of members for tc::ExecutionEngine, including all inherited members.
addTC(const std::string &tc) | tc::ExecutionEngine | |
clear(size_t handle) | tc::ExecutionEngine | |
compile(const std::string &name, const std::vector< const DLTensor * > &inputs, const MappingOptions &options) | tc::ExecutionEngine | |
define(const std::string &language) | tc::ExecutionEngine | |
define(const std::vector< lang::TreeRef > &treeRefs) | tc::ExecutionEngine | |
emplaceExecutor(std::unique_ptr< ExecutorInfo > p) | tc::ExecutionEngine | private |
ExecutionEngine()=default | tc::ExecutionEngine | |
executorInfoMutex | tc::ExecutionEngine | private |
executors_ | tc::ExecutionEngine | private |
getHandle(const std::string &name, const std::vector< const DLTensor * > &inputsInfo, const MappingOptions &options) | tc::ExecutionEngine | private |
inferOutputTensorInfo(const std::string &name, const std::vector< const DLTensor * > &inTensorPtrs) | tc::ExecutionEngine | |
makeExecutorInfo(const std::string &name, const std::vector< const DLTensor * > &inputsInfo, const MappingOptions &options) | tc::ExecutionEngine | private |
run(size_t handle, const std::vector< const DLTensor * > &inputs, const std::vector< DLTensor * > &outputs, bool profile=false, std::function< bool(const ExecutorInfo *)> pruningFunction=[](const ExecutorInfo *){return false;}) | tc::ExecutionEngine | |
tcNameMap_ | tc::ExecutionEngine | private |
treeForFunction(const std::string &name) | tc::ExecutionEngine | inline |
uidCounter | tc::ExecutionEngine | private |
uncheckedRun(size_t handle, const std::vector< const void * > &inputs, const std::vector< void * > &outputs) | tc::ExecutionEngine |