Searched refs:scratch_buffer_size (Results 1 – 8 of 8) 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()
962 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local963 scratch_buffer_size->data[0] = n_batch; in Prepare()966 scratch_buffer_size->data[1] = n_cell * 3; in Prepare()969 scratch_buffer_size->data[1] = n_cell * 4; in Prepare()972 scratch_buffer_size)); in Prepare()1188 TfLiteIntArray* scratch_buffer_size = TfLiteIntArrayCreate(2); in Prepare() local1189 scratch_buffer_size->data[0] = n_batch; in Prepare()1190 scratch_buffer_size->data[1] = n_cell; in Prepare()1193 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 …]
264 constexpr int scratch_buffer_size = 40; in TF_LITE_MICRO_TEST() local265 unsigned char scratch_buffer[scratch_buffer_size]; in TF_LITE_MICRO_TEST()266 tflite::GreedyMemoryPlanner planner(scratch_buffer, scratch_buffer_size); in TF_LITE_MICRO_TEST()
57 GreedyMemoryPlanner(unsigned char* scratch_buffer, int scratch_buffer_size);
42 int scratch_buffer_size) in GreedyMemoryPlanner() argument45 max_buffer_count_ = scratch_buffer_size / per_buffer_size(); in GreedyMemoryPlanner()
407 size_t scratch_buffer_size = Eigen::kGpuScratchSize + sizeof(unsigned int); in InitScratchBuffers() local410 Allocator::kAllocatorAlignment, scratch_buffer_size); in InitScratchBuffers()416 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()