- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 52 for registry_ (0.19 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/tools/stablehlo_quant_opt.cc
mlir::tf_executor::TensorFlowExecutorDialect, mlir::vhlo::VhloDialect>(); mlir::mhlo::registerAllMhloDialects(registry); mlir::func::registerAllExtensions(registry); return failed( mlir::MlirOptMain(argc, argv, "StableHLO quant Pass Driver\n", registry));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 07:37:34 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/legalize_tf_mlir.cc
"or compilation failure from the MLIR bridge (full fallback mode)."; mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::stablehlo::registerAllDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 20:29:34 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
} private: void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { mlir::mhlo::registerAllMhloDialects(registry); mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::func::FuncDialect, mlir::arith::ArithDialect>(); registry.insert<shape::ShapeDialect>(); } public: StringRef getArgument() const final { return "tf-mhlo"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
LegalizeTFToQuant(const LegalizeTFToQuant &) {} /// Performs the lowering to Quant ops dialect. void runOnOperation() override; void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); } StringRef getArgument() const final { // This is the argument used to refer to the pass in
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/python/converter_python_api_wrapper.cc
tflite::RegisterCustomOpdefs(custom_opdefs_txt_raw.ptr())); }, py::arg("custom_opdefs_txt_raw"), R"pbdoc( Registers the given custom opdefs to the TensorFlow global op registry. )pbdoc"); m.def( "RetrieveCollectedErrors", []() { std::vector<std::string> collected_errors = tflite::RetrieveCollectedErrors();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 18:18:30 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
"allowlist) as select Tensorflow ops"), llvm::cl::init(false)); // The following approach allows injecting opdefs in addition // to those that are already part of the global TF registry to be linked in // prior to importing the graph. The primary goal is for support of custom ops. // This is not intended to be a general solution for custom ops for the future
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/tftext_utils_test.cc
void Register(const std::string& op_name, OpRegistry* registry) { registry->Register([op_name](OpRegistrationData* op_reg_data) -> Status { op_reg_data->op_def.set_name(op_name); return absl::OkStatus(); }); } } // namespace TEST(TfTextUtilsTest, TestTfTextRegistered) { std::unique_ptr<OpRegistry> registry(new OpRegistry); Register("WhitespaceTokenizeWithOffsets", registry.get());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 29 02:14:37 UTC 2024 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
MlirToFlatBufferFileTranslateFunction, [](DialectRegistry& registry) { registry.insert<quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); mlir::RegisterAllTensorFlowDialects(registry); registry.insert<TFL::TensorFlowLiteDialect>(); registry.insert<arith::ArithDialect>(); registry.insert<func::FuncDialect>(); registry.insert<mlir::vhlo::VhloDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/python/tfr_wrapper.cc
PYBIND11_MODULE(tfr_wrapper, m) { m.def("verify", [](std::string input) { mlir::DialectRegistry registry; registry.insert<mlir::arith::ArithDialect, mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::TFR::TFRDialect>(); mlir::MLIRContext ctx(registry); ctx.loadAllAvailableDialects(); llvm::SourceMgr source_mgr = llvm::SourceMgr();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 13 06:54:12 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_mlir_util_test.cc
mlir::DialectRegistry registry; mlir::registerAllDialects(registry); mlir::RegisterAllTensorFlowDialects(registry); mlir::registerAllPasses(); mlir::registerTensorFlowPasses(); EXPECT_TRUE(mlir::MlirOptMain(output_stream->os(), std::move(input_file), registry, mlir::MlirOptMainConfig{}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 18 13:40:21 UTC 2024 - 5.8K bytes - Viewed (0)