- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetVariableFromSession (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/session_utils.cc
std::vector<std::string> names; names.reserve(var_handle_ops.size()); for (auto var_handle_op : var_handle_ops) names.push_back(GetVariableName(var_handle_op)); return names; } tensorflow::Var* GetVariableFromSession(mlir::TF::VarHandleOp var_handle_op, llvm::StringRef device_name, const tensorflow::DeviceMgr* mgr) { tensorflow::Device* device = nullptr;
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
InitializeVariable(var_op, &resource_tensor, session_init_func, builder); continue; } auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()(); auto* var_ptr = GetVariableFromSession(var_op, handle.device(), mgr); if (!var_ptr) { // If no value in session, then just skip this variable. // This can happen if the variable is not saved in checkpoint.
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/transforms/tf_saved_model_freeze_variables.cc
return GetTensorValueAsElementsAttr(resource_tensor, builder); } auto handle = resource_tensor.scalar<tensorflow::ResourceHandle>()(); auto* var_ptr = tf_saved_model::GetVariableFromSession(var_handle_op, handle.device(), mgr); if (!var_ptr) { return nullptr; } tensorflow::core::RefCountPtr<tensorflow::Var> var(var_ptr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0)