- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 24 for TensorFlowLiteDialect (0.55 sec)
-
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) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_translate.cc
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>(); registry.insert<mlir::stablehlo::StablehloDialect>(); });
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/lite/utils/lstm_utils_test.cc
context_->loadDialect<arith::ArithDialect, mlir::func::FuncDialect, tensor::TensorDialect, mlir::TF::TensorFlowDialect, TensorFlowLiteDialect>(); builder_ = std::make_unique<mlir::Builder>(context_.get()); fused_lstm_func_ = createLstmCompositeFunc(builder_.get(), false, false); fused_lstm_func_cifg_ =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/passes.h
} namespace quantfork { class QuantizationForkDialect; } namespace mhlo { class MhloDialect; } namespace TF { class TensorFlowDialect; } namespace TFL { class TFLDialect; typedef TFLDialect TensorFlowLiteDialect; } // namespace TFL namespace func { class FuncOp; } class ModuleOp; template <typename T> class OperationPass; class Type; namespace TFL {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 21:29:34 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
registry.insert<mlir::BuiltinDialect>(); registry.insert<mlir::arith::ArithDialect>(); registry.insert<mlir::func::FuncDialect>(); registry.insert<mlir::TFL::TensorFlowLiteDialect>(); return registry; }()) { context_.loadAllAvailableDialects(); mlir::OpBuilder builder(&context_); module_ = builder.create<mlir::ModuleOp>(builder.getUnknownLoc());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ReduceWhileOperandsPass) void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TFL::TensorFlowLiteDialect, TF::TensorFlowDialect>(); } void runOnOperation() override; }; LogicalResult FindImplicityProducers( const std::vector<uint64_t> &explicitly_consumed_ids, std::vector<bool> &is_consumed_id,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
return "Fallback TF-Quantization-unsupported ops to TFLite Flex ops."; } void getDependentDialects(DialectRegistry ®istry) const override { registry.insert<TFL::TensorFlowLiteDialect>(); } private: // The mode of TF Quantization, might indicate different users/devices. Option<std::string> mode_{*this, "mode",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc
: public impl::PrepareCompositeFunctionsPassBase< PrepareCompositeFunctionsPass> { void getDependentDialects(DialectRegistry& registry) const override { registry.insert<TFL::TensorFlowLiteDialect>(); } public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(PrepareCompositeFunctionsPass) explicit PrepareCompositeFunctionsPass() {} private:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.6K bytes - Viewed (0)