Searched refs:scratch_buffer_size (Results 1 – 5 of 5) sorted by relevance
158 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(lhs_rank); in InitializeTemporaries() local160 scratch_buffer_size->data[i] = lhs->dims->data[i]; in InitializeTemporaries()163 scratch_buffer_size->data[lhs_rank - 2] = lhs->dims->data[lhs_rank - 1]; in InitializeTemporaries()164 scratch_buffer_size->data[lhs_rank - 1] = lhs->dims->data[lhs_rank - 2]; in InitializeTemporaries()169 scratch_buffer_size)); in InitializeTemporaries()183 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(rhs_rank); in InitializeTemporaries() local185 scratch_buffer_size->data[i] = rhs->dims->data[i]; in InitializeTemporaries()188 scratch_buffer_size->data[rhs_rank - 2] = rhs->dims->data[rhs_rank - 1]; in InitializeTemporaries()189 scratch_buffer_size->data[rhs_rank - 1] = rhs->dims->data[rhs_rank - 2]; in InitializeTemporaries()198 scratch_buffer_size)); in InitializeTemporaries()
971 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local972 scratch_buffer_size->data[0] = n_batch; in Prepare()975 scratch_buffer_size->data[1] = n_cell * 3; in Prepare()978 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()981 scratch_buffer_size)); in Prepare()1197 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1198 scratch_buffer_size->data[0] = n_batch; in Prepare()1199 scratch_buffer_size->data[1] = n_cell; in Prepare()1202 scratch_buffer_size)); in Prepare()
1418 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1419 scratch_buffer_size->data[0] = n_batch; in Prepare()1422 scratch_buffer_size->data[1] = n_cell * 3; in Prepare()1425 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()1428 scratch_buffer_size)); in Prepare()1772 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1773 scratch_buffer_size->data[0] = n_batch; in Prepare()1774 scratch_buffer_size->data[1] = n_cell; in Prepare()1777 scratch_buffer_size)); in Prepare()1810 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local[all …]
409 size_t scratch_buffer_size = Eigen::kGpuScratchSize + sizeof(unsigned int); in InitScratchBuffers() local412 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()419 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()