- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for UnwrapXlaCallModuleOpPass (0.19 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
// Unwraps XlaCallModule ops without quantizable trait that call function with // '_from_xla_call_module' trait. class UnwrapXlaCallModuleOpPass : public impl::UnwrapXlaCallModuleOpPassBase<UnwrapXlaCallModuleOpPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(UnwrapXlaCallModuleOpPass) explicit UnwrapXlaCallModuleOpPass() = default; private: void runOnOperation() override; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td
let summary = "Merge quantized conv/dot_general fusion with subsequent dequantize."; let dependentDialects = [ "chlo::ChloDialect", "mlir::stablehlo::StablehloDialect", ]; } def UnwrapXlaCallModuleOpPass : Pass<"stablehlo-unwrap-xla-call-module-op", "ModuleOp"> { let summary = "Unwrap XlaCallModuleOps into inline functions if not used for quantizing fused patterns."; let dependentDialects = ["TF::TensorFlowDialect"];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 10.3K bytes - Viewed (0)