Searched refs:m_allocation_index (Results 1 – 1 of 1) sorted by relevance
528 : m_device(device), m_allocation_index(0) {} in TensorBlockScratchAllocator()542 const bool has_allocation = m_allocation_index < num_allocations; in allocate()545 eigen_assert(m_allocation_index <= num_allocations); in allocate()552 if (has_allocation && m_allocations[m_allocation_index].size < size) { in allocate()553 m_device.deallocate(m_allocations[m_allocation_index].ptr); in allocate()554 m_allocations[m_allocation_index].ptr = m_device.allocate(size); in allocate()555 m_allocations[m_allocation_index].size = size; in allocate()566 eigen_assert(m_allocations[m_allocation_index].ptr != NULL); in allocate()567 eigen_assert(m_allocations[m_allocation_index].size >= size); in allocate()569 return m_allocations[m_allocation_index++].ptr; in allocate()[all …]