Function SetLightHeader

tensorflow/core/common_runtime/tensorpool_allocator.cc:25–32

}

void* SetLightHeader(void* p, size_t total_bytes, size_t header_size) {
  // LightHeader *KB max(sizeof(LightHeader)=8B, alignment)
  //   { | .....| checksum (4B) | header_size (4B)}
  auto user_ptr = (char*)p + header_size;
  auto header = new((char*)user_ptr - sizeof(LightHeader))
                    LightHeader(header_size);
  return user_ptr;
}

LightHeader* GetLightHeader(void* p) {
  auto light_header = (LightHeader*)((char*)p - sizeof(LightHeader));