- Sort Score
- Result 10 results
- Languages All
Results 171 - 180 of 271 for getOperation (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
registry.insert<mlir::tf_saved_model::TensorFlowSavedModelDialect, ml_program::MLProgramDialect>(); } void runOnOperation() override { if (failed(convertTFGlobals(getOperation()))) { signalPassFailure(); } } }; std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass() { return std::make_unique<LowerGlobalsToMlProgram>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/lift_quantizable_spots_as_functions.cc
} } return success(); } void LiftQuantizableSpotsAsFunctionsPass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx); ModuleOp module_op = getOperation(); simple_patterns::populateWithGenerated(patterns); fusion_patterns::populateWithGenerated(patterns); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
if (dequantize_op.use_empty()) { dequantize_op.erase(); } } } return success(); } void ModifyIONodesPass::runOnOperation() { auto func = getOperation(); auto attrs = func->getAttrOfType<mlir::DictionaryAttr>("tf.entry_function"); // Handle the entry functions only. if (func.getName() != "main" && (!attrs || attrs.empty())) { return; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/pin_ops_with_side_effects.cc
private: // This will be used recursively. const std::function<bool(Operation *)> op_has_side_effects_; }; void PinOpsWithSideEffectsPass::runOnOperation() { auto fn = getOperation(); // We're assuming (and checking) that there are no tfl::ControlNodeOps present // before this pass. We could relax this requirement by defining folding logic // for them. if (fn.walk([&](TFL::ControlNodeOp) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
class XlaCallModuleSerializationPass : public impl::XlaCallModuleSerializationPassBase< XlaCallModuleSerializationPass> { public: void runOnOperation() override { mlir::ModuleOp module = getOperation(); mlir::SymbolTableCollection symbol_table; mlir::WalkResult result = module.walk([&](mlir::TF::XlaCallModuleOp xla_call_module) { if (failed(SerializeXlaCallModule(symbol_table, module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
plugin/pkg/admission/certificates/ctbattest/admission_test.go
return t.obj } func (t *testAttributes) GetOldObject() runtime.Object { return t.oldObj } func (t *testAttributes) GetName() string { return t.name } func (t *testAttributes) GetOperation() admission.Operation { return t.operation } func (t *testAttributes) GetUserInfo() user.Info { return &user.DefaultInfo{Name: "ignored"}
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Wed Apr 24 18:25:29 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
builder.create<func::ReturnOp>( NameLoc::get(builder.getStringAttr(kTfQuantSaveReturnOpName))); } void InsertSaveOpPass::runOnOperation() { ModuleOp module_op = getOperation(); func::FuncOp session_init_func = GetInitializerFunction( module_op, /*initializer_type=*/kTfSavedModelInitializerRestoreType); if (!session_init_func) {
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
} }; // TODO: b/264218457 - Refactors the current file to parse preset quantization // options and allow modular control of quantization specs. void QuantizeWeightPass::runOnOperation() { func::FuncOp func = getOperation(); MLIRContext* ctx = func.getContext(); RewritePatternSet patterns(ctx); patterns.add<QuantizeWeight>(ctx, quantization_component_spec_); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
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
<< key; break; } } } return attrs; } }; void TflToStablehloPass::runOnOperation() { func::FuncOp fn = getOperation(); OpBuilder builder(fn.getContext()); fn.walk([&](TFL::CustomOp custom_op) { builder.setInsertionPoint(custom_op); const uint8_t* option_buf = reinterpret_cast<const uint8_t*>(
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/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
<< " outside compilation regions."; } } void MarkOpsForOutsideCompilation::runOnOperation() { auto module = getOperation(); const Dialect* tf_dialect = getContext().getLoadedDialect("tf"); if (!tf_dialect) { getOperation().emitError() << "'tf' dialect is not registered"; return signalPassFailure(); } RewritePatternSet patterns(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0)