| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
75 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
86 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
92 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
100 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
105 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
|
| D | local_client_test_base.cc | 44 StatusOr<se::OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()
58 Status TestAllocator::Deallocate(int device_ordinal, se::DeviceMemoryBase mem) { in Deallocate()
202 auto device_ordinal = in ExecuteLocally() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | tf_allocator_adapter.h | 81 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
89 port::Status Deallocate(int device_ordinal, DeviceMemoryBase mem) override { in Deallocate()
104 port::StatusOr<Stream *> GetStream(int device_ordinal) override { in GetStream()
|
| D | tf_allocator_adapter.cc | 39 int device_ordinal, uint64 size, bool retry_on_failure, in Allocate()
56 port::Status TfAllocatorAdapter::Deallocate(int device_ordinal, in Deallocate()
62 port::StatusOr<Stream *> TfAllocatorAdapter::GetStream(int device_ordinal) { in GetStream()
|
| D | device_memory_allocator.h | 59 ScopedDeviceMemory(DeviceMemoryBase mem, int device_ordinal, in ScopedDeviceMemory()
135 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
183 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size) { in Allocate()
189 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | asm_compiler.cc | 36 port::StatusOr<std::vector<uint8>> CompileGpuAsm(int device_ordinal, in CompileGpuAsm()
46 int device_ordinal, const char* ptx, GpuAsmOpts compilation_options) { in CompileGpuAsmOrGetCached()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service_executable_run_options.h | 49 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal() function
53 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream()
|
| 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 | shaped_buffer_test.cc | 59 StatusOr<se::OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
72 Status Deallocate(int device_ordinal, se::DeviceMemoryBase mem) override { in Deallocate()
89 StatusOr<se::Stream*> GetStream(int device_ordinal) override { in GetStream()
|
| D | shaped_buffer.cc | 35 const se::Platform* platform, int device_ordinal) in ShapedBuffer()
122 int device_ordinal) in ScopedShapedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer()
120 int device_ordinal, in XRTBufferAllocation()
147 XRTTupleAllocation::XRTTupleAllocation(int device_ordinal, in XRTTupleAllocation()
171 xla::Backend* backend, int device_ordinal, in CreateAndTransfer()
199 xla::Backend* backend, int device_ordinal, in CreateUninitialized()
223 int device_ordinal, XRTTupleAllocation** allocation) { in CreateFromBuffer()
236 int device_ordinal, XRTTupleAllocation** allocation) { in CreateFromBuffer()
356 int XRTTupleAllocation::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in tensorflow::XRTTupleAllocation
414 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal, in ExpandTreeOfTuples()
455 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in MakeTuple()
[all …]
|
| D | xrt_memory_manager.cc | 37 int64 MakeDeviceHandle(int64 device_ordinal, int64 rnd_value) { in MakeDeviceHandle()
226 int device_ordinal = GetDeviceFromHandle(handle); in Lookup() local
240 int device_ordinal = GetDeviceFromHandle(handle); in Release() local
250 int device_ordinal) { in CompactAllocations()
268 xla::Backend* backend, int device_ordinal, size_t size) { in Allocate()
305 int device_ordinal, bool create_if_missing) { in GetDeviceContext()
|
| D | xrt_memory_manager.h | 138 const int device_ordinal = 0; member
160 int device_ordinal, size_t requested_free_size) { in Run()
|
| D | xrt_device.h | 48 int device_ordinal() const { return ordinal_; } 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 | 68 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
88 BufferAllocations(BufferAllocation::Index buffer_count, int device_ordinal, in BufferAllocations()
|
| /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()
262 se::Platform* platform, int device_ordinal) { in MaybeOwningShapeTreeToShapedBuffer()
367 const LiteralSlice& literal, int device_ordinal, in LiteralToShapedBuffer()
396 int device_ordinal) { in TransferToInfeedLocal()
404 int device_ordinal) { in TransferFromOutfeedLocal()
418 const ::xla::BorrowingLiteral& literal, int device_ordinal) { in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 52 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager()
173 int device_ordinal = in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local
197 int device_ordinal = in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local
216 int device_ordinal = in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local
240 int device_ordinal = in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local
397 int device_ordinal = [&] { in __xla_cpu_runtime_AllReduce() local
447 int device_ordinal = [&]() { in __xla_cpu_runtime_ReplicaId() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | shared_device_buffer.cc | 60 int device_ordinal, se::DeviceMemoryAllocator* allocator, in BufferFromScopedShapedBufferIterator()
113 se::DeviceMemoryAllocator* allocator, int device_ordinal, in MakeTuple()
132 se::DeviceMemoryAllocator* allocator, int device_ordinal, in MakeArray()
180 se::DeviceMemoryAllocator* allocator, int device_ordinal, in SharedDeviceBuffer()
|
| D | shared_device_buffer.h | 122 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 34 int device_ordinal) { in set_device_ordinal()
39 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 | 286 device_ordinal, argument
323 device_ordinal, argument
381 device_ordinal, argument
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_platform.cc | 68 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 | 106 int CudaPlatform::DeviceToBus(int device_ordinal) { in DeviceToBus()
158 int device_ordinal, const PluginConfig& plugin_config) { in ExecutorForDeviceWithPluginConfig()
|