- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for CreateDevices (0.44 sec)
-
tensorflow/compiler/jit/xla_cpu_device.cc
namespace tensorflow { using tensorflow::IdentityShapeRepresentationFn; class XlaCpuDeviceFactory : public DeviceFactory { public: Status ListPhysicalDevices(std::vector<string>* devices) override; Status CreateDevices(const SessionOptions& options, const string& name_prefix, std::vector<std::unique_ptr<Device>>* devices) override; }; Status XlaCpuDeviceFactory::ListPhysicalDevices(std::vector<string>* devices) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_gpu_device.cc
#include "tensorflow/core/lib/core/status.h" namespace tensorflow { class XlaGpuDeviceFactory : public DeviceFactory { public: Status ListPhysicalDevices(std::vector<string>* devices) override; Status CreateDevices(const SessionOptions& options, const string& name_prefix, std::vector<std::unique_ptr<Device>>* devices) override; }; Status XlaGpuDeviceFactory::ListPhysicalDevices(std::vector<string>* devices) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_tpu_device.cc
if (!status.ok()) { done(status); } } class TpuNodeDeviceFactory : public DeviceFactory { public: Status ListPhysicalDevices(std::vector<string>* devices) override; Status CreateDevices(const SessionOptions& options, const string& name_prefix, std::vector<std::unique_ptr<Device>>* devices) override; }; Status TpuNodeDeviceFactory::ListPhysicalDevices(std::vector<string>* devices) {
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_host_send_recv_device_context_test.cc
void SetDevice(const string& device_type) { auto device_factory = DeviceFactory::GetFactory(device_type); SessionOptions options; std::vector<std::unique_ptr<Device>> devices; Status s = device_factory->CreateDevices( options, "/job:worker/replica:0/task:0", &devices); device_ = std::move(devices[0]); AllocatorAttributes host_alloc_attr; host_alloc_attr.set_on_host(true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
if (!cpu_factory) { return errors::NotFound( "CPU Factory not registered. Can't run EncapsulateSubgraphsPass"); } TF_RETURN_IF_ERROR(cpu_factory->CreateDevices( session_options, "/job:localhost/replica:0/task:0", &devices)); if (devices.empty()) { return errors::NotFound( "Failed to create a CPU device for EncapsulateSubgraphsPass"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
RELEASE.md
`state_is_tuple=True`. For a quick fix while transitioning to the new default, simply pass the argument `state_is_tuple=False`. * DeviceFactory's AddDevices and CreateDevices functions now return a Status instead of void. * Int32 elements of list(type) arguments are no longer placed in host memory by default. If necessary, a list(type) argument to a kernel can be placed in
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)