- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 145 for registry_ (0.17 sec)
-
tensorflow/compiler/mlir/tfr/passes/tfr_opt.cc
#include "tensorflow/compiler/mlir/tfr/ir/tfr_ops.h" int main(int argc, char **argv) { tensorflow::InitMlir y(&argc, &argv); mlir::registerAllPasses(); mlir::DialectRegistry registry; registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect, mlir::quant::QuantizationDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 15 16:15:16 UTC 2023 - 2.1K bytes - Viewed (0) -
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/quantization/tensorflow/passes/tf_quant_opt.cc
int main(int argc, char **argv) { tensorflow::InitMlir y(&argc, &argv); mlir::registerAllPasses(); mlir::registerTensorFlowPasses(); mlir::DialectRegistry registry; registry.insert<mlir::scf::SCFDialect, mlir::TF::TensorFlowDialect, mlir::tf_saved_model::TensorFlowSavedModelDialect, mlir::func::FuncDialect, mlir::shape::ShapeDialect,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 06 18:20:34 UTC 2023 - 2.4K 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/experimental/tac/py_wrapper/tac_wrapper.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(model_file_path)); tac_module.SetExporter( CreateTfLiteExporter(model_output_path, options.hardware_backends));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tpu_embedding_ops_registry.cc
TPUEmbeddingOpsRegistry::GetOpsTypeIds() { return ops_type_ids_; } // static TPUEmbeddingOpsRegistry& TPUEmbeddingOpsRegistry::Global() { static TPUEmbeddingOpsRegistry* registry = new TPUEmbeddingOpsRegistry; return *registry; } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 14 05:42:28 UTC 2022 - 1.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.h
namespace mlir { namespace tf_saved_model { // Create a pass that removes function arguments that map to global tensors. std::unique_ptr<Pass> CreateLowerGlobalsToMlProgramPass(); // Register this pass in the global registry of MLIR. void RegisterLowerGlobalsToMlProgramPass(); } // namespace tf_saved_model } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 14:17:18 UTC 2022 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/order_by_dialect.h
namespace mlir { namespace TF { // Create an instance of a pass that reorders ops so ops of the same dialect are // next to each other. std::unique_ptr<Pass> CreateOrderByDialectPass(); // Register this pass in the global registry of MLIR. void RegisterOrderByDialectPass(); } // namespace TF } // namespace mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 01 14:17:18 UTC 2022 - 1.2K bytes - Viewed (0) -
tensorflow/c/experimental/pluggable_profiler/pluggable_profiler_internal.h
using TFInitProfilerFn = void (*)(TF_ProfilerRegistrationParams* const, TF_Status* const); // Registers plugin's profiler to TensorFlow's profiler registry. Status InitPluginProfiler(TFInitProfilerFn init_fn); } // namespace profiler } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 1.7K bytes - Viewed (0)