Home
last modified time | relevance | path

Searched defs:device_ordinal (Results 1 – 25 of 122) sorted by relevance

12345

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc37 LocalExecutable* executable, int device_ordinal, LocalClient* client, in CompileAndExecute()
76 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
87 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
93 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
101 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
106 for (int device_ordinal = 0; device_ordinal < device_count; in TestWithDeviceCount() local
Dlocal_client_test_base.cc42 StatusOr<se::OwningDeviceMemory> TestAllocator::Allocate(int device_ordinal, in Allocate()
56 Status TestAllocator::Deallocate(int device_ordinal, se::DeviceMemoryBase mem) { in Deallocate()
199 auto device_ordinal = in ExecuteLocally() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dtf_allocator_adapter.h82 int device_ordinal = p.second->parent()->device_ordinal(); in MultiDeviceAdapter() local
93 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64_t size, in Allocate()
101 port::Status Deallocate(int device_ordinal, DeviceMemoryBase mem) override { in Deallocate()
116 port::StatusOr<Stream *> GetStream(int device_ordinal) override { in GetStream()
Ddevice_memory_allocator.h61 ScopedDeviceMemory(DeviceMemoryBase mem, int device_ordinal, in ScopedDeviceMemory()
140 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
188 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, in Allocate()
195 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64_t size, in Allocate()
Dtf_allocator_adapter.cc39 int device_ordinal, uint64_t 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()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc77 OpKernelContext* ctx, int device_ordinal) { in RunTransferWithOrdinal()
123 const int device_ordinal = device_ordinal_tensor.scalar<int32>()(); in RunTransfer() local
153 int device_ordinal, in TransferBuffersToInfeed()
160 int device_ordinal, const xla::LiteralSlice& literal) { in TransferLiteralToInfeed()
166 int device_ordinal, xla::MutableBorrowingLiteral literal) { in TransferLiteralFromOutfeed()
172 StreamExecutorTransferOpImpl::GetStreamExecutor(int device_ordinal) { in GetStreamExecutor()
Dhost_compute_ops.cc39 int device_ordinal = 0; in RecvAtHostOp() local
84 const int device_ordinal = device_ordinal_tensor.flat<int64_t>()(0); in ComputeAsync() local
190 int device_ordinal = 0; in SendFromHostOp() local
240 const int device_ordinal = device_ordinal_tensor.flat<int64_t>()(0); in Compute() local
Doutfeed_ops.h50 Status DoWork(OpKernelContext* ctx, int device_ordinal) override { in DoWork()
105 Status DoWork(OpKernelContext* ctx, int device_ordinal) override { in DoWork()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dtpu_node_context.cc28 int device_ordinal) { in Create()
60 Status TpuNodeContext::Initialize(int device_ordinal) { in Initialize()
71 int TpuNodeContext::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in tensorflow::tpu::TpuNodeContext
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_xfeed.cc103 Status TransferBufferToInfeed(int device_ordinal, int64_t size, in TransferBufferToInfeed()
116 int device_ordinal, absl::Span<const std::pair<void*, int64_t>> buffer_data, in TransferBuffersFromOutfeedInternal()
161 StatusOr<Shape> TransferArrayBufferFromOutfeed(int device_ordinal, in TransferArrayBufferFromOutfeed()
169 int device_ordinal, in TransferTupleBuffersFromOutfeed()
176 Status TransferLiteralToInfeedOnCpu(int device_ordinal, in TransferLiteralToInfeedOnCpu()
222 Status TransferLiteralFromOutfeedOnCpu(int device_ordinal, in TransferLiteralFromOutfeedOnCpu()
Dcpu_runtime.cc63 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager()
179 int64_t device_ordinal; member
204 int64_t device_ordinal; member
303 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local
326 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local
344 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local
367 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local
699 int device_ordinal = GetDeviceOrdinal(run_options); in GetRendezvousKey() local
723 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AllToAll() local
766 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AllReduce() local
[all …]
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc73 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in AllocateScopedShapedBuffer()
121 int device_ordinal, in XRTBufferAllocation()
148 XRTTupleAllocation::XRTTupleAllocation(int device_ordinal, in XRTTupleAllocation()
172 xla::Backend* backend, int device_ordinal, XRTTupleAllocation** allocation, in CreateAndTransfer()
199 xla::Backend* backend, int device_ordinal, XRTTupleAllocation** allocation, in CreateUninitialized()
223 int device_ordinal, XRTTupleAllocation** allocation, in CreateFromBuffer()
235 int device_ordinal, XRTTupleAllocation** allocation, in CreateFromBuffer()
357 int XRTTupleAllocation::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in tensorflow::XRTTupleAllocation
415 const xla::ShapeTree<ExpandedTupleInput>& elements, int device_ordinal, in ExpandTreeOfTuples()
456 XRTMemoryManager* memory_manager, xla::Backend* backend, int device_ordinal, in MakeTuple()
[all …]
Dxrt_memory_manager.cc37 int64_t MakeDeviceHandle(int64_t device_ordinal, int64_t rnd_value) { in MakeDeviceHandle()
229 int device_ordinal = GetDeviceFromHandle(handle); in Lookup() local
243 int device_ordinal = GetDeviceFromHandle(handle); in Release() local
253 xla::Backend* backend, int device_ordinal, in CompactAllocations()
272 xla::Backend* backend, int device_ordinal, size_t size, in Allocate()
309 int device_ordinal, bool create_if_missing) { in GetDeviceContext()
Dxrt_tpu_device.cc37 Status XRTTpuDeviceAccessor::ScopedRef::Acquire(int device_ordinal) { in Acquire()
51 OpKernelContext* /*unused ctx*/, int device_ordinal, in InitScopedRef()
Dxrt_memory_manager.h145 const int device_ordinal = 0; member
167 int device_ordinal, size_t requested_free_size, in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshaped_buffer.cc34 ShapedBuffer::ShapedBuffer(Shape on_device_shape, int device_ordinal) in ShapedBuffer()
42 int device_ordinal) in ShapedBuffer()
118 int device_ordinal) in ScopedShapedBuffer()
125 int device_ordinal) in ScopedShapedBuffer()
Dservice_executable_run_options.h49 int device_ordinal() const { return run_options_.device_ordinal(); } in device_ordinal() function
53 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream()
Dbackend.h136 bool device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
142 std::string device_name(int device_ordinal) const { in device_name()
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py299 device_ordinal, argument
336 device_ordinal, argument
394 device_ordinal, argument
474 device_ordinal, argument
552 device_ordinal, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_allocations.h39 int device_ordinal, in BufferAllocations()
53 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
Dnccl_all_reduce_thunk.cc46 int device_ordinal = stream.parent()->device_ordinal(); in RunAllReduce() local
257 int device_ordinal = stream.parent()->device_ordinal(); in RunNcclCollective() local
302 int device_ordinal = async_comms_stream.parent()->device_ordinal(); in RunNcclCollective() local
314 StatusOr<se::Event> NcclAllReduceStartThunk::TakeDoneEvent(int device_ordinal) { in TakeDoneEvent()
329 int device_ordinal = params.stream->parent()->device_ordinal(); in ExecuteOnStream() local
373 int device_ordinal = stream.parent()->device_ordinal(); in RunReduceScatter() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc36 StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
273 const ShapeTree<MaybeOwningDeviceMemory>& tree, int device_ordinal) { in MaybeOwningShapeTreeToShapedBuffer()
447 const LiteralSlice& literal, int device_ordinal, in LiteralToShapedBuffer()
476 int device_ordinal) { in TransferToInfeedLocal()
483 Status LocalClient::TransferFromOutfeedLocal(int device_ordinal, in TransferFromOutfeedLocal()
496 const ::xla::BorrowingLiteral& literal, int device_ordinal) { in TransferToLocalServer()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc141 xla::ScopedShapedBuffer output_scoped_buffer, int device_ordinal) { in AllocateOutputTuple()
178 xla::ScopedShapedBuffer output_scoped_buffer, int device_ordinal) { in AllocateOutputTensors()
194 int device_ordinal, string rendezvous_key_base) { in RunExecutable()
236 int device_ordinal, string rendezvous_key_base) { in ExecuteTPUProgram()
275 const int device_ordinal = metadata->device_ordinal(); in DoWork() local
397 const int device_ordinal = metadata->device_ordinal(); in DoWork() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_driver.h79 static GpuContext* Add(CUcontext context, int device_ordinal) { in Add()
115 int device_ordinal; in GetAnyContext() local
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc36 int device_ordinal) { in set_device_ordinal()
41 int ExecutableRunOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableRunOptions

12345