- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 75 for registry_ (0.1 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.cc
} if (input_mlir) { mlir::DialectRegistry registry; registry.insert<mlir::TFL::TensorFlowLiteDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect>(); mlir::func::registerAllExtensions(registry); context->appendDialectRegistry(registry); source_mgr->AddNewSourceBuffer(std::move(buffer), llvm::SMLoc()); return mlir::OwningOpRef<mlir::ModuleOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 03:47:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
// This is a brief description of the pass. return "Convert tf.CustomAggregator op to quant.Stats"; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TF::TensorFlowDialect>(); registry.insert<quant::QuantizationDialect>(); registry.insert<quantfork::QuantizationForkDialect>(); } void runOnOperation() override; }; class ConvertCustomAggregationOpToQuantStats
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api.h
// Sparsifies model to encode sparse tensors with proper format. Throws error if // sparsification fails. PyObject* MlirSparsifyModel(PyObject* data); // Registers the given custom opdefs to TensorFlow global op registry. PyObject* RegisterCustomOpdefs(PyObject* list); // Returns the collected TFLite conversion errors. std::vector<std::string> RetrieveCollectedErrors(); // Returns MLIR string dump of the given Flatbuffer model.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 18:18:30 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
std::vector<TensorShape>& arg_shapes, tpu::TPUCompileMetadataProto& metadata_proto) { mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR( DeserializeMlirModule(mlir_module_str, &context, &mlir_module));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate.cc
// back to do it properly in the future mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); registry .insert<mlir::func::FuncDialect, mlir::stablehlo::StablehloDialect>(); context.appendDialectRegistry(registry); } absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> module; std::unordered_set<std::string> tags; tensorflow::GraphImportConfig specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/tf_to_stablehlo_translate.cc
llvm::cl::ParseCommandLineOptions(argc, argv, "TF Saved Model to Stablehlo converter\n"); mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); bool is_input_mlir_module = absl::EndsWith(input_path, ".mlir"); std::vector<std::string> exported_model_signatures_in_vector =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 4.6K bytes - Viewed (0) -
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/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)