Searched refs:mem_alloc_tracker (Results 1 – 1 of 1) sorted by relevance
853 struct xnn_value_allocation_tracker mem_alloc_tracker; in xnn_create_runtime_v2() local854 xnn_init_value_allocation_tracker(&mem_alloc_tracker, subgraph); in xnn_create_runtime_v2()865 …xnn_add_value_allocation_tracker(&mem_alloc_tracker, i, round_up_po2(blob->size, XNN_EXTRA_BYTES)); in xnn_create_runtime_v2()873 xnn_plan_value_allocation_tracker(&mem_alloc_tracker); in xnn_create_runtime_v2()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()881 xnn_release_value_allocation_tracker(&mem_alloc_tracker); in xnn_create_runtime_v2()890 … blob->data = (void*) ((uintptr_t) runtime->workspace + mem_alloc_tracker.usage[i].alloc_offset); in xnn_create_runtime_v2()895 xnn_release_value_allocation_tracker(&mem_alloc_tracker); in xnn_create_runtime_v2()