- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for LocalDevice (0.1 sec)
-
tensorflow/compiler/jit/pjrt_base_device.cc
absl::StrCat("device: ", device_name, " device")); } } // namespace PjRtBaseDevice::PjRtBaseDevice(const SessionOptions& session_options, const Options& options) : LocalDevice(session_options, BuildPjRtBaseDeviceAttributes(options.device_name_prefix, options.device_name,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 2.5K bytes - Viewed (0) -
tensorflow/compiler/jit/pjrt_base_device.h
namespace tensorflow { // tensorflow::PjRtBaseDevice replaces the deprecated tensorflow::XlaDevice. // This accelerator agnostic device is mainly used to store metadata. class PjRtBaseDevice : public LocalDevice { public: // Stores metadata about the PjRtBaseDevice. class Metadata { public: Metadata(const DeviceType& jit_device_type, std::vector<XlaShapeLayoutHelpers::ShapeDeterminationFns>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.h
#include "tensorflow/core/platform/mutex.h" #include "tensorflow/core/platform/stream_executor_no_cuda.h" #include "tensorflow/core/tfrt/common/async_value_tensor.h" namespace tensorflow { class XlaDevice : public LocalDevice { public: // Given a tensor, sets `xla::Shape*` the shape of tensor's representation // on device, fully padded. On error, the contents of `xla::Shape*` // are undefined.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_device.cc
XlaDevice::global_compute_streams_ = new std::vector<std::shared_ptr<se::Stream>>; XlaDevice::XlaDevice(const SessionOptions& session_options, const Options& options) : LocalDevice(session_options, BuildXlaDeviceAttributes(options.device_name_prefix, options.device_name, options.device_ordinal)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 21:05:42 UTC 2024 - 24.3K bytes - Viewed (0)