- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for TensorFlowLiteDialect (0.42 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.td
} def LegalizeHloToTfLitePass : Pass<"tfl-legalize-hlo", "mlir::ModuleOp"> { let summary = "Legalize from MHLO to the TFLite dialect"; let dependentDialects = ["TFL::TensorFlowLiteDialect"]; let constructor = "mlir::odml::CreateLegalizeHloToTfLitePass()"; } def LegalizeStablehloToVhloPass : Pass<"stablehlo-legalize-vhlo", "ModuleOp"> { let summary = "Legalize StableHLO to VHLO for ODML.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.h
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h" #define GET_ATTRDEF_CLASSES #include "tensorflow/compiler/mlir/lite/ir/tfl_ops_attrdefs.h.inc" namespace mlir { namespace TFL { typedef TFLDialect TensorFlowLiteDialect; // The Control type is a token-like value that models control dependencies class ControlType : public Type::TypeBase<ControlType, Type, TypeStorage> { public: using Base::Base;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/register_common_dialects.cc
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>(); registry.insert<mlir::quant::QuantizationDialect>(); registry.insert<mlir::quantfork::QuantizationForkDialect>();
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/lite/stablehlo/transforms/composite_lowering_pass.cc
void CompositeLoweringPass::runOnOperation() { RewritePatternSet patterns(&getContext()); populateWithGenerated(patterns); ConversionTarget target(getContext()); target.addLegalDialect<TFL::TensorFlowLiteDialect>(); target.addLegalDialect<arith::ArithDialect>(); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 23:16:05 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/test_base.h
builder_(ctx_.get()) { ctx_->loadDialect< arith::ArithDialect, mlir::stablehlo::StablehloDialect, func::FuncDialect, TF::TensorFlowDialect, TFL::TensorFlowLiteDialect, tf_saved_model::TensorFlowSavedModelDialect, tf_executor::TensorFlowExecutorDialect, quant::QuantizationDialect, quantfork::QuantizationForkDialect>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 3.5K 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>(
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/lite/experimental/tac/utils/utils.cc
return absl::InvalidArgumentError(absl::StrFormat( "Cannot open input file: %s. %s", input_filename, error)); } if (input_mlir) { mlir::DialectRegistry registry; registry.insert<mlir::TFL::TensorFlowLiteDialect, mlir::arith::ArithDialect, mlir::func::FuncDialect>(); mlir::func::registerAllExtensions(registry); context->appendDialectRegistry(registry);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 03:47:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
void SetUp() override { context_ = std::make_unique<mlir::MLIRContext>(); context_->loadDialect<mlir::arith::ArithDialect, mlir::func::FuncDialect, mlir::TF::TensorFlowDialect, TensorFlowLiteDialect>(); builder_ = std::make_unique<mlir::Builder>(context_.get()); fused_max_unpooling_func_ = createMaxUnpoolingFunc(builder_.get(), {2, 4, 4, 2}, {2, 2, 2, 2});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tflite_legalize_hlo.cc
RewritePatternSet patterns(&getContext()); // Add new conversion patterns here. PopulateLegalizeHloToTFLitePatterns(&patterns, &context); ConversionTarget target(context); target.addLegalDialect<TFL::TensorFlowLiteDialect, mhlo::MhloDialect>(); target.addLegalOp<func::CallOp, func::ConstantOp, arith::ConstantOp>(); target.addDynamicallyLegalOp<mhlo::CustomCallOp>(IsCustomCallLegal);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/mlir_tflite_runner.cc
<< "': " << error.message() << "\n"; 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;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 03 00:14:05 UTC 2023 - 6.3K bytes - Viewed (0)