Searched refs:scratch_buffer_size (Results 1 – 5 of 5) sorted by relevance
160 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(lhs_rank); in InitializeTemporaries() local162 scratch_buffer_size->data[i] = lhs->dims->data[i]; in InitializeTemporaries()165 scratch_buffer_size->data[lhs_rank - 2] = lhs->dims->data[lhs_rank - 1]; in InitializeTemporaries()166 scratch_buffer_size->data[lhs_rank - 1] = lhs->dims->data[lhs_rank - 2]; in InitializeTemporaries()171 scratch_buffer_size)); in InitializeTemporaries()186 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(rhs_rank); in InitializeTemporaries() local188 scratch_buffer_size->data[i] = rhs->dims->data[i]; in InitializeTemporaries()191 scratch_buffer_size->data[rhs_rank - 2] = rhs->dims->data[rhs_rank - 1]; in InitializeTemporaries()192 scratch_buffer_size->data[rhs_rank - 1] = rhs->dims->data[rhs_rank - 2]; in InitializeTemporaries()201 scratch_buffer_size)); in InitializeTemporaries()
972 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local973 scratch_buffer_size->data[0] = n_batch; in Prepare()977 scratch_buffer_size->data[1] = n_cell * 4 + 16; in Prepare()981 scratch_buffer_size->data[1] = n_cell * 5 + 16; in Prepare()984 scratch_buffer_size)); in Prepare()1201 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1202 scratch_buffer_size->data[0] = n_batch; in Prepare()1203 scratch_buffer_size->data[1] = n_cell; in Prepare()1206 scratch_buffer_size)); in Prepare()
1419 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1420 scratch_buffer_size->data[0] = n_batch; in Prepare()1424 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()1428 scratch_buffer_size->data[1] = n_cell * 5; in Prepare()1431 scratch_buffer_size)); in Prepare()1776 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1777 scratch_buffer_size->data[0] = n_batch; in Prepare()1778 scratch_buffer_size->data[1] = n_cell; in Prepare()1781 scratch_buffer_size)); in Prepare()1814 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local[all …]
418 size_t scratch_buffer_size = Eigen::kGpuScratchSize + sizeof(unsigned int); in InitScratchBuffers() local421 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()428 se::DeviceMemoryBase(scratch_buffer, scratch_buffer_size)); in InitScratchBuffers()
689 …VkDeviceSize scratch_buffer_size = std::max(bot_level_as.build_scratch_memory_requirements().memor… in TEST_F() local692 …scratch_buffer.init(*m_device, scratch_buffer_size, VK_MEMORY_PROPERTY_DEVICE_LOCAL_BIT, VK_BUFFER… in TEST_F()