Searched refs:StreamExecutorMemoryAllocator (Results 1 – 17 of 17) sorted by relevance
/external/tensorflow/tensorflow/stream_executor/ |
D | device_memory_allocator.h | 231 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator { 235 explicit StreamExecutorMemoryAllocator(StreamExecutor *executor); 240 StreamExecutorMemoryAllocator(
|
D | stream_executor_pimpl.cc | 885 StreamExecutorMemoryAllocator::StreamExecutorMemoryAllocator( in StreamExecutorMemoryAllocator() function in stream_executor::StreamExecutorMemoryAllocator 891 StreamExecutorMemoryAllocator::StreamExecutorMemoryAllocator( in StreamExecutorMemoryAllocator() function in stream_executor::StreamExecutorMemoryAllocator 897 port::StatusOr<OwningDeviceMemory> StreamExecutorMemoryAllocator::Allocate( in Allocate() 916 port::Status StreamExecutorMemoryAllocator::Deallocate(int device_ordinal, in Deallocate() 929 StreamExecutorMemoryAllocator::GetStreamExecutor(int device_ordinal) const { in GetStreamExecutor() 944 bool StreamExecutorMemoryAllocator::AllowsAsynchronousDeallocation() const { in AllowsAsynchronousDeallocation() 948 port::StatusOr<Stream *> StreamExecutorMemoryAllocator::GetStream( in GetStream()
|
D | stream_executor_pimpl.h | 529 StreamExecutorMemoryAllocator *GetAllocator() { return &allocator_; } in GetAllocator() 748 StreamExecutorMemoryAllocator allocator_;
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | local_client_test_base.h | 46 class TestAllocator : public se::StreamExecutorMemoryAllocator { 49 : se::StreamExecutorMemoryAllocator( in TestAllocator()
|
D | local_client_test_base.cc | 54 return se::StreamExecutorMemoryAllocator::Allocate( in Allocate() 65 return se::StreamExecutorMemoryAllocator::Deallocate(device_ordinal, mem); in Deallocate()
|
D | buffer_donation_test.cc | 82 se::StreamExecutorMemoryAllocator memory_allocator( in RunAndCheck()
|
D | dynamic_ops_test.cc | 756 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_DynamicSlice()
|
D | local_client_execute_test.cc | 952 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_LocalClientOverhead()
|
D | cpu_gpu_fusion_test.cc | 832 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_ParallelFusion()
|
D | while_test.cc | 1267 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_WhileLoop()
|
D | dot_operation_test.cc | 1816 se::StreamExecutorMemoryAllocator allocator(platform, executors); in DOT_ReorderContracting()
|
/external/tensorflow/tensorflow/stream_executor/cuda/ |
D | redzone_allocator_test.cc | 57 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST() 132 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | backend.h | 182 std::unique_ptr<se::StreamExecutorMemoryAllocator> memory_allocator_;
|
D | backend.cc | 132 memory_allocator_ = absl::make_unique<se::StreamExecutorMemoryAllocator>( in Backend()
|
D | shaped_buffer_test.cc | 35 xla::se::StreamExecutorMemoryAllocator allocator(platform, executors); in TEST()
|
/external/tensorflow/tensorflow/compiler/xla/tools/ |
D | replay_computation.cc | 306 se::StreamExecutorMemoryAllocator allocator( in ReplayComputation()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | gpu_conv_algorithm_picker.cc | 319 optional<se::StreamExecutorMemoryAllocator> se_allocator; in PickBestAlgorithm()
|