Home
last modified time | relevance | path

Searched refs:StreamExecutorMemoryAllocator (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/
Ddevice_memory_allocator.h231 class StreamExecutorMemoryAllocator : public DeviceMemoryAllocator {
235 explicit StreamExecutorMemoryAllocator(StreamExecutor *executor);
240 StreamExecutorMemoryAllocator(
Dstream_executor_pimpl.cc885 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()
Dstream_executor_pimpl.h529 StreamExecutorMemoryAllocator *GetAllocator() { return &allocator_; } in GetAllocator()
748 StreamExecutorMemoryAllocator allocator_;
/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_test_base.h46 class TestAllocator : public se::StreamExecutorMemoryAllocator {
49 : se::StreamExecutorMemoryAllocator( in TestAllocator()
Dlocal_client_test_base.cc54 return se::StreamExecutorMemoryAllocator::Allocate( in Allocate()
65 return se::StreamExecutorMemoryAllocator::Deallocate(device_ordinal, mem); in Deallocate()
Dbuffer_donation_test.cc82 se::StreamExecutorMemoryAllocator memory_allocator( in RunAndCheck()
Ddynamic_ops_test.cc756 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_DynamicSlice()
Dlocal_client_execute_test.cc952 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_LocalClientOverhead()
Dcpu_gpu_fusion_test.cc832 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_ParallelFusion()
Dwhile_test.cc1267 se::StreamExecutorMemoryAllocator allocator(platform, executors); in BM_WhileLoop()
Ddot_operation_test.cc1816 se::StreamExecutorMemoryAllocator allocator(platform, executors); in DOT_ReorderContracting()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dredzone_allocator_test.cc57 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST()
132 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST()
/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.h182 std::unique_ptr<se::StreamExecutorMemoryAllocator> memory_allocator_;
Dbackend.cc132 memory_allocator_ = absl::make_unique<se::StreamExecutorMemoryAllocator>( in Backend()
Dshaped_buffer_test.cc35 xla::se::StreamExecutorMemoryAllocator allocator(platform, executors); in TEST()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc306 se::StreamExecutorMemoryAllocator allocator( in ReplayComputation()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc319 optional<se::StreamExecutorMemoryAllocator> se_allocator; in PickBestAlgorithm()