- Sort Score
- Result 10 results
- Languages All
Results 91 - 100 of 445 for FuncOp (0.15 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
// is "restore_op". Creates and returns a new initializer function iff such // `FuncOp` is not found. The newly created initializer function's // initializer_type is "restore_op" and its symbol will be added to the symbol // table and session_initializer op's "intializer" attribute. func::FuncOp GetOrCreateInitializerFunc(ModuleOp module_op) { if (auto init_func_op = GetInitializerFunction(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
namespace mlir { namespace TFL { namespace { template <int NInput, int NOutput> func::FuncOp createMaxUnpoolingFunc( mlir::Builder* builder, const SmallVector<mlir::Type, NInput>& input_types, const SmallVector<mlir::Type, NOutput>& output_types) { auto func_type = builder->getFunctionType(input_types, output_types); auto func = func::FuncOp::create( mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.cc
} // Returns the entry function, i.e. the callee of `xla_call_module_op`. func::FuncOp GetEntryFuncOp(TF::XlaCallModuleOp xla_call_module_op, const SymbolTable symbol_table) { const auto entry_function_symbol_ref = xla_call_module_op->getAttrOfType<FlatSymbolRefAttr>(kEntryFuncAttrName); return dyn_cast_or_null<func::FuncOp>( symbol_table.lookup(entry_function_symbol_ref.getValue())); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func.h
#include "mlir/IR/BuiltinOps.h" // from @llvm-project namespace mlir::quant { // Returns a public `func::FuncOp` in `module_op` whose name matches either // `main` or `serving_default`. If `func::FuncOps` with both names exist, the // function with name "main" takes precedence. Returns null if no such a // function exists. func::FuncOp FindMainFuncOp(ModuleOp module_op); } // namespace mlir::quant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/nms_utils.cc
builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); return success(); } LogicalResult ConvertSSDPostProcessFunc::CreateNMSCustomOptions( func::FuncOp func, DictionaryAttr attrs, std::string& custom_option_buffer) { flexbuffers::Builder fbb; size_t start_map = fbb.StartMap(); if (failed(AddIntAttr(func, attrs, "max_detections", &fbb)) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
auto fail = [&](Operation* op, std::string message) { op->emitError(message); signalPassFailure(); }; DenseMap<StringRef, func::FuncOp> entrypoints; auto module = getOperation(); module.walk([&](func::FuncOp op) { auto visibility = SymbolTable::getSymbolVisibility(op); if (visibility != SymbolTable::Visibility::Public) return; entrypoints[op.getSymName()] = op; });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor.cc
/*legacy_graph_export=*/true)); add_pass(mlir::TFDevice::CreateLaunchToDeviceAttributePass( /*legacy_graph_export=*/true)); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUDevicePropagationPass()); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUColocateSplitsPass()); pm.addPass(mlir::createSymbolDCEPass()); if (tensorflow::GetMlirCommonFlags() ->tf_mlir_enable_convert_control_to_data_outputs_pass) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 16 03:41:02 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
op->removeAttr(TF::kClusterOutlinedFunctionNameAttr); func_name = outlined_func_name.str(); } else { func_name = "_func"; } func::FuncOp outlined_func = func::FuncOp::create(op.getLoc(), func_name, func_type); // This function is not externally visible and marking it private would allow // symbol-dce pass to remove it when it is not referenced anymore.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
using mlir::ModuleOp; using mlir::OpBuilder; using mlir::Operation; using mlir::OperationPass; using mlir::SymbolTable; using mlir::SymbolTableCollection; using mlir::SymbolUserOpInterface; using mlir::func::FuncOp; #define GEN_PASS_DEF_XLACLUSTERFORMATIONPASS #include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" // Outlines partitioned call ops with `_XlaMustCompile` to device clusters.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/passes.h
// Verifies that the TF/XLA ops have all been lowered to MHLO. std::unique_ptr<OperationPass<func::FuncOp>> CreateVerifyTFXLALegalizationPass( bool legalize_chlo = true); // Transforms TFXLA Device specific ops into device independent ops. std::unique_ptr<OperationPass<func::FuncOp>> CreateTFXLADeviceSpecificTransformsPass( std::optional<StringRef> tf2xla_fallback_device_type = std::nullopt);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 4.8K bytes - Viewed (0)