- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 17 for getNumRegions (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc
<< "Cannot find function " << sym.getRootReference(); } callees.push_back(callee); } return success(); } bool HasSingleBlock(func::FuncOp func) { return func->getNumRegions() == 1 && func.getBody().hasOneBlock(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
output_types, candidate_op->getAttrs()); for (int i = 0; i < candidate_op->getNumRegions(); ++i) { new_state.addRegion(); } Operation* quantized_op = rewriter.create(new_state); if (candidate_op->getNumRegions() != 0) { for (const auto& indexed_regions : llvm::enumerate(candidate_op->getRegions())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
// This is a lengthy bit of code, since it has to recreate the operation. // TODO(kramm): Move this under utils/ somewhere. void EraseResults(Operation* op, llvm::BitVector erase) { assert(!op->getNumRegions()); std::vector<Type> result_types; for (auto result : op->getResults()) { if (!erase[result.getResultNumber()]) { result_types.push_back(result.getType()); } } OpBuilder builder(op);
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/constant_fold_utils.cc
// could be folded should have a custom folder instead of relying on the // TensorFlow folding hook. if (inst == nullptr || inst->getNumResults() == 0 || inst->hasTrait<OpTrait::TF::NoConstantFold>() || inst->getNumRegions() != 0 || !isMemoryEffectFree(inst)) { return false; } // If any of the result types are variants, don't try to constant fold them. // This creates opaque variant constants which lose information and would
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc
} 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(); } // Gets the GraphOp from the function op. Returns an empty op iff it doesn't // exist. GraphOp GetGraphOpFromFuncOp(func::FuncOp func_op) {
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/analysis/resource_alias_analysis.cc
} } } else if (isa<tf_device::LaunchOp, tf_device::ClusterOp, tf_executor::IslandOp, tf_executor::GraphOp>(op) && op->getNumRegions() == 1) { Region& region = op->getRegion(0); const auto& body_info = backtrack_analysis.GetAnalysisForRegion(region); for (auto result : filter_resources(op->getResults())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
} num_new_results_ = op_->getNumResults(); for (auto resource : all_resources) { ResourceInfo info; info.data_type = GetResourceSubtype(resource); llvm::BitVector written_regions(op_->getNumRegions()); bool unsupported_use = false; for (OpOperand& use : resource.getUses()) { Operation* user = use.getOwner(); // If the user is not in one of the regions, we are not interested in it.
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/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
tf_saved_model::TensorFlowSavedModelDialect>(); } }; // 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(); } // Gets the GraphOp from the function op. Returns an empty op iff it doesn't // exist. GraphOp GetGraphOpFromFuncOp(func::FuncOp func_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
builder.setInsertionPoint(region_if); if (!region_idx) { RewriteRegionIfOp(builder, region_if, ops_to_visit, token); return true; } if (*region_idx < region_if.getNumRegions()) { // For the region-blocks of If op, we create a dummy token argument. Later // we replace that block-argument's uses with the same (implicitly captured) // token 'token', used for If op, and erase the argument.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting_cleanup.cc
OpBuilder builder(op); Operation *new_op = Operation::create( op->getLoc(), op->getName(), new_result_types, op->getOperands(), 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);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.7K bytes - Viewed (0)