- Sort Score
- Result 10 results
- Languages All
Results 171 - 180 of 293 for runOnOperation (0.16 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
return "Inserts a new function that wraps a SaveV2 op. The SaveV2 op saves " "the values of the VarHandleOps that are found in the initializer " "function of 'restore_op' type."; } void runOnOperation() override; }; // Finds `tf.AssignVariableOp(tf.VarHandleOp, tf.Const)` patterns and removes // `tf.AssignVariableOp`s and `tf.Const`s. Collects and returns the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_weight.cc
public: explicit QuantizeWeightPass( QuantizationComponentSpec quantization_component_spec) : quantization_component_spec_(quantization_component_spec) {} private: void runOnOperation() override; QuantizationComponentSpec quantization_component_spec_; }; // Collects quantizable target ops, then insert Q-DQ quantization patterns. class QuantizeWeight : public OpRewritePattern<ConstantOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
StringRef getArgument() const final { return "tfl-parse-stablehlo-ops"; } StringRef getDescription() const final { return "This pass will legalize TFLite custom Ops to StableHLO ops."; } private: void runOnOperation() override; void getDependentDialects(DialectRegistry& registry) const override { mlir::stablehlo::registerAllDialects(registry); } inline TFL::ConstBytesAttr CustomOption(OpBuilder* builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/drop_savedmodel_semantics.cc
} // NOTE: The implementation is mostly copypasted from // third_party/tensorflow/compiler/mlir/tfrt/transforms/lower_saved_model.cc // with the original code trimmed and adapted as needed. void runOnOperation() override { auto module = getOperation(); if (!tf_saved_model::HasTfSavedModelSemantics(module)) return; // Clean up functions from tf_saved_model attributes. OpBuilder builder(module);
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/lite/transforms/legalize_variables.cc
class LegalizeVariablesPass : public impl::LegalizeVariablesPassBase<LegalizeVariablesPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeVariablesPass) void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_tensor_list_init_ops_pass.cc
// This transformation pass replicates TensorList initialization ops. class ReplicateTensorListInitOps : public impl::ReplicateTensorListInitOpsPassBase< ReplicateTensorListInitOps> { public: void runOnOperation() override { getOperation().walk([](Operation* op) { if (auto tl_reserve = dyn_cast<TensorListReserveOp>(op)) { ReplicateTensorListForUses(tl_reserve); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Jan 22 17:28:34 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/import_quant_stats_pass.cc
return "quant-import-stats"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Import quantization stats to the model"; } void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 08 10:41:08 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
} void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, quant::QuantizationDialect>(); } private: void runOnOperation() override; bool test_mode_; tensorflow::quantization::QuantizationOptions quant_options_; // Initialize for tests. void initializeForTest() { if (!test_mode_) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/per_function_aggregate_analysis.h
OperationPass<ModuleOp>> { public: static ::mlir::TypeID resolveTypeID() { static ::mlir::SelfOwningTypeID id; return id; } private: void runOnOperation() override { ModuleOp op = this->getOperation(); DerivedT& derived = *static_cast<DerivedT*>(this); auto& analysis = this->template getAnalysis<AnalysisT>(); for (auto func : op.getOps<func::FuncOp>())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
public: StringRef getArgument() const final { return "rename-entrypoint-to-main"; } StringRef getDescription() const final { return "Renames the entrypoint in SavedModel to `main`"; } void runOnOperation() override { auto fail = [&](Operation* op, std::string message) { op->emitError(message); signalPassFailure(); }; DenseMap<StringRef, func::FuncOp> entrypoints;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0)