- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for GetInitializerFunction (0.18 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables.cc
#include "tsl/platform/logging.h" #include "tsl/platform/status.h" namespace tensorflow { namespace quantization { namespace { using ::mlir::func::FuncOp; using ::mlir::tf_saved_model::GetInitializerFunction; using ::mlir::tf_saved_model::kTfSavedModelInitializerRestoreType; // Adds the tensor that initializes the variable through the provided // `assign_var_op` to the `bundle_writer` for saving to checkpoint. Returns the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:36:55 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h" #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h" namespace mlir { namespace quant { namespace { using ::mlir::tf_saved_model::GetInitializerFunction; using ::mlir::tf_saved_model::kTfSavedModelInitializerRestoreType; constexpr StringRef kTfQuantSaveV2OpName = "tf_quant__save_save_v2"; constexpr StringRef kTfQuantSaveReturnOpName = "tf_quant__save_return";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h
// Returns the initializer function whose `tf_saved_model.initializer_type` // attribute matches `initializer_type`. Returns a null op if it doesn't exist. func::FuncOp GetInitializerFunction(ModuleOp module_op, StringRef initializer_type); // Checks if the module restores variables from a Checkpoint. bool IsRestoreGraph(ModuleOp module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc
func::FuncOp GetOrCreateSessionInitFunc(ModuleOp module) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module); if (!session_init_op) return CreateSessionInitFunc(module); auto init_func_op = GetInitializerFunction( module, /*initializer_type=*/kTfSavedModelInitializerRestoreType); if (init_func_op) { return init_func_op; } else if (!session_init_op.getInitializers().empty()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0)