- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 58 for getRegions (0.72 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc
// Returns true if the op contains a `CalibrationStatisticsSaverOp`. bool ContainCalibrationStatisticsSaverOp(Operation* op) { // Check the region for CaseRegionOp, IfRegionOp and WhileRegionOp. for (Region& region : op->getRegions()) { if (!region.getOps<TF::CalibrationStatisticsSaverOp>().empty()) { return true; } } SymbolTable symbol_table(op->getParentOfType<ModuleOp>());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
op->getAttrs(), op->getPropertiesStorage(), op->getSuccessors(), op->getNumRegions()); builder.insert(new_op); // Move region bodies to the new operation. for (auto it : llvm::zip(op->getRegions(), new_op->getRegions())) { Region &old_region = std::get<0>(it); Region &new_region = std::get<1>(it); new_region.takeBody(old_region); } // Replace used results and erase the old op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/move_tpu_compile_to_front.cc
// Followed by everything else. return 0; } void MoveTpuCompileToFrontPass::runOnOperation() { MarkCompilationOps(getOperation()); getOperation().walk([](Operation* op) { for (Region& region : op->getRegions()) { for (Block& block : region.getBlocks()) { if (block.empty()) continue; auto ops = SortBlockTopologically(block, OutsideCompilationOrdering); // Replace the block with the reordered block.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 00:26:18 UTC 2023 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/tf_dataflow.h
if (auto cast = dyn_cast<TF::CastOp>(op)) { this->join(results[0], *operands[0]); } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) { for (auto ®ion : while_op->getRegions()) { for (auto [arg, value] : llvm::zip(region.getArguments(), while_op->getOperands())) { this->join(getLatticeElement(arg), *getLatticeElement(value)); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/raise_target_subgraphs.cc
added_call_op->setAttr(kInterfaceNameAttr, interface_name); StringAttr device = mlir::cast<StringAttr>( added_func_op->getRegion(0).getBlocks().front().front().getAttr(kDevice)); StringAttr inference_type = mlir::cast<StringAttr>( added_func_op->getRegion(0).getBlocks().front().front().getAttr( kInferenceType)); added_call_op->setAttr(kDevice, device);
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/tensorflow/transforms/region_control_flow_to_functional.cc
llvm::SmallVector<Value, 4> extern_values = CollectExternValues(case_region.getRegions()); // These external values need to be added as inputs to the generated Case. The // order is determined by the order of these values the `extern_vales`. for (const auto& item : llvm::enumerate(case_region.getRegions())) { std::string branch_name =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
op_->getAttrs(), op_->getPropertiesStorage(), op_->getSuccessors(), op_->getNumRegions()); builder.insert(new_op); // Move regions to the new op. for (auto it : llvm::zip(op_->getRegions(), new_op->getRegions())) { Region& old_region = std::get<0>(it); Region& new_region = std::get<1>(it); new_region.takeBody(old_region); } // Insert stores to all written resources.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
PropagatePotentiallyWrittenUpFromCallee(func.getRegion(), call.getArgOperands()); } return; } if (auto if_op = dyn_cast<TF::IfOp>(op)) { for (auto callee : {if_op.then_function(), if_op.else_function()}) { PropagatePotentiallyWrittenUpFromCallee(callee.getRegion(), if_op.getInput());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst.cc
if (absl::StartsWith(op_name, "tf.") || absl::StartsWith(op_name, "tfl.")) { loc_to_name_.emplace(op->getLoc(), op_name); } }; for (auto ®ion : module->getRegions()) { region.walk(collectOps); } pass_name_ = extract_pass_name(pass->getName().str()); error_collector_->Clear(); } void ErrorCollectorInstrumentation::runAfterPass(Pass *pass,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/rewrite_tpu_embedding_ops.cc
void RewriteTPUEmbeddingOps::runOnOperation() { func::FuncOp func = getOperation(); if (failed(RunOnRegion(&func.getBody()))) return signalPassFailure(); func.walk([&](Operation* op) { for (Region& region : op->getRegions()) { if (failed(RunOnRegion(®ion))) return signalPassFailure(); } }); } } // anonymous namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateRewriteTPUEmbeddingOpsPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 22:55:42 UTC 2024 - 4.5K bytes - Viewed (0)