#include <TensorBlock.h>
Classes | |
struct | Allocation |
Public Member Functions | |
void * | allocate (size_t size) |
void | reset () |
TensorBlockScratchAllocator (const Device &device) | |
~TensorBlockScratchAllocator () | |
Private Attributes | |
int | m_allocation_index |
std::vector< Allocation > | m_allocations |
const Device & | m_device |
Definition at line 525 of file TensorBlock.h.
|
inlineexplicit |
Definition at line 527 of file TensorBlock.h.
|
inline |
Definition at line 530 of file TensorBlock.h.
|
inline |
Definition at line 536 of file TensorBlock.h.
|
inline |
Definition at line 572 of file TensorBlock.h.
|
private |
Definition at line 581 of file TensorBlock.h.
|
private |
Definition at line 583 of file TensorBlock.h.
|
private |
Definition at line 580 of file TensorBlock.h.