| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | owning_device_memory.h | 44 OwningDeviceMemory() : device_ordinal_(-1), allocator_(nullptr) {} in OwningDeviceMemory()
48 : mem_(mem), device_ordinal_(device_ordinal), allocator_(allocator) { in OwningDeviceMemory()
54 device_ordinal_(other.device_ordinal_), in OwningDeviceMemory()
65 device_ordinal_ = other.device_ordinal_;
90 int device_ordinal() const { return device_ordinal_; } in device_ordinal()
125 int device_ordinal_; variable
|
| D | shaped_buffer.cc | 40 device_ordinal_(device_ordinal), in ShapedBuffer()
47 device_ordinal_(s.device_ordinal_), in ShapedBuffer()
59 device_ordinal_ = s.device_ordinal_; in operator =()
|
| D | shaped_buffer.h | 69 int device_ordinal() const { return device_ordinal_; } in device_ordinal()
116 int device_ordinal_; variable
|
| D | owning_device_memory.cc | 26 auto status = allocator_->Deallocate(device_ordinal_, mem_); in Free()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 37 device_ordinal_ = device_ordinal; in set_device_ordinal()
41 int ExecutableBuildOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal()
75 device_ordinal_, result_layout, num_replicas_); in ToString()
|
| D | executable_build_options.h | 76 int device_ordinal_ = -1;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 66 int device_ordinal() const { return device_ordinal_; } in device_ordinal()
90 device_ordinal_(device_ordinal), in BufferAllocations()
106 int device_ordinal_; variable
|
| D | scratch_allocator.h | 33 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {} in ScratchAllocator()
52 const int device_ordinal_;
|
| D | scratch_allocator.cc | 33 memory_allocator_->Allocate(device_ordinal_, byte_size, in AllocateBytes()
|
| D | fft_thunk.cc | 33 : device_ordinal_(device_ordinal), memory_allocator_(memory_allocator) {} in FftScratchAllocator()
52 memory_allocator_->Allocate(device_ordinal_, byte_size, in AllocateBytes()
|
| D | fft_thunk.h | 51 const int device_ordinal_;
|
| D | buffer_allocations.cc | 143 memory_allocator_->Deallocate(device_ordinal_, buffer_address); in TearDown()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 137 device_ordinal_(device_ordinal), in XRTBufferAllocation()
141 GetAllocStats()->ReportAlloc(device_ordinal_, allocation_.size()); in XRTBufferAllocation()
142 LOG(INFO) << "XRT Allocation Stats: device=" << device_ordinal_ in XRTBufferAllocation()
149 GetAllocStats()->ReportFree(device_ordinal_, allocation_.size()); in ~XRTBufferAllocation()
152 Status s = allocator_->Deallocate(device_ordinal_, allocation_); in ~XRTBufferAllocation()
171 : device_ordinal_(device_ordinal), in XRTTupleAllocation()
266 int XRTTupleAllocation::device_ordinal() { return device_ordinal_; } in device_ordinal()
511 allocator_->platform(), device_ordinal_); in ToShapedBuffer()
549 buffer.second->allocation(), device_ordinal_, allocator_); in ToDeviceMemoryTree()
|
| D | xrt_state.h | 71 int device_ordinal_; variable
217 int device_ordinal_; variable
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 22 device_ordinal_ = device_ordinal; in set_device_ordinal()
26 int ExecutableRunOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal()
|
| D | executable_run_options.h | 85 int device_ordinal_ = -1; variable
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 144 : device_ordinal_(device_ordinal), in Metadata()
151 int XlaDevice::Metadata::device_ordinal() const { return device_ordinal_; } in device_ordinal()
201 device_ordinal_(options.device_ordinal), in XlaDevice()
253 backend, device_ordinal_); in GetAllocatorLocked()
269 TF_ASSIGN_OR_RETURN(ptr, backend->BorrowStream(device_ordinal_)); in EnsureStreamOkLocked()
|
| D | xla_device.h | 80 const int device_ordinal_;
200 const int device_ordinal_; variable
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 145 device_ordinal_ = device_ordinal; in Init()
152 status = GpuDriver::GetDevice(device_ordinal_, &device_); in Init()
157 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options, in Init()
896 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal_); in PopulateDeviceDescription()
901 if (GpuDriver::GetDeviceProperties(&prop, device_ordinal_)) { in PopulateDeviceDescription()
|
| D | rocm_driver.cc | 56 GpuContext(const int v) : device_ordinal_(v) {} in GpuContext()
58 int device_ordinal() const { return device_ordinal_; } in device_ordinal()
67 const int device_ordinal_; member in stream_executor::gpu::GpuContext
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 132 device_ordinal_ = device_ordinal; in Init()
139 status = GpuDriver::GetDevice(device_ordinal_, &device_); in Init()
144 status = GpuDriver::CreateContext(device_ordinal_, device_, device_options, in Init()
1049 int numa_node = TryToReadNumaNode(pci_bus_id, device_ordinal_); in PopulateDeviceDescription()
1090 CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE, device_ordinal_); in PopulateDeviceDescription()
1092 CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH, device_ordinal_); in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 51 device_ordinal_(0), in GpuExecutor()
327 int device_ordinal_; variable
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 163 device_ordinal_(-1), in StreamExecutor()
185 device_ordinal_(-1), in StreamExecutor()
226 device_ordinal_ = device_ordinal; in Init()
505 << device_ordinal_ in Allocate()
|
| D | stream_executor_pimpl.h | 434 int device_ordinal() const { return device_ordinal_; } in device_ordinal()
686 int device_ordinal_; variable
|