- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 86 for GetOperation (0.55 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc
: public impl::LegalizeVariablesPassBase<LegalizeVariablesPass> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeVariablesPass) void runOnOperation() override { auto module = getOperation(); // If TFLite variable legalization is not allowed, then we skip this pass. if (auto legalize_tfl_variables_attr = module->getAttr(kLegalizeTflVariables)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
explicit ConvertFuncToBfloat16Pass() = default; private: void runOnOperation() override; }; void ConvertFuncToBfloat16Pass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext* context = func_op.getContext(); RewritePatternSet patterns(context); BFloat16TypeConverter converter; patterns.add<BFloat16TypePattern, BitcastConvertOpPattern>(converter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass.cc
public: void runOnOperation() override; }; void InputMetricsLoweringPass::runOnOperation() { bool has_dynamic_op = false; Operation* func_op = getOperation(); func_op->walk([&](Operation* op) { auto abstractOp = op->getRegisteredInfo(); if (!abstractOp) return WalkResult::advance(); if (mlir::mhlo::IsDynamicPadderOp(abstractOp->getTypeID())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 08:55:35 UTC 2024 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
auto fail = [&](Operation* op, std::string message) { op->emitError(message); signalPassFailure(); }; DenseMap<StringRef, func::FuncOp> entrypoints; auto module = getOperation(); module.walk([&](func::FuncOp op) { auto visibility = SymbolTable::getSymbolVisibility(op); if (visibility != SymbolTable::Visibility::Public) return; entrypoints[op.getSymName()] = op; });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_rewrite.cc
cluster_func_op.replaceAllUsesWith(xla_launch_op.getResults()); cluster_func_op.erase(); } void XlaRewritePass::runOnOperation() { ModuleOp module = getOperation(); SymbolTable symtab(module); OpBuilder builder(&getContext()); module.walk([&](tf_device::ClusterFuncOp cluster_func_op) { RewriteCall(cluster_func_op, symtab, builder); }); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_composite_functions.cc
// TODO: b/321729008 - move this implementation to quantization_patterns.cc. if (merge_fusion_with_dequantize_) { pm.addPass(createMergeFusionWithDequantizePass()); } ModuleOp module_op = getOperation(); if (const absl::Status pm_run_status = RunPassesOnModuleOp(mlir_dump_file_name_, pm, module_op); !pm_run_status.ok()) { signalPassFailure(); } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 02:59:01 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
quant::AccumulatorScaleFunc func); quant::QuantParams default_quant_params_; }; } // namespace void DefaultQuantParamsPass::runOnOperation() { func::FuncOp func = getOperation(); OpBuilder builder(func); std::vector<Value> activation_values; std::vector<Value> bias_values; // First of all, collect all the values (block arguments and op results) which
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_validate_inputs.cc
"marker."; return failure(); } } return success(); } void XlaValidateInputsPass::runOnOperation() { ModuleOp module = getOperation(); SymbolTable symtab(module); llvm::SmallVector<func::FuncOp> entry_funcs = GetEntryFunctions(module); if (entry_funcs.empty()) { LOG(WARNING) << "missing entry functions"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 19:29:14 UTC 2024 - 3.8K 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. module.walk([&](mhlo::ConstantOp const_op) {
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/quantization/stablehlo/passes/testing/test_post_calibration_component.cc
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestPostCalibrationComponentPass) private: void runOnOperation() override; }; void TestPostCalibrationComponentPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext& ctx = getContext(); OpPassManager pm(ModuleOp::getOperationName()); QuantizationConfig config = QuantizationConfig::default_instance(); config.mutable_static_range_ptq_preset();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 12:53:33 UTC 2024 - 3.5K bytes - Viewed (0)