- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateFreezeGlobalTensorsPass (0.36 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/transforms.cc
// freezing of TF v2 GlobalTensor ops. (Ref: b/206855389) pm.addPass(mlir::tf_saved_model::CreateOptimizeGlobalTensorsPass()); pm.addPass(mlir::tf_saved_model::CreateFreezeGlobalTensorsPass( /*allow_mutable_tensors=*/true)); // Generic MLIR optimization passes. pm.addPass(mlir::createCanonicalizerPass()); pm.addPass(mlir::TF::CreateTFShapeInferencePass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 04:34:23 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
// freezing of TF v2 GlobalTensor ops. (Ref: b/206855389) pm.addPass(mlir::tf_saved_model::CreateOptimizeGlobalTensorsPass()); pm.addPass(mlir::tf_saved_model::CreateFreezeGlobalTensorsPass( /*allow_mutable_tensors=*/true)); // Generic MLIR optimization passes. pm.addPass(mlir::createCanonicalizerPass()); pm.addPass(mlir::TF::CreateTFShapeInferencePass(input_arg_shapes));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
module.emitError() << "could not freeze all global tensors in the module"; return signalPassFailure(); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeGlobalTensorsPass( bool allow_mutable_tensors) { return std::make_unique<FreezeGlobalTensorsPass>(allow_mutable_tensors); } } // namespace tf_saved_model
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
if (!pass_config.disable_variable_freezing) { // This pass 'freezes' immutable global tensors and inlines them as tf // constant ops. pass_manager->addPass(mlir::tf_saved_model::CreateFreezeGlobalTensorsPass( /*allow_mutable_tensors=*/pass_config.enable_tflite_variables)); } if (!saved_model_dir.empty()) { // This pass 'freezes' tf saved model asset ops and inlines as string values
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0)