![]() |
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 |