- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for mgr2 (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc
tensorflow::Var* GetVariableFromSession(mlir::TF::VarHandleOp var_handle_op, llvm::StringRef device_name, const tensorflow::DeviceMgr* mgr) { tensorflow::Device* device = nullptr; if (!mgr || !mgr->LookupDevice(StringRefToView(device_name), &device).ok()) return nullptr; tensorflow::Var* var_ptr = nullptr; const auto& container = var_handle_op.getContainer().str();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.2K bytes - Viewed (0) -
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)