Searched refs:mem_arena_size (Results 1 – 4 of 4) sorted by relevance
127 tracker->mem_arena_size = 0; in xnn_init_value_allocation_tracker()172 size_t mem_arena_size = 0; in xnn_plan_value_allocation_tracker() local186 if (mem_arena_size < current->alloc_offset + current->tensor_size) { in xnn_plan_value_allocation_tracker()187 mem_arena_size = current->alloc_offset + current->tensor_size; in xnn_plan_value_allocation_tracker()191 tracker->mem_arena_size = mem_arena_size; in xnn_plan_value_allocation_tracker()195 tracker->mem_arena_size = 0; in xnn_plan_value_allocation_tracker()198 tracker->usage[i].alloc_offset = tracker->mem_arena_size; in xnn_plan_value_allocation_tracker()199 tracker->mem_arena_size += tracker->usage[i].tensor_size; in xnn_plan_value_allocation_tracker()
158 size_t mem_arena_size = mem_alloc_tracker->mem_arena_size; in initialize_workspace_blobs() local159 if (mem_arena_size == 0) { in initialize_workspace_blobs()163 mem_arena_size += 2 * XNN_EXTRA_BYTES; in initialize_workspace_blobs()168 if (runtime->workspace->size < mem_arena_size) { in initialize_workspace_blobs()174 void* new_workspace_data = xnn_allocate_simd_memory(mem_arena_size); in initialize_workspace_blobs()176 xnn_log_error("failed to allocate %zu bytes for runtime workspace", mem_arena_size); in initialize_workspace_blobs()180 runtime->workspace->size = mem_arena_size; in initialize_workspace_blobs()187 assert(runtime->workspace->size >= mem_arena_size); in initialize_workspace_blobs()
83 EXPECT_EQ(192, tracker.mem_arena_size); in TEST()90 EXPECT_EQ(332, tracker.mem_arena_size); in TEST()144 EXPECT_EQ(124, tracker.mem_arena_size); in TEST()154 EXPECT_EQ(234, tracker.mem_arena_size); in TEST()
34 size_t mem_arena_size; member