| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | maybe_owning_device_memory.cc | 25 return *absl::get<tensorflow::se::OwningDeviceMemory>(mem_); in AsDeviceMemoryBase()
32 return absl::holds_alternative<tensorflow::se::OwningDeviceMemory>(mem_); in HasOwnership()
35 absl::optional<tensorflow::se::OwningDeviceMemory>
40 return std::move(absl::get<tensorflow::se::OwningDeviceMemory>(mem_)); in Release()
43 const tensorflow::se::OwningDeviceMemory*
45 return HasOwnership() ? &absl::get<tensorflow::se::OwningDeviceMemory>(mem_) in AsOwningDeviceMemory()
|
| D | maybe_owning_device_memory.h | 31 explicit MaybeOwningDeviceMemory(tensorflow::se::OwningDeviceMemory owned) in MaybeOwningDeviceMemory()
43 MaybeOwningDeviceMemory& operator=(tensorflow::se::OwningDeviceMemory owned) {
58 absl::optional<tensorflow::se::OwningDeviceMemory> Release();
62 const tensorflow::se::OwningDeviceMemory* AsOwningDeviceMemory() const;
68 absl::variant<tensorflow::se::OwningDeviceMemory,
|
| D | executable.h | 153 std::vector<se::OwningDeviceMemory> to_be_released) in ExecutionOutput()
171 result_.set_buffer(se::OwningDeviceMemory(), index); in ~ExecutionOutput()
179 void AddToBeReleased(se::OwningDeviceMemory mem) { in AddToBeReleased()
199 const std::vector<se::OwningDeviceMemory>& ToBeReleased() const { in ToBeReleased()
203 std::vector<se::OwningDeviceMemory> ConsumeToBeReleased() { in ConsumeToBeReleased()
218 std::vector<se::OwningDeviceMemory> to_be_released_;
228 se::OwningDeviceMemory output_shape_table_;
|
| D | shaped_buffer_test.cc | 59 StatusOr<se::OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
64 return se::OwningDeviceMemory(); in Allocate()
68 return se::OwningDeviceMemory(se::DeviceMemoryBase(buf, size), in Allocate()
126 se::OwningDeviceMemory m, in TEST()
163 se::OwningDeviceMemory m, in TEST()
|
| D | allocation_tracker.h | 80 se::OwningDeviceMemory device_memory;
|
| D | executable.cc | 67 const tensorflow::se::OwningDeviceMemory* mem = in ToShapedBuffer()
307 if (absl::optional<se::OwningDeviceMemory> maybe_owning_buffer = in MarkToBeReleasedArguments()
|
| D | shaped_buffer.h | 182 void set_buffer(se::OwningDeviceMemory buffer, const ShapeIndex& index) { in set_buffer()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_memory_allocator.h | 153 using OwningDeviceMemory = ScopedDeviceMemory<uint8>; variable
176 virtual port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal,
187 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size) { in Allocate()
193 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
244 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
|
| D | tf_allocator_adapter.cc | 38 port::StatusOr<OwningDeviceMemory> TfAllocatorAdapter::Allocate( in Allocate()
53 return OwningDeviceMemory(DeviceMemoryBase(data, size), device_ordinal, this); in Allocate()
|
| D | tf_allocator_adapter.h | 42 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
81 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 114 TF_ASSIGN_OR_RETURN(se::OwningDeviceMemory out, in MemoryForAllocation()
272 if (absl::optional<se::OwningDeviceMemory> owning = in CreateResultShapedBuffer()
296 se::OwningDeviceMemory allocated_buffer, in CreateResultShapedBuffer()
312 if (absl::optional<se::OwningDeviceMemory> owned_buffer = in CreateResultShapedBuffer()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.h | 124 std::vector<std::pair<OwningDeviceMemory, int64>> allocated_buffers_;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | fft_thunk.h | 54 std::vector<se::OwningDeviceMemory> allocated_buffers_;
|
| D | gpu_executable.cc | 345 se::OwningDeviceMemory buffer, in BufferForAllocation()
494 absl::optional<tensorflow::se::OwningDeviceMemory> owning = in ExecuteAsyncOnStreamImpl()
521 se::OwningDeviceMemory allocated_buffer, in ExecuteAsyncOnStreamImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.h | 53 StatusOr<se::OwningDeviceMemory> Allocate(int device_ordinal, uint64 size,
|
| D | local_client_test_base.cc | 44 StatusOr<se::OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_memory_manager.h | 107 xla::StatusOr<se::OwningDeviceMemory> Allocate(xla::Backend* backend,
|
| D | xrt_state.cc | 102 se::OwningDeviceMemory buffer, in AllocateScopedShapedBuffer()
489 se::OwningDeviceMemory buffer, in MakeTuple()
671 xla::MaybeOwningDeviceMemory(se::OwningDeviceMemory( in ToExecutionInput()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 58 TF_ASSIGN_OR_RETURN(se::OwningDeviceMemory buffer, in AllocateShapedBuffer()
|
| D | xla_launch_util.cc | 212 *in_buffer = se::OwningDeviceMemory(buffer, device_ordinal, allocator); in PopulateExecutionInputBuffer()
524 output.set_buffer(se::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
570 output.set_buffer(se::OwningDeviceMemory(), {output_num}); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.h | 98 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem);
|
| D | c_api_conversions.cc | 49 const stream_executor::OwningDeviceMemory* owned = in ToC()
70 stream_executor::OwningDeviceMemory(ApiConverter::FromC(se_mem->memory), in FromC()
122 SE_MaybeOwningDeviceMemory ToC(stream_executor::OwningDeviceMemory* mem) { in ToC()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | tracked_device_buffer_test.cc | 37 se::OwningDeviceMemory device_memory, in MakeArray()
|
| D | tracked_device_buffer.cc | 157 se::OwningDeviceMemory(buf, device_ordinal_, allocator)); in AddToInputAsDonated()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 113 std::vector<se::OwningDeviceMemory> owning_memory; in RunMlirModuleWithHostBuffers()
|