- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 145 for registry_ (0.21 sec)
-
tensorflow/compiler/mlir/python/mlir_wrapper/mlir_wrapper.cc
PYBIND11_MODULE(mlir_wrapper, m) { m.def("preloadTensorFlowDialects", [](mlir::MLIRContext &context) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); context.appendDialectRegistry(registry); context.loadAllAvailableDialects(); }); m.def("verify", [](std::string input) { llvm::SourceMgr SM = llvm::SourceMgr();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 08:41:42 UTC 2023 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/analysis/update_op_cost_in_tfrt_mlir_test.cc
"tensorflow/compiler/mlir/tfrt/tests/analysis/testdata/test.mlir"); mlir::DialectRegistry registry; tfrt::RegisterTFRTDialects(registry); registry.insert<tfrt::fallback_async::FallbackAsyncDialect>(); registry.insert<tfrt::fallback_sync::FallbackSyncDialect>(); mlir::MLIRContext context(registry); auto module = mlir::parseSourceFile<mlir::ModuleOp>(saved_model_mlir_path, &context); ASSERT_TRUE(module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jul 21 22:52:12 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/register_common_dialects_test.cc
#include "mlir/IR/DialectRegistry.h" // from @llvm-project namespace mlir { namespace { TEST(RegisterCommonDialectsTest, DoesntCrash) { mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); EXPECT_FALSE(registry.getDialectNames().empty()); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 21 20:13:57 UTC 2023 - 1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/pywrap_tensorflow_to_stablehlo_lib.cc
const std::vector<std::string>& exported_model_signatures, const std::vector<std::string>& tag_names, absl::string_view input_arg_shapes_str) { mlir::DialectRegistry registry; RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); context.loadAllAvailableDialects(); auto module = TfToStablehlo(input_path, &context, exported_model_signatures, tag_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 22:58:42 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_fake_quant_to_qdq.cc
return "Convert Fake Quant op to quant.qcast and quant.dcast pairs"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); registry.insert<quant::QuantizationDialect>(); registry.insert<quantfork::QuantizationForkDialect>(); } void runOnOperation() override; }; static PassRegistration<ConvertFakeQuantToQdqPass> pass;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 15 00:56:15 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tpu_embedding_ops_registry.h
namespace mlir { namespace TF { // A global ops registry that is used to hold TPU embedding ops. // // Example: // TPUEmbeddingOpsRegistry::Global().Add<TF::FooOp>(); // for (auto op_type_id : TPUEmbeddingOpsRegistry::Global().GetOpsTypeIds()) // { // ... // } class TPUEmbeddingOpsRegistry { public: // Add the op to the registry. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 05:42:28 UTC 2022 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
static void RegisterMlirInputDialects(mlir::DialectRegistry& registry) { // TODO(b/259459405): Remove support for stablehlo as an input. registry .insert<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::TF::TensorFlowDialect, mlir::stablehlo::StablehloDialect, mlir::quant::QuantizationDialect>(); mlir::func::registerAllExtensions(registry); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/mlir_tflite_runner.cc
return 1; } // Load the MLIR module. mlir::DialectRegistry registry; registry.insert<mlir::TF::TensorFlowDialect, mlir::TFL::TensorFlowLiteDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect>(); mlir::MLIRContext context(registry); llvm::SourceMgr source_mgr; source_mgr.AddNewSourceBuffer(std::move(*file_or_err), llvm::SMLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tac_translate.cc
} options.enable_inliner = true; options.legalize_to_tflite_ops = true; mlir::TFL::tac::TacModule tac_module(options); mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); tac_module.RegisterExtraDialects(registry); tac_module.SetImporter(CreateTfLiteImporter()); tac_module.SetExporter(CreateTfLiteExporter(options.hardware_backends)); return tac_module.Run(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 14:25:57 UTC 2022 - 5.9K bytes - Viewed (0) -
tensorflow/c/experimental/ops/gen/BUILD
"//tensorflow/c/experimental/ops/gen/cpp", "//tensorflow/c/experimental/ops/gen/model", "//tensorflow/core:lib", # Without this line to link in ops, the global registry will be empty: "//tensorflow/core:ops", ], ) tf_cc_binary( name = "generate_cpp", args = ["--stderrthreshold=1"], # Warnings and errors to stderr. deps = [":generate_cpp_main"],
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 17 15:20:54 UTC 2022 - 864 bytes - Viewed (0)