Method BigDeallocate

tensorflow/core/common_runtime/tensorpool_allocator.cc:392–408

}

void TensorPoolAllocator::BigDeallocate(Header* header) {
  auto ptr = header->raw_ptr;
  auto num_bytes = header->total_size;
  if (!inited_.load()) {
    mem_planner_->TrackDeallocate(header);
    sub_allocator_->Free(ptr, num_bytes);
    return;
  }
  
  if (header->bin == nullptr) {
    sub_allocator_->Free(ptr, num_bytes);
    return;
  }

  auto bin = (Bin*) (header->bin);
  bin->Deallocate(header);
}

class TensorPoolAllocatorFactory : public AllocatorFactory {
 public: