- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 70 for getRegions (0.69 sec)
-
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) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
cluster_result_types.size() * num_cores_per_replica); for (mlir::Region& region : old_parallel_execute.getRegions()) { if (!llvm::isa<tf_device::ClusterFuncOp>(region.front().front())) { for (Type t : region.front().front().getResultTypes()) concatenated_output_types.emplace_back(t); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc
} } else if (isa<tf_device::LaunchOp, tf_device::ClusterOp, tf_executor::IslandOp, tf_executor::GraphOp, IfRegionOp, CaseRegionOp>(op)) { for (Region& region : op->getRegions()) { AddRegionSideEffectsForOp(region, op); } } else if (auto xla_call_module_op = dyn_cast<XlaCallModuleOp>(op)) { for (auto func_symbol : xla_call_module_op.getFunctionList().getAsRange<
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 41.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
// _TPUDeviceOrdinalPlaceholder implicitly depends on the replica. if (llvm::isa<TF::_TPUDeviceOrdinalPlaceholderOp>(op)) return false; bool has_replicate_operands = false; visitUsedValuesDefinedAbove(op->getRegions(), [&](OpOperand* operand) { if (!ancestor_of_replicate(operand->get().getParentRegion())) has_replicate_operands = true; }); return !has_replicate_operands; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0)