- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 30 for getRegions (0.25 sec)
-
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/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/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/analysis/resource_alias_analysis.cc
// but we also need to analyze regions attached to other ops. module->walk([this](Operation* op) { if (op->hasTrait<OpTrait::NoTerminator>()) return; for (Region& region : op->getRegions()) GetOrCreateAnalysis(region); }); } // Backtracks the definition of `value` looking through passthrough ops. // Returns a non-null value and can return `value` if backtracking is not // possible.
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/quantization/stablehlo/passes/quantization_patterns.h
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/quantization/stablehlo/passes/quantization_patterns.cc
new_state.addRegion(); } Operation* quantized_op = rewriter.create(new_state); for (const auto& [index, region] : llvm::enumerate(op_with_region->getRegions())) { Region& target_region = quantized_op->getRegion(index); IRMapping mapping; region.cloneInto(&target_region, mapping); } const Type operand_type = quantized_op->getOperandTypes()[0];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 06:04:36 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
(*buffer_to_size)[new_op.getResult(std::get<0>(entry))] = { new_op.getResult(std::get<1>(entry)), std::get<2>(entry)}; } for (auto pair : llvm::zip(new_op.getRegions(), case_op.getRegions())) { std::get<0>(pair)->takeBody(*std::get<1>(pair)); } case_op.replaceAllUsesWith( new_op.getResults().take_front(case_op.getNumResults())); case_op.erase(); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0)