Method TfAllocatorAdapter

tensorflow/stream_executor/tf_allocator_adapter.cc:26–30

namespace stream_executor {

TfAllocatorAdapter::TfAllocatorAdapter(tensorflow::Allocator *wrapped,
                                       Stream *stream)
    : DeviceMemoryAllocator(stream->parent()->platform()),
      wrapped_(wrapped),
      stream_(stream) {}

TfAllocatorAdapter::TfAllocatorAdapter(tensorflow::Allocator *wrapped,
                                       Platform *platform)