| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
74 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
85 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
91 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
99 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
104 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
|
| D | local_client_test_base.cc | 39 StatusOr<OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()
52 Status TestAllocator::Deallocate(int device_ordinal, se::DeviceMemoryBase mem) { in Deallocate()
195 auto device_ordinal = in ExecuteLocally() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | device_memory_allocator.cc | 34 int device_ordinal, uint64 size, bool retry_on_failure) { in Allocate()
50 Status StreamExecutorMemoryAllocator::Deallocate(int device_ordinal, in Deallocate()
63 int device_ordinal) { in GetStreamExecutor()
|
| D | service_executable_run_options.h | 47 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal() function
51 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream()
|
| D | owning_device_memory.h | 46 explicit OwningDeviceMemory(se::DeviceMemoryBase mem, int device_ordinal, in OwningDeviceMemory()
90 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
|
| D | backend.h | 135 bool device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
141 string device_name(int device_ordinal) const { in device_name()
|
| D | device_memory_allocator.h | 58 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size) { in Allocate()
|
| D | shaped_buffer.cc | 36 const se::Platform* platform, int device_ordinal) in ShapedBuffer()
109 int device_ordinal) in ScopedShapedBuffer()
|
| D | shaped_buffer_test.cc | 61 StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
73 Status Deallocate(int device_ordinal, se::DeviceMemoryBase mem) override { in Deallocate()
|
| D | allocation_tracker.cc | 219 se::DeviceMemoryBase device_memory, int device_ordinal) { in AddAllocationOrIncrementRefCount()
233 int device_ordinal) { in DecrementRefCount()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 33 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager()
111 int device_ordinal = in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local
135 int device_ordinal = in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local
154 int device_ordinal = in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local
178 int device_ordinal = in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | executable_build_options.cc | 35 int device_ordinal) { in set_device_ordinal()
41 int ExecutableBuildOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableBuildOptions
|
| D | local_client.cc | 35 StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
263 const Literal& literal, int device_ordinal, in LiteralToShapedBuffer()
292 int device_ordinal) { in TransferToInfeedLocal()
300 int device_ordinal) { in TransferFromOutfeedLocal()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 86 xla::Backend* backend, int device_ordinal, const xla::Shape& shape, in AllocateScopedShapedBuffer()
133 int device_ordinal, in XRTBufferAllocation()
167 XRTTupleAllocation::XRTTupleAllocation(int device_ordinal, in XRTTupleAllocation()
184 const xla::LiteralBase& literal, xla::Backend* backend, int device_ordinal, in CreateAndTransfer()
211 int device_ordinal, XRTTupleAllocation** allocation) { in CreateFromBuffer()
222 Status XRTTupleAllocation::ToLiteral(xla::Backend* backend, int device_ordinal, in ToLiteral()
266 int XRTTupleAllocation::device_ordinal() { return device_ordinal_; } in device_ordinal() function in tensorflow::XRTTupleAllocation
340 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal, in ExpandTreeOfTuples()
381 xla::Backend* backend, int device_ordinal, in MakeTuple()
501 xla::DeviceMemoryAllocator* allocator, int device_ordinal) { in InitializeFromShapedBuffer()
|
| D | xrt_device.h | 48 int device_ordinal() { return 0; } in device_ordinal() function
|
| D | xrt_device.cc | 34 OpKernelContext* ctx, int device_ordinal, ScopedRef* scoped_ref) { in InitScopedRef()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 66 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
86 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal, in BufferAllocations()
|
| D | buffer_comparator.cc | 107 auto device_ordinal = stream_exec->device_ordinal(); in Create() local
137 auto device_ordinal = stream_exec->device_ordinal(); in CompareEqualImpl() local
|
| D | scratch_allocator.h | 32 ScratchAllocator(int device_ordinal, DeviceMemoryAllocator* memory_allocator) in ScratchAllocator()
|
| /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/python/tpu/ops/ |
| D | tpu_ops.py | 283 device_ordinal, argument
320 device_ordinal, argument
378 device_ordinal, argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 73 int ROCmPlatform::DeviceToBus(int device_ordinal) { in DeviceToBus()
121 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_platform.cc | 108 int CudaPlatform::DeviceToBus(int device_ordinal) { in DeviceToBus()
155 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 95 const xla::Backend* backend, int device_ordinal) { in GetOrCreateXlaDeviceAllocator()
131 int device_ordinal) { in BuildXlaDeviceAttributes()
141 int device_ordinal, se::Platform* platform, const DeviceType& device_type, in Metadata()
151 int XlaDevice::Metadata::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in tensorflow::XlaDevice::Metadata
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_platform.cc | 53 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|