- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for LocalDeviceManager (0.88 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
} } // namespace LogicalResult InitializeVariablesInSessionInitializer( ModuleOp module, tensorflow::Session* session) { const tensorflow::DeviceMgr* mgr = nullptr; auto status = session->LocalDeviceManager(&mgr); if (!status.ok()) { module->emitError( absl::StrCat("failed to fetch device manager: ", status.message())); return failure(); } // Fetch all VarHandleOp.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/fake_session.cc
} Status FakeSession::ListDevices( std::vector<tensorflow::DeviceAttributes>* response) { return tensorflow::errors::Unimplemented("not available"); } Status FakeSession::LocalDeviceManager( const tensorflow::DeviceMgr** deviceMgrPtr) { Initialize(); if (kSessionOptions->fail_to_fetch_local_device_manager) return Status(absl::StatusCode::kUnknown, "No Local Device Manager");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
if (!status.ok()) { return module.emitOpError() << "failed to run the provided session: " << status.message(); } const DeviceMgr* device_manager; if (!(session->LocalDeviceManager(&device_manager).ok())) { return module.emitOpError() << "failed to get local device manager"; } // Read all underlying tensors of the variables from the session. std::vector<Tensor> tensors;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0)