- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 36 for registry_ (0.12 sec)
-
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/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/tf_mlir_opt_main.cc
tensorflow::tfrt_compiler:: RegisterNonTPULowerClusterToRuntimeOpsPassPipeline(); mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); return failed( mlir::MlirOptMain(argc, argv, "TensorFlow pass driver\n", registry));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate_registration.cc
} static TranslateFromMLIRRegistration mlir_to_graph_translate( /*name=*/"mlir-to-graph", /*description=*/"convert mlir to graph", MlirToGraphTranslateFunction, [](DialectRegistry& registry) { mlir::RegisterAllTensorFlowDialects(registry); }); static LogicalResult MlirToGraphdefTranslateFunction( ModuleOp module, llvm::raw_ostream& output) { if (!module) return failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/execution_metadata_exporter_test.cc
func.return %3 : tensor<2x1xf32> })"; const std::string kExpectedFB = CreateRuntimeMetadata(); mlir::DialectRegistry registry; registry.insert<mlir::TFL::TensorFlowLiteDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect>(); mlir::MLIRContext context(registry); auto module = mlir::OwningOpRef<mlir::ModuleOp>( mlir::parseSourceString<mlir::ModuleOp>(kMLIR, &context));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 06:11:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/register_common_dialects.cc
namespace mlir { void RegisterCommonToolingDialects(mlir::DialectRegistry& registry) { mlir::RegisterAllTensorFlowDialects(registry); mlir::mhlo::registerAllMhloDialects(registry); mlir::registerAllDialects(registry); mlir::registerAllExtensions(registry); mlir::stablehlo::registerAllDialects(registry); registry.insert<mlir::TFL::TensorFlowLiteDialect>(); registry.insert<mlir::kernel_gen::tf_framework::TFFrameworkDialect>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 22:45:35 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/runtime_fallback/runtime_fallback_executor.cc
} void RuntimeFallbackExecutor::Prepare(llvm::StringRef mlir_input) { // We only support IR written in the Tensorflow dialect. mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); llvm::SourceMgr source_mgr; source_mgr.AddNewSourceBuffer( llvm::MemoryBuffer::getMemBuffer(mlir_input, "test_ir"), llvm::SMLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/backend_compiler.h
#include "tensorflow/core/tfrt/runtime/runtime.h" namespace tensorflow { class BackendCompiler { public: virtual ~BackendCompiler(); virtual void GetDependentDialects(mlir::DialectRegistry& registry) const {} // Compile the `module` in TF dialect. The result module should be also in TF // dialect. virtual absl::Status CompileTensorflow( tfrt_stub::ModelRuntimeContext& model_context,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 10 10:31:36 UTC 2024 - 1.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_converter/odml_converter_main.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 19:05:30 UTC 2024 - 2.2K bytes - Viewed (0)