- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for CreateConvertTFQuantTypesPass (0.77 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.td
argument/result allow qint type only. For such cases, add qint <-> int tf.Cast around the ops so that they are still valid. }]; let constructor = "mlir::quant::stablehlo::CreateConvertTFQuantTypesPass()"; let dependentDialects = ["TF::TensorFlowDialect", "tf_type::TFTypeDialect"]; } def VerifyQuantLegalization : Pass<"verify-quant-legalization", "mlir::func::FuncOp"> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.h
// Creates an instance of the ConvertTFQuantTypes pass, which will convert TF // qint types to int types and surround TF UniformQuantized ops with qint <-> // int casts. std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertTFQuantTypesPass(); // Creates an instance of the VerifyQuantLegalization pass, which verifies all // quant ops and types are lowered. std::unique_ptr<OperationPass<func::FuncOp>> CreateVerifyQuantLegalizationPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 01:41:18 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types_test.cc
tensorflow::DeserializeMlirModule(module_string, &context_, &module_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>( quant::stablehlo::CreateConvertTFQuantTypesPass()); } mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: MLIRContext context_; OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
if (failed(applyFullConversion(getOperation(), target, std::move(patterns)))) return signalPassFailure(); } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateConvertTFQuantTypesPass() { return std::make_unique<ConvertTFQuantTypes>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
// Run the Convert TF Quant Types, TF Quant -> MHLO Quant and MHLO Quant -> // MHLO int passes. PassManager pm(module_op->getContext()); pm.addNestedPass<func::FuncOp>(CreateConvertTFQuantTypesPass()); AddQuantizationLoweringPasses(pm); CHECK(succeeded(pm.run(module_op.get()))); // Compile the program. return pjrt_client_->Compile(*module_op, xla::CompileOptions{}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
} pm.addNestedPass<mlir::func::FuncOp>(mlir::TF::CreateLowerQuantizedPass()); pm.addNestedPass<mlir::func::FuncOp>( mlir::quant::stablehlo::CreateConvertTFQuantTypesPass()); if (lower_to_xla_hlo) { for (auto& target_pass : custom_legalization_passes) { pm.addNestedPass<mlir::func::FuncOp>(std::move(target_pass)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0)