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()
875 if (mem_alloc_tracker.mem_arena_size != 0) { in xnn_create_runtime_v2()877 const size_t mem_arena_size = mem_alloc_tracker.mem_arena_size + XNN_EXTRA_BYTES; in xnn_create_runtime_v2() local878 runtime->workspace = xnn_allocate_simd_memory(mem_arena_size); in xnn_create_runtime_v2()880 xnn_log_error("failed to allocate %zu bytes for runtime workspace", mem_arena_size); in xnn_create_runtime_v2()
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