- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for allow_mutable_tensors (0.54 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
struct FreezeGlobalTensorsPass : public impl::FreezeGlobalTensorsPassBase<FreezeGlobalTensorsPass> { explicit FreezeGlobalTensorsPass(bool allow_mutable_tensors) { this->allow_mutable_tensors = allow_mutable_tensors; } void runOnOperation() override; }; void FreezeGlobalTensorsPass::runOnOperation() { auto module = getOperation();
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/tensorflow/tests/tf_saved_model_freeze_global_tensors_mutable_tensors.mlir
// RUN: tf-opt "-tf-saved-model-freeze-global-tensors=allow-mutable-tensors=true" -split-input-file %s | FileCheck %s module attributes {tf_saved_model.semantics} { // Test case: Do not fail if the tensor is mutable but allow_mutable_tensors is true "tf_saved_model.global_tensor"() { is_mutable, sym_name = "v", type = tensor<f32>, value = dense<1.0> : tensor<f32> } : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 703 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_passes.h
std::unique_ptr<OperationPass<ModuleOp>> CreateOptimizeGlobalTensorsPass(); // Creates a pass that freezes tf_saved_model.global_tensor ops. std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeGlobalTensorsPass( bool allow_mutable_tensors = false); // Creates a pass that freezes tf_saved_model.asset ops. std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeAssetsPass( std::string saved_model_dir = "");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 09 19:11:34 UTC 2023 - 3.8K bytes - Viewed (0) -
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()); // Legalizes TF UniformQuantized types into MHLO.
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/tf_savedmodel_passes.td
pass lowers from saved model to pure TF. Hence it fails, if it cannot lower. }]; let constructor = "::mlir::tf_saved_model::CreateFreezeGlobalTensorsPass()"; let options = [ Option<"allow_mutable_tensors", "allow-mutable-tensors", "bool", /*default=*/"false", "Allows mutable tensors to be in the graph."> ]; } def OptimizeGlobalTensorsPass : Pass<"tf-saved-model-optimize-global-tensors", "ModuleOp"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 09 19:11:34 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_passes.cc
// 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 // in a format of the tf constant op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 18:45:51 UTC 2024 - 25.5K bytes - Viewed (0)