- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for session_init_funcs (0.45 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
absl::StrCat("failed to fetch device manager: ", status.message())); return failure(); } SmallVector<func::FuncOp, 2> session_init_funcs = tf_saved_model::GetInitializerFunctions(module); func::FuncOp session_init_func = session_init_funcs.empty() ? nullptr : session_init_funcs[0]; TF::ResourceAnalyzer analyzer(module, /*skip_session_init=*/true); llvm::SmallVector<TF::VarHandleOp, 4> variables;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
const_op.getResult().replaceAllUsesWith(read_variable_op); } return const_op_name_map; } // Inside `session_init_func`, creates AssignVariableOps(VarHandleOp, ConstOp) // for each VarHandleOp that replaces a ConstOp. The `session_init_func` will // essentially behave like restore_op for the newly created VarHandleOps whose // shared names are the values of `const_op_name_map`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0)