| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | owning_device_memory.h | 42 class OwningDeviceMemory {
44 OwningDeviceMemory() : device_ordinal_(-1), allocator_(nullptr) {} in OwningDeviceMemory() function
46 explicit OwningDeviceMemory(se::DeviceMemoryBase mem, int device_ordinal, in OwningDeviceMemory() function
52 OwningDeviceMemory(OwningDeviceMemory&& other) in OwningDeviceMemory() function
60 OwningDeviceMemory& operator=(OwningDeviceMemory&& other) {
74 OwningDeviceMemory& operator=(std::nullptr_t) {
81 ~OwningDeviceMemory() { in ~OwningDeviceMemory()
|
| D | maybe_owning_device_memory.cc | 22 return absl::get<OwningDeviceMemory>(mem_).AsDeviceMemoryBase(); in AsDeviceMemoryBase()
29 return absl::holds_alternative<OwningDeviceMemory>(mem_); in HasOwnership()
32 absl::optional<OwningDeviceMemory> MaybeOwningDeviceMemory::Release() { in Release()
36 OwningDeviceMemory result = std::move(absl::get<OwningDeviceMemory>(mem_)); in Release()
38 return absl::make_optional<OwningDeviceMemory>(std::move(result)); in Release()
|
| D | maybe_owning_device_memory.h | 32 explicit MaybeOwningDeviceMemory(OwningDeviceMemory owned) in MaybeOwningDeviceMemory()
44 MaybeOwningDeviceMemory& operator=(OwningDeviceMemory owned) {
59 absl::optional<OwningDeviceMemory> Release();
65 absl::variant<OwningDeviceMemory, se::DeviceMemoryBase> mem_;
|
| D | device_memory_allocator.h | 50 virtual StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
58 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size) { in Allocate()
73 friend class OwningDeviceMemory;
85 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
|
| D | shaped_buffer_test.cc | 61 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
65 return OwningDeviceMemory(); in Allocate()
69 return OwningDeviceMemory(se::DeviceMemoryBase(buf, size), device_ordinal, in Allocate()
123 OwningDeviceMemory m, in TEST()
|
| D | device_memory_allocator.cc | 33 StatusOr<OwningDeviceMemory> StreamExecutorMemoryAllocator::Allocate( in Allocate()
47 return OwningDeviceMemory(result, device_ordinal, this); in Allocate()
|
| D | executable.h | 50 std::vector<OwningDeviceMemory> to_be_released) in ExecutionOutput()
56 std::vector<OwningDeviceMemory> to_be_released;
|
| D | owning_device_memory.cc | 22 void OwningDeviceMemory::Free() { in Free()
|
| D | allocation_tracker.h | 80 OwningDeviceMemory device_memory;
|
| D | shaped_buffer.h | 163 void set_buffer(OwningDeviceMemory buffer, const ShapeIndex& index) { in set_buffer()
|
| D | allocation_tracker.cc | 224 OwningDeviceMemory(device_memory, device_ordinal, in AddAllocationOrIncrementRefCount()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 76 std::vector<OwningDeviceMemory>>>
82 std::vector<OwningDeviceMemory> owning_buffers( in CreateBufferTable()
210 absl::Span<OwningDeviceMemory> buffers) { in CreateResultShapedBuffer()
238 OwningDeviceMemory& buffer = buffers[buffer_index]; in CreateResultShapedBuffer()
301 std::vector<OwningDeviceMemory> owning_buffers; in ExecuteAsyncOnStreamImpl()
329 std::shared_ptr<std::vector<OwningDeviceMemory>> buffers; in ExecuteAsyncOnStreamImpl()
341 std::make_shared<std::vector<OwningDeviceMemory>>( in ExecuteAsyncOnStreamImpl()
|
| D | cpu_executable.h | 114 std::vector<OwningDeviceMemory>>>
129 absl::Span<OwningDeviceMemory> buffers);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 173 xla::StatusOr<xla::OwningDeviceMemory> XlaAllocator::Allocate( in Allocate()
185 return xla::OwningDeviceMemory(se::DeviceMemoryBase(data, size), in Allocate()
372 output.set_buffer(xla::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
433 output.set_buffer(xla::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
|
| D | xla_tensor.cc | 62 TF_ASSIGN_OR_RETURN(xla::OwningDeviceMemory buffer, in AllocateShapedBuffer()
|
| D | xla_launch_util.h | 115 xla::StatusOr<xla::OwningDeviceMemory> Allocate(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | scratch_allocator.cc | 32 TF_ASSIGN_OR_RETURN(OwningDeviceMemory allocated_buffer, in AllocateBytes()
|
| D | scratch_allocator.h | 54 std::vector<OwningDeviceMemory> allocated_buffers_;
|
| D | fft_thunk.h | 53 std::vector<OwningDeviceMemory> allocated_buffers_;
|
| D | buffer_allocations.cc | 80 OwningDeviceMemory buffer; in Build()
|
| D | fft_thunk.cc | 51 TF_ASSIGN_OR_RETURN(OwningDeviceMemory allocated_buffer, in AllocateBytes()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 116 xla::OwningDeviceMemory buffer, in AllocateScopedShapedBuffer()
414 TF_ASSIGN_OR_RETURN(xla::OwningDeviceMemory buffer, in MakeTuple()
548 *shaped_tree.mutable_element(buffer.first) = xla::OwningDeviceMemory( in ToDeviceMemoryTree()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.h | 49 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
|
| D | local_client_test_base.cc | 39 StatusOr<OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()
|