- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for removeResultAttr (0.13 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/drop_savedmodel_semantics.cc
func.removeArgAttr(i, bound_input); } func.removeArgAttr(i, index_path); } for (unsigned i = 0, e = func.getNumResults(); i != e; ++i) { func.removeResultAttr(i, bound_input); func.removeResultAttr(i, index_path); } }); // Clean up modules from tf_saved_model attributes. module->removeAttr("tf_saved_model.semantics"); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 13 23:39:10 UTC 2022 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_saved_module_metadata.cc
func.getResultAttrs(i), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); })); for (auto namedAttr : stripAttrs) { func.removeResultAttr(i, namedAttr.getName()); } } } void StripSavedModuleMetadataPass::runOnOperation() { auto module = getOperation(); StripModule(module); module.walk(StripFunction); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/strip_tf_attributes.cc
func.getResultAttrs(i), [](NamedAttribute namedAttr) { return ShouldStripAttr(namedAttr); })); for (auto namedAttr : stripAttrs) { func.removeResultAttr(i, namedAttr.getName()); } } } void StripTfAttributesPass::runOnOperation() { ModuleOp module = getOperation(); // strip module itself
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/add_functions_for_exported_names.cc
} } } for (int i = 0; i < f.getNumResults(); ++i) { for (NamedAttribute a : f.getResultAttrs(i)) { if (a.getName().strref().starts_with("tf_saved_model.")) { f.removeResultAttr(i, a.getName()); } } } } void AddFunctionsForExportedNamesPass::runOnOperation() { Block& module_body = getOperation().getRegion().front(); for (auto f :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 08:06:04 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
for (auto& attr : iface.getResultAttrs(i)) { const StringAttr& attr_name = attr.getName(); if (attr_name.getValue().starts_with("tf_saved_model.")) { func.removeResultAttr(i, attr_name); } } } } // Information to identify an output in its node and in the model output list. // Ex: If the model output list is ["add:0", "topk:0": "topk:1"], then the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0)