- Sort Score
- Result 10 results
- Languages All
Results 131 - 140 of 293 for runOnOperation (0.35 sec)
-
tensorflow/compiler/mlir/lite/experimental/tac/transforms/fold_constants_to_subgraph.cc
this->fold_all_constants_flag_ = other.fold_all_constants_flag_; } explicit FoldConstantsToSubgraphPass(bool fold_all_constants) { fold_all_constants_flag_ = fold_all_constants; } private: void runOnOperation() override; Option<bool> fold_all_constants_flag_{ *this, "fold-all-constants", llvm::cl::desc("Whether to fold all constants or just i32."), llvm::cl::init(false)}; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeTFToQuant) 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>(); }
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/experimental/tac/transforms/tac_filter.cc
} Option<bool> use_test_setting_{ *this, "use-test-setting", llvm::cl::desc( "Whether to use the test config for the tac filter protobuf."), llvm::cl::init(false)}; void runOnOperation() override; }; void ApplyFunctionTacFilter(func::FuncOp func, FunctionFilter::FunctionFilterType type, OpBuilder& builder) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.h.inc" struct DecomposeResourceOpsPass : public impl::DecomposeResourceOpsPassBase<DecomposeResourceOpsPass> { void runOnOperation() override { // Add lowering patterns to the list. RewritePatternSet patterns(&getContext()); TF::PopulateDecomposeResourceOpsPatterns(&getContext(), &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fold_broadcast.cc
#define GEN_PASS_DEF_BROADCASTFOLDPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class BroadcastFoldPass : public impl::BroadcastFoldPassBase<BroadcastFoldPass> { public: void runOnOperation() override; }; LogicalResult ConvertResultsBroadcastableShapeOp::matchAndRewrite( Operation* op, PatternRewriter& rewriter) const { if (op->hasTrait<OpTrait::ResultsBroadcastableShape>())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
#define GEN_PASS_DEF_GROUPBYDIALECTPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class GroupByDialectPass : public impl::GroupByDialectPassBase<GroupByDialectPass> { public: void runOnOperation() override; private: void processFunction(mlir::func::FuncOp func, int& counter, llvm::SmallDenseSet<StringRef>& dialects, Operation* module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
return "quant-propagate-quantize-type"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Propagate quantized type through allowed ops."; } void runOnOperation() override; }; // Propagate dequantize op if the next op supports the data type. // Given the below graph, // op_before_dequantize -> dequantize_op -> user_op -> rest_op
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_quantized.cc
#define GEN_PASS_DEF_LOWERQUANTIZEDPASS #include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" class LowerQuantizedPass : public impl::LowerQuantizedPassBase<LowerQuantizedPass> { public: void runOnOperation() override { RewritePatternSet patterns(&getContext()); mlir::TF::PopulateLoweringQuantizedPatterns(&getContext(), &patterns); (void)applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)); } };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_tf_test_pass.cc
// Lowers some of the TensorFlow operations that can be represented using other // TensorFlow operations. struct LowerTF : public impl::TestTensorFlowLowerTFPassBase<LowerTF> { void runOnOperation() override { // Add lowering patterns to the list. RewritePatternSet patterns(&getContext()); if (default_patterns_) { mlir::TF::PopulateLoweringTFPatterns(&getContext(), &patterns); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
: public QuantizeVariablesPassBase<QuantizeVariablesPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(QuantizeVariablesPass) explicit QuantizeVariablesPass() = default; void runOnOperation() override; private: // Outlines the regions of the WhileOp's cond and body and insert function // calls instead. void QuantizeVariable(OpBuilder &builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0)