HomeSort by relevance Sort by last modified time
    Searched refs:StreamExecutorMemoryAllocator (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
device_memory_allocator.cc 26 StreamExecutorMemoryAllocator::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 {
device_memory_allocator.h 64 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator {
66 StreamExecutorMemoryAllocator(
backend.h 185 std::unique_ptr<StreamExecutorMemoryAllocator> memory_allocator_;
backend.cc 140 MakeUnique<StreamExecutorMemoryAllocator>(platform, stream_executors);
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_test_base.h 42 class TestAllocator : public StreamExecutorMemoryAllocator {
45 : StreamExecutorMemoryAllocator(
local_client_test_base.cc 46 return StreamExecutorMemoryAllocator::Allocate(device_ordinal, size);
57 return StreamExecutorMemoryAllocator::Deallocate(device_ordinal, mem);
dynamic_ops_test.cc 668 StreamExecutorMemoryAllocator allocator(platform, executors);
fusion_test.cc     [all...]
local_client_execute_test.cc     [all...]
while_test.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
cudnn_convolution_algorithm_picker.cc 177 // StreamExecutorMemoryAllocator for stream_exec_.
179 optional<StreamExecutorMemoryAllocator> se_allocator;

Completed in 170 milliseconds