- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOps (0.5 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
// Replaces the StableHLO ops in the main function block with // tf.XlaCallModuleOps as separate subgraphs. Wires them back to the main // function block to be compatible with SavedModel structure. void ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOps( ModuleOp module_op, func::FuncOp main_func, int& stablehlo_func_id) { Block& main_func_block = main_func.getBody().front();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0)