- Sort Score
- Result 10 results
- Languages All
Results 191 - 200 of 293 for runOnOperation (0.21 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc
public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(OptimizeBatchMatmulPass) OptimizeBatchMatmulPass() = default; OptimizeBatchMatmulPass(const OptimizeBatchMatmulPass &) {} void runOnOperation() override; }; // Converts batch_matmul operation to fully_connected if rhs is a // constant tensor with rank 2 struct ConvertBatchMatMulOp2FullyConnectedOp : public OpRewritePattern<TFL::BatchMatMulOp> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/unfold_splat_constant_pass.cc
// pass and folders are not applied by default. class UnfoldSplatConstantPass : public impl::UnfoldSplatConstantPassBase<UnfoldSplatConstantPass> { public: void runOnOperation() override { auto module = getOperation(); mlir::OpBuilder op_builder(&module.getBodyRegion()); // Cannot use the pattern rewriter because the driver applies folders by // default.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/unroll_batch_matmul.cc
// the end. struct UnrollBatchMatMulPass : public impl::UnrollBatchMatMulPassBase<UnrollBatchMatMulPass> { void runOnOperation() override; }; void UnrollBatchMatMulPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); PopulateUnrollTfBatchMatMul(&getContext(), patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/quantize.cc
enable_weight_only_quantization_ = quant_specs.weight_only_quantization; } void runOnOperation() override; private: quant::QuantizationSpecs quant_specs; }; #include "tensorflow/compiler/mlir/lite/transforms/generated_quantize.inc" void QuantizePass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); auto* ctx = func.getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
} StringRef getDescription() const override { return "Merge the save function's ops to the main function. The save " "function will be removed after the pass."; } void runOnOperation() override; }; // Returns true iff func_op has either no Region or the body has no Blocks. bool IsFuncOpEmpty(func::FuncOp func_op) { return func_op->getNumRegions() == 0 || func_op.getBody().empty(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc" struct TPUDevicePropagation : public impl::TPUDevicePropagationPassBase<TPUDevicePropagation> { void runOnOperation() override; }; void TPUDevicePropagation::runOnOperation() { func::FuncOp func = getOperation(); if (!IsSupportedGraph(func)) return; llvm::DenseMap<Value, llvm::StringRef> value_to_device;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/preprocess_op.cc
return "quant-preprocess-op"; } StringRef getDescription() const final { // This is a brief description of the pass. return "Preprocess TF op prior to quantization"; } void runOnOperation() override; private: Option<OpSet> op_set_{ *this, "target-opset", llvm::cl::init(OpSet::UNIFORM_QUANTIZED), llvm::cl::desc("Choose target opset."), llvm::cl::values(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc
if (tf2xla_fallback_device_type.has_value()) { device_type_ = tf2xla_fallback_device_type.value().str(); } } /// Performs the lowering to XLA dialect. void runOnOperation() override; }; #define GEN_PASS_DEF_LEGALIZETFMODULEPASS #include "tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_passes.h.inc"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
} // namespace class DeferActivationTransposePass : public impl::DeferActivationTransposePassBase< DeferActivationTransposePass> { private: void runOnOperation() override; }; void DeferActivationTransposePass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); patterns.add<DeferActivationTransposeForAddOp,
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/quantization/tensorflow/passes/prepare_quantize_drq.cc
// processing the function for dynamic range quantization. Therefore, this // method preprocess the function to remove all stats ops. void removeAllStatsOp(func::FuncOp func); void runOnOperation() override; private: QuantizationSpecs quant_specs_; OpSet op_set_; Option<bool> enable_per_channel_quantization_{ *this, "enable-per-channel-quantization", llvm::cl::init(false),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0)