This is a little tricky to expose because the `heap_size` attribute on a tracked function currently applies to the output, not the input.