Tensor Comprehensions
|
Public Attributes | |
std::string | kernelName |
std::vector < dlutils::DLTensorUPtr > | inputsInfo |
std::vector < dlutils::DLTensorUPtr > | outputsInfo |
std::vector< int > | kernelParams |
std::string | kernelSpecializedName |
std::unique_ptr < tc::MappingOptions > | options |
std::string | cudaSource |
Grid | grid {{0, 0, 0}} |
Block | block {{0, 0, 0}} |
std::shared_ptr< CudaRTCFunction > | rtcFun |
Block tc::TcExecutor::TcExecutionInfo::block {{0, 0, 0}} |
std::string tc::TcExecutor::TcExecutionInfo::cudaSource |
Grid tc::TcExecutor::TcExecutionInfo::grid {{0, 0, 0}} |
std::vector<dlutils::DLTensorUPtr> tc::TcExecutor::TcExecutionInfo::inputsInfo |
std::string tc::TcExecutor::TcExecutionInfo::kernelName |
std::vector<int> tc::TcExecutor::TcExecutionInfo::kernelParams |
std::string tc::TcExecutor::TcExecutionInfo::kernelSpecializedName |
std::unique_ptr<tc::MappingOptions> tc::TcExecutor::TcExecutionInfo::options |
std::vector<dlutils::DLTensorUPtr> tc::TcExecutor::TcExecutionInfo::outputsInfo |
std::shared_ptr<CudaRTCFunction> tc::TcExecutor::TcExecutionInfo::rtcFun |