- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass (1 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
constexpr StringRef kPlatformTpu = "TPU"; class ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass : public impl:: ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPassBase< ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID( ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/unwrap_xla_call_module_op.cc
.getValue(); func::FuncOp func_op = symbol_table.lookup<func::FuncOp>(function_name); // We should not unwrap if the function is not from // ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass. if (!func_op->hasAttr(TF::kFromXlaCallModuleAttrName)) { return; } MLIRContext* context = call_op.getContext(); OpBuilder builder(context);
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/cc/pass_pipeline.cc
void AddCallModuleSerializationPasses(OpPassManager& pm) { AddShapeLegalizationPasses(pm); pm.addPass(createReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass()); // ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass may create // duplicate constants. Add canonicalizer to deduplicate. pm.addNestedPass<func::FuncOp>(mlir::createCanonicalizerPass()); pm.addPass(TF::CreateXlaCallModuleSerializationPass()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td
}]; let dependentDialects = [ "mlir::func::FuncDialect", "mlir::stablehlo::StablehloDialect", "TF::TensorFlowDialect", ]; } def ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass : Pass<"stablehlo-replace-stablehlo-ops-in-main-function-with-xla-call-module-ops", "mlir::ModuleOp"> { let summary = "Replaces the StableHLO ops with a separate XlaCallModuleOps."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 10.3K bytes - Viewed (0)