Searched refs:se_allocator (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | redzone_allocator_test.cc | 57 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST() local
62 &stream, &se_allocator, opts, in TEST()
132 StreamExecutorMemoryAllocator se_allocator(platform, {stream_exec}); in TEST() local
136 &stream, &se_allocator, opts, in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.cc | 79 SE_DeviceMemoryAllocator se_allocator; in ToC() local
81 se_allocator.ctx = nullptr; in ToC()
82 se_allocator.platform = nullptr; in ToC()
83 se_allocator.allocate = nullptr; in ToC()
84 se_allocator.deallocate = nullptr; in ToC()
85 return se_allocator; in ToC()
88 se_allocator.platform = nullptr; in ToC()
89 se_allocator.ctx = allocator; in ToC()
90 se_allocator.allocate = [](void* ctx, int device_ordinal, uint64_t size, in ToC()
109 se_allocator.deallocate = [](void* ctx, SE_DeviceMemoryBase* base, in ToC()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_algorithm_picker.cc | 319 optional<se::StreamExecutorMemoryAllocator> se_allocator; in PickBestAlgorithm() local
323 se_allocator.emplace(stream_exec_); in PickBestAlgorithm()
324 allocator = &*se_allocator; in PickBestAlgorithm()
|