- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for CreateInsertSaveOpPass (0.22 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/python/unfreeze_constants.cc
/*name=*/kTfQuantInsertRestoreOpStepName, /*add_passes_func=*/ [](mlir::PassManager &pm) { pm.addPass(mlir::quant::CreateInsertRestoreOpPass()); pm.addPass(mlir::quant::CreateInsertSaveOpPass()); // Initialization by `tf.ConstOp` is no longer required as there is // a `tf.RestoreV2Op` now. pm.addPass( mlir::quant::CreateRemoveVariableInitializationByConstPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 11:07:03 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
return; } CreateSaveFunc(module_op, target_var_handle_ops); } static PassRegistration<InsertSaveOpPass> pass{}; } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateInsertSaveOpPass() { return std::make_unique<InsertSaveOpPass>(); } } // namespace quant
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/quantization/tensorflow/passes/passes.h
// `tf.AssignVariableOp(tf.VarHandleOp, tf.Const)` pattern in the initializer // function of type "restore_op" to identify the VarHandleOps that should be // saved using the SaveV2 op. std::unique_ptr<OperationPass<ModuleOp>> CreateInsertSaveOpPass(); // Creates a pass that marks functions with the attribute `tf._noinline = true` // to avoid being inlined by the `InlinerPass`. `noinline_functions` is the name // of the functions to mark.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 12.3K bytes - Viewed (0)