| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multiple_devices_on_host_test.cc | 37 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
|
| 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 | device_memory_allocator.h | 60 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()
|
| 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()
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_node_context.cc | 30 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
|
| D | tpu_node_context.h | 50 explicit TpuNodeContext(int device_ordinal, XLA_TpuNodeContext* node_context) in TpuNodeContext()
|
| /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_tpu_device.cc | 37 Status XRTTpuDeviceAccessor::ScopedRef::Acquire(int device_ordinal) { in Acquire()
51 OpKernelContext* /*unused ctx*/, int device_ordinal, in InitScopedRef()
|
| D | xrt_memory_manager.h | 138 const int device_ordinal = 0; member
160 int device_ordinal, size_t requested_free_size) { in Run()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | host_compute_ops.cc | 39 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
|
| D | transfer_ops.cc | 73 OpKernelContext* ctx, int device_ordinal) { in RunTransferWithOrdinal()
129 const int device_ordinal = device_ordinal_tensor.scalar<int32>()(); in RunTransfer() local
|
| /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 | shaped_buffer.cc | 34 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()
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_allocations.h | 39 int device_ordinal, in BufferAllocations()
53 int device_ordinal() const { return device_ordinal_; } in device_ordinal() function
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 35 StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
270 const ShapeTree<MaybeOwningDeviceMemory>& tree, int device_ordinal) { in MaybeOwningShapeTreeToShapedBuffer()
400 const LiteralSlice& literal, int device_ordinal, in LiteralToShapedBuffer()
429 int device_ordinal) { in TransferToInfeedLocal()
436 Status LocalClient::TransferFromOutfeedLocal(int device_ordinal, in TransferFromOutfeedLocal()
449 const ::xla::BorrowingLiteral& literal, int device_ordinal) { in TransferToLocalServer()
|
| D | executable_build_options.cc | 36 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/python/tpu/ops/ |
| D | tpu_ops.py | 286 device_ordinal, argument
323 device_ordinal, argument
381 device_ordinal, argument
461 device_ordinal, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_runtime.cc | 52 XfeedManager* GetXfeedManager(int device_ordinal) { in GetXfeedManager()
235 int device_ordinal = in __xla_cpu_runtime_AcquireInfeedBufferForDequeue() local
259 int device_ordinal = in __xla_cpu_runtime_ReleaseInfeedBufferAfterDequeue() local
278 int device_ordinal = in __xla_cpu_runtime_AcquireOutfeedBufferForPopulation() local
302 int device_ordinal = in __xla_cpu_runtime_ReleaseOutfeedBufferAfterPopulation() local
601 int device_ordinal = GetDeviceOrdinal(run_options); in GetRendezvousKey() local
623 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AllToAll() local
665 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_AllReduce() local
711 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_ReplicaId() local
724 int device_ordinal = GetDeviceOrdinal(run_options); in __xla_cpu_runtime_CollectivePermute() local
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 141 xla::ScopedShapedBuffer output_scoped_buffer, int device_ordinal) { in AllocateOutputTuple()
177 xla::ScopedShapedBuffer output_scoped_buffer, int device_ordinal) { in AllocateOutputTensors()
193 int device_ordinal, string rendezvous_key_base) { in RunExecutable()
235 int device_ordinal, string rendezvous_key_base) { in ExecuteTPUProgram()
273 const int device_ordinal = metadata->device_ordinal(); in DoWork() local
395 const int device_ordinal = metadata->device_ordinal(); in DoWork() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 36 int device_ordinal) { in set_device_ordinal()
41 int ExecutableRunOptions::device_ordinal() const { return device_ordinal_; } in device_ordinal() function in xla::ExecutableRunOptions
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_extract_outside_compilation.cc | 94 Value device_ordinal, in CreateSendFromHostOp()
112 Value device_ordinal, in CreateRecvAtHostOp()
267 Value device_ordinal, int& communication_key_index) { in MoveOpsToHost()
358 Value compilation_key, Value device_ordinal, in DecomposeControlFlow()
444 Value device_ordinal = nullptr; in CreateParallelExecuteForOutsideCompilation() local
|