OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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