- Sort Score
- Result 10 results
- Languages All
Results 51 - 60 of 99 for registry_ (0.1 sec)
-
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
StringRef getDescription() const final { return "Smuggle disallowed ops via stablehlo.custom_calls"; } void getDependentDialects(DialectRegistry& registry) const final { registry.insert<mlir::stablehlo::StablehloDialect>(); } void runOnOperation() override { RewritePatternSet patterns(&getContext()); patterns.add<SmuggleOpPattern<TF::ResizeBilinearOp>>(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc
StringPiece tfr_raw_text, mlir::MLIRContext* mlir_ctx) { mlir_ctx->allowUnregisteredDialects(/*allow=*/true); // Load dialects involved in the conversion mlir::DialectRegistry registry; // clang-format off registry.insert<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::scf::SCFDialect, mlir::shape::ShapeDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:34:43 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
"initializer functions will be removed after this pass."; } void runOnOperation() override; private: void getDependentDialects(DialectRegistry& registry) const override { registry .insert<TF::TensorFlowDialect, tf_executor::TensorFlowExecutorDialect, tf_saved_model::TensorFlowSavedModelDialect>(); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc
EXPECT_EQ(non_categorized_count, 428); } // Just a counter test to see which ops have duplicate lowerings. This isn't a // correctness test versus a test to easily ensure the op registry is kept // in sync. TEST_F(LegalizationOpConfigTest, CountTypesWhichHaveBothMlirAndTf2xlaFallback) { int double_lowering_count = 0; DialectRegistry dialect_registry;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/api_template.__init__.py
setattr(_current_module, "initializers", _initializers) # Do an eager load for Keras' code so that any function/method that needs to # happen at load time will trigger, eg registration of optimizers in the # SavedModel registry. # See b/196254385 for more details. try: if _tf_uses_legacy_keras: importlib.import_module("tf_keras.src.optimizers") else: importlib.import_module("keras.src.optimizers")
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 06:27:59 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc
return "Replace quantization candidates with composite functions into the " "module for post-training dynamic range case"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: Option<OpSet> target_opset_{ *this, "target-opset", llvm::cl::init(OpSet::TF),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc
StringRef getDescription() const final { // This is a brief description of the pass. return "Insert quantized functions into the module"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, func::FuncDialect>(); } private: void runOnOperation() override; // Returns the function library for the given quantization method and opset // pair.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
// name should be unique to the graph it is being inserted to. If the // `ignore_unregistered_attrs` argument is set to true, the attributes which are // not in the op registry will be ignored. If the `ignore_unregistered_attrs` // argument is not set to true, _output_shapes attribute is added to nodes with // ShapedType for the leading values with ShapedType in the results of the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
// Explicitly disable dumping Op details on failures. module.getContext()->printOpOnDiagnostic(false); mlir::DialectRegistry registry; mlir::func::registerAllExtensions(registry); module.getContext()->appendDialectRegistry(registry); mlir::StatusScopedDiagnosticHandler status_handler(module.getContext(), /*propagate=*/true);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0)