| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | device_memory_allocator.cc | 34 StreamExecutorMemoryAllocator::Allocate(int device_ordinal, uint64 size, in Allocate()
50 int device_ordinal, perftools::gputools::DeviceMemoryBase* mem) { in Deallocate()
65 StreamExecutorMemoryAllocator::GetStreamExecutor(int device_ordinal) { in GetStreamExecutor()
|
| D | service_executable_run_options.h | 50 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal() function
55 int device_ordinal) const { in BorrowStream()
|
| D | backend.h | 131 bool device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
137 string device_name(int device_ordinal) const { in device_name()
|
| D | shaped_buffer.cc | 39 const se::Platform* platform, int device_ordinal) in ShapedBuffer()
122 int device_ordinal) in ScopedShapedBuffer()
|
| D | allocation_tracker.cc | 159 perftools::gputools::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount()
171 perftools::gputools::DeviceMemoryBase device_memory, int device_ordinal) { in DecrementRefCount()
|
| D | allocation_tracker.h | 68 int device_ordinal; member
|
| D | transfer_manager.cc | 185 int device_ordinal) { in AllocateShapedBuffer()
215 int device_ordinal) { in AllocateScopedShapedBuffer()
|
| D | shaped_buffer.h | 56 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
|
| D | backend.cc | 103 StatusOr<Backend::StreamPtr> Backend::BorrowStream(int device_ordinal) { in BorrowStream()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 63 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
87 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal, in BufferAllocations()
|
| D | cudnn_convolution_algorithm_picker.cc | 37 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
174 const auto device_ordinal = stream_exec_->device_ordinal(); in PickBestAlgorithm() local
|
| D | buffer_allocations.cc | 42 const BufferAssignment& buffer_assignment, int device_ordinal, in Build()
|
| D | fft_thunk.cc | 33 int device_ordinal, DeviceMemoryAllocator* memory_allocator) in FftScratchAllocator()
|
| D | gpu_executable.cc | 290 auto device_ordinal = executor->device_ordinal(); in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 34 int device_ordinal) { in set_device_ordinal()
40 int ExecutableBuildOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableBuildOptions
|
| D | local_client.cc | 34 StatusOr<Backend::StreamPtr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
269 LocalClient::LiteralToShapedBuffer(const Literal& literal, int device_ordinal, in LiteralToShapedBuffer()
294 int device_ordinal) { in TransferToInfeedLocal()
302 const Shape& shape, int device_ordinal) { in TransferFromOutfeedLocal()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 92 const xla::Backend* backend, int device_ordinal) { in GetOrCreateXlaDeviceAllocator()
109 const string& platform_name, const string& device_name, int device_ordinal, in Create()
144 XlaDevice::Metadata::Metadata(int device_ordinal, se::Platform* platform, in Metadata()
150 int XlaDevice::Metadata::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in tensorflow::XlaDevice::Metadata
179 const DeviceAttributes& attrs, int device_ordinal, in XlaDevice()
|
| D | xla_device_context.cc | 30 int device_ordinal) in XlaDeviceAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 21 int device_ordinal) { in set_device_ordinal()
26 int ExecutableRunOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableRunOptions
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 109 int CudaPlatform::DeviceToBus(int device_ordinal) { in DeviceToBus()
156 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 39 int device_ordinal, uint64 size, bool retry_on_failure) { in Allocate()
50 int device_ordinal, perftools::gputools::DeviceMemoryBase* mem) { in Deallocate()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 56 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| D | host_gpu_executor.h | 50 port::Status Init(int device_ordinal, DeviceOptions device_options) override { in Init()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | platform.cc | 59 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_launch_op.cc | 89 int device_ordinal, uint64 size, bool retry_on_failure) { in Allocate()
117 Status XlaAllocator::Deallocate(int device_ordinal, in Deallocate()
|