ONNX Runtime
|
This is the complete list of members for Ort::KernelContext, including all inherited members.
GetAllocator(const OrtMemoryInfo &memory_info) const | Ort::KernelContext | |
GetGPUComputeStream() const | Ort::KernelContext | |
GetInput(size_t index) const | Ort::KernelContext | |
GetInputCount() const | Ort::KernelContext | |
GetLogger() const | Ort::KernelContext | |
GetOrtKernelContext() const | Ort::KernelContext | inline |
GetOutput(size_t index, const int64_t *dim_values, size_t dim_count) const | Ort::KernelContext | |
GetOutput(size_t index, const std::vector< int64_t > &dims) const | Ort::KernelContext | |
GetOutputCount() const | Ort::KernelContext | |
KernelContext(OrtKernelContext *context) | Ort::KernelContext | explicit |
ParallelFor(void(*fn)(void *, size_t), size_t total, size_t num_batch, void *usr_data) const | Ort::KernelContext |