#include <tc_executor.h>
|
| static const size_t | InvalidHandle = std::numeric_limits<size_t>::max() |
| |
| tc::TcExecutor::TcExecutor |
( |
const std::string & |
TCDefinition, |
|
|
const std::vector< const DLTensor * > & |
inputsInfo |
|
) |
| |
| tc::TcExecutor::TcExecutor |
( |
lang::TreeRef |
TCDefinition, |
|
|
const std::vector< const DLTensor * > & |
inputsInfo |
|
) |
| |
| tc::TcExecutor::~TcExecutor |
( |
| ) |
|
| Block tc::TcExecutor::block |
( |
| ) |
const |
|
inline |
| void tc::TcExecutor::checkInputsCompliant |
( |
const std::vector< const DLTensor * > & |
inputsInfo | ) |
const |
|
private |
| void tc::TcExecutor::clearRTC |
( |
| ) |
|
|
inline |
| void tc::TcExecutor::compileWithTcMapper |
( |
| ) |
|
|
private |
| std::string tc::TcExecutor::getCudaSource |
( |
| ) |
|
|
inline |
| Grid tc::TcExecutor::grid |
( |
| ) |
const |
|
inline |
| bool tc::TcExecutor::hasRTCFun |
( |
| ) |
|
|
inline |
| std::vector<const DLTensor*> tc::TcExecutor::inferOutputTensorInfo |
( |
| ) |
|
| std::string tc::TcExecutor::kernelName |
( |
| ) |
const |
|
inline |
| Duration tc::TcExecutor::run |
( |
const std::vector< const DLTensor * > & |
inputs, |
|
|
const std::vector< DLTensor * > & |
outputs, |
|
|
bool |
profile = false |
|
) |
| const |
| void tc::TcExecutor::uncheckedRun |
( |
const std::vector< const void * > & |
inputs, |
|
|
const std::vector< void * > & |
outputs |
|
) |
| const |
| isl::ctx tc::TcExecutor::ctx_ |
|
mutableprivate |
| const size_t tc::TcExecutor::InvalidHandle = std::numeric_limits<size_t>::max() |
|
static |
The documentation for this class was generated from the following file: