- Sort Score
- Result 10 results
- Languages All
Results 11 - 11 of 11 for FindMainFuncOp (0.11 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
} } } void ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass:: runOnOperation() { ModuleOp module_op = getOperation(); func::FuncOp main_func = FindMainFuncOp(module_op); if (!main_func) return; // In case the model has tf.StatefulPartitionedCallOp or tf.PartitionedCallOp, // we recursively find called functions and process StableHLO ops in them.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0)