- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 29 for getDependentDialects (0.33 sec)
-
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(
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/tensorflow/transforms/xla_call_module_deserialization.cc
class XlaCallModuleDeserializationPass : public impl::XlaCallModuleDeserializationPassBase< XlaCallModuleDeserializationPass> { public: void getDependentDialects(mlir::DialectRegistry ®istry) const override { XlaCallModuleDeserializationPassBase::getDependentDialects(registry); mlir::func::registerAllExtensions(registry); } void runOnOperation() override { ModuleOp module = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_custom_aggregation_op_to_quant_stats.cc
} StringRef getDescription() const final { // 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>(); }
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/stablehlo/transforms/tf_stablehlo_pass.cc
skip_resize_ = pass.skip_resize_; skip_partitioned_calls_ = pass.skip_partitioned_calls_; } 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>();
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/quantization/tensorflow/passes/lift_quantizable_spots_as_functions_drq.cc
// This is a brief description of the pass. return "Replace quantization candidates with composite functions into the " "module for post-training dynamic range case"; } void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect>(); } void runOnOperation() override; private: Option<OpSet> target_opset_{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_globals_to_ml_program.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.h.inc" class LowerGlobalsToMlProgram : public impl::LowerGlobalsToMlProgramPassBase<LowerGlobalsToMlProgram> { public: void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<mlir::tf_saved_model::TensorFlowSavedModelDialect, ml_program::MLProgramDialect>(); } void runOnOperation() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
explicit LegalizeTFToQuant() = default; 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 {
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/tensorflow/transforms/tf_graph_optimization_pass.cc
// belong to 'tf' or '_tf' dialect. The output is in '_tf' dialect. class GraphOptPass : public mlir::PassWrapper<GraphOptPass, mlir::OperationPass<mlir::ModuleOp>> { void getDependentDialects(mlir::DialectRegistry& registry) const override { mlir::RegisterAllTensorFlowDialects(registry); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(GraphOptPass)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
// Applies prepare quantization on the model in TF dialect for dynamic range // quantization case. class PrepareQuantizeDRQPass : public PassWrapper<PrepareQuantizeDRQPass, OperationPass<ModuleOp>> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TF::TensorFlowDialect, ::mlir::quant::QuantizationDialect, ::mlir::quantfork::QuantizationForkDialect>(); } public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
StringRef getArgument() const final { return "constant-fold-broadcast-pass"; } StringRef getDescription() const final { return "Constant folds BroadcastInDimOp before binary elementwise ops"; } void getDependentDialects(::mlir::DialectRegistry ®istry) const override {} void runOnOperation() override { RewritePatternSet patterns(&getContext()); patterns.add<FoldBroadcastInDimBeforeMulOp>(&getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0)