Home
last modified time | relevance | path

Searched defs:device_ordinal (Results 1 – 25 of 112) 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
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
Dlocal_client_test_base.cc44 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/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>()(0); in ComputeAsync() local
190 int device_ordinal = 0; in SendFromHostOp() local
240 const int device_ordinal = device_ordinal_tensor.flat<int64>()(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/stream_executor/
Ddevice_memory_allocator.h60 ScopedDeviceMemory(DeviceMemoryBase mem, int device_ordinal, in ScopedDeviceMemory()
139 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
187 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size) { in Allocate()
193 port::StatusOr<OwningDeviceMemory> Allocate(int device_ordinal, uint64 size, in Allocate()
Dtf_allocator_adapter.h81 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()
Dtf_allocator_adapter.cc39 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()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_node_context.cc30 int device_ordinal) { in Create()
62 Status TpuNodeContext::Initialize(int device_ordinal) { in Initialize()
73 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>> buffer_data, in TransferBuffersFromOutfeedInternal()
160 StatusOr<Shape> TransferArrayBufferFromOutfeed(int device_ordinal, in TransferArrayBufferFromOutfeed()
168 int device_ordinal, absl::Span<const std::pair<void*, int64>> buffer_data) { in TransferTupleBuffersFromOutfeed()
174 Status TransferLiteralToInfeedOnCpu(int device_ordinal, in TransferLiteralToInfeedOnCpu()
220 Status TransferLiteralFromOutfeedOnCpu(int device_ordinal, in TransferLiteralFromOutfeedOnCpu()
Dcpu_runtime.cc54 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager()
149 xla::int64 device_ordinal; member
174 xla::int64 device_ordinal; member
269 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local
292 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local
310 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local
333 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local
623 int device_ordinal = GetDeviceOrdinal(run_options); in GetRendezvousKey() local
645 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AllToAll() local
687 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 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.h138 bool device_ordinal_supported(int device_ordinal) const { in device_ordinal_supported()
144 string device_name(int device_ordinal) const { in device_name()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc35 StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
272 const ShapeTree<MaybeOwningDeviceMemory>& tree, int device_ordinal) { in MaybeOwningShapeTreeToShapedBuffer()
402 const LiteralSlice& literal, int device_ordinal, in LiteralToShapedBuffer()
431 int device_ordinal) { in TransferToInfeedLocal()
438 Status LocalClient::TransferFromOutfeedLocal(int device_ordinal, in TransferFromOutfeedLocal()
451 const ::xla::BorrowingLiteral& literal, int device_ordinal) { in TransferToLocalServer()
Dexecutable_build_options.cc36 int device_ordinal) { in set_device_ordinal()
42 int ExecutableBuildOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableBuildOptions
/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.cc44 int device_ordinal = stream.parent()->device_ordinal(); in RunAllReduce() local
253 int device_ordinal = stream.parent()->device_ordinal(); in RunNcclCollective() local
294 int device_ordinal = async_comms_stream.parent()->device_ordinal(); in RunNcclCollective() local
306 StatusOr<se::Event> NcclAllReduceStartThunk::TakeDoneEvent(int device_ordinal) { in TakeDoneEvent()
321 int device_ordinal = params.stream->parent()->device_ordinal(); in ExecuteOnStream() local
354 int device_ordinal = params.stream->parent()->device_ordinal(); in RunNcclCollective() local
Dnccl_collective_thunk.cc141 int device_ordinal = params.stream->parent()->device_ordinal(); in ExecuteOnStream() local
165 int device_ordinal = params.stream->parent()->device_ordinal(); in GetDeviceString() local
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py303 device_ordinal, argument
340 device_ordinal, argument
398 device_ordinal, argument
478 device_ordinal, argument
/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

12345