#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: