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()
|