- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for shaped_buffer (0.49 sec)
-
tensorflow/compiler/jit/xla_tpu_device.cc
<< " dst: " << dst_compute_stream->parent()->device_ordinal() << ", " << " input buffers: " << xla_input->shaped_buffer().ToString() << " output buffers: " << xla_output->shaped_buffer().ToString(); // Wait for definition event of the source tensor so the input buffers are // available.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device_context.cc
VLOG(2) << "Transfer to device as literal: " << literal.ToString() << " " << xla_tensor->shaped_buffer().ToString(); if (UseMultipleStreams() && !transfer_manager_->CanShapedBufferBeAccessedNow( stream_->parent(), xla_tensor->shaped_buffer())) { // Initially wait for the compute stream so that memory allocations are // synchronized.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
tensorflow::XlaTensor::FromTensor(&tensor); if (xla_tensor == nullptr) { return TensorShapeToXLAShape(tensor.dtype(), tensor.shape(), shape); } const xla::ShapedBuffer& shaped_buffer = xla_tensor->shaped_buffer(); *shape = shaped_buffer.on_device_shape(); return absl::OkStatus(); } // Caches a XlaDeviceAllocator per <backend, device ordinal> pair. A // XlaDeviceAllocator is created on demand and is associated with a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_launch_util.cc
// buffer with a nullptr root index table. This allows the code below to treat // output as a tuple unconditionally. if (!output.on_host_shape().IsTuple()) { ShapedBuffer nontuple_buffer = output.release(); ShapedBuffer buffer( xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_host_shape()}), xla::ShapeUtil::MakeTupleShape({nontuple_buffer.on_device_shape()}), output.device_ordinal());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 16 00:36:08 UTC 2024 - 40.4K bytes - Viewed (0)