Searched defs:StreamExecutorMemoryAllocator (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddevice_memory_allocator.cc26 StreamExecutorMemoryAllocator::StreamExecutorMemoryAllocator( function in class:xla::StreamExecutorMemoryAllocator
34 StreamExecutorMemoryAllocator::Allocate(int device_ordinal, uint64 size,
49 tensorflow::Status StreamExecutorMemoryAllocator::Deallocate(
65 StreamExecutorMemoryAllocator::GetStreamExecutor(int device_ordinal) {
82 bool StreamExecutorMemoryAllocator::AllowsAsynchronousDeallocation() const {
H A Ddevice_memory_allocator.h64 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator { class in namespace:xla
66 StreamExecutorMemoryAllocator(

Completed in 90 milliseconds