- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 51 for setRegion (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_executor_ops.td
}]; let results = (outs Variadic<AnyType>:$results ); let regions = (region SizedRegion<1>:$body); let extraClassDeclaration = [{ Block &GetBody() { return getOperation()->getRegion(0).front(); } FetchOp GetFetch(); }]; let hasCanonicalizer = 1; } def TfExecutor_FetchOp : TfExecutor_Op<"fetch", [Terminator, ControlOperandsAfterAllData, HasParent<"GraphOp">]> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 23 19:35:12 UTC 2023 - 22K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
state.addRegion(); } Operation* new_op = builder.create(state); for (const auto& indexed_regions : llvm::enumerate(op->getRegions())) { Region& region = op->getRegion(indexed_regions.index()); IRMapping mapping; indexed_regions.value().cloneInto(®ion, mapping); } int new_position = 0; for (auto result : op->getResults()) {
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/quantization/stablehlo/passes/quantization_patterns.h
if (candidate_op->getNumRegions() != 0) { for (const auto& indexed_regions : llvm::enumerate(candidate_op->getRegions())) { Region& target_region = quantized_op->getRegion(indexed_regions.index()); IRMapping mapping; indexed_regions.value().cloneInto(&target_region, mapping); } } for (auto output : outputs_replaced) {
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/analysis/resource_alias_analysis.cc
if (!passthrough_arg) break; value = call.getArgOperands()[passthrough_arg.value()]; } else if (isa<tf_device::LaunchOp, tf_device::ClusterOp>(op)) { value = op->getRegion(0).front().getTerminator()->getOperand(res_index); } else { break; } } return value; } // Analyze the region. BacktrackAnalysisInfo::BacktrackAnalysisInfo(
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.cc
} entry_func_op.setFunctionType( rewriter.getFunctionType(arg_types, output_types)); // Replace argument types and locs. Block& entry = entry_func_op->getRegion(0).front(); for (auto [arg, arg_type, arg_loc] : llvm::zip_equal(entry.getArguments(), arg_types, arg_locs)) { arg.setType(arg_type); arg.setLoc(arg_loc); } }
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/utils/tf_xla_mlir_translate.cc
// have the proper shapes. TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); mlir::func::FuncOp main = module_op.lookupSymbol<mlir::func::FuncOp>("main"); mlir::Block& block = main.getRegion().front(); xla::XlaBuilder builder("main"); // Create xla_params. std::vector<xla::XlaOp> xla_params; for (mlir::BlockArgument& arg : block.getArguments()) { auto num = arg.getArgNumber();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_executor.cc
<< ")"; } } return success(); } void GraphOp::print(OpAsmPrinter &p) { p << ' '; p.printRegion(getOperation()->getRegion(0)); p.printOptionalAttrDict(getOperation()->getAttrs()); } ParseResult GraphOp::parse(OpAsmParser &parser, OperationState &result) { llvm::SMLoc loc = parser.getCurrentLocation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
SmallVectorImpl<OpVisitorState>& ops_to_visit, const llvm::SmallPtrSetImpl<Block*>& control_flow_blocks, Value token) { ops_to_visit.push_back({region_idx + 1, token, region_op}); Region& region = region_op->getRegion(region_idx); assert(llvm::hasSingleElement(region)); auto block_token = UpdateControlFlowBlockArgWithToken(builder, region.front(), block_arg_types);
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
for (OpResult result : op->getResults()) { if (!IsResource(result)) continue; has_resource_result = true; int result_idx = result.getResultNumber(); Value ret0 = op->getRegion(0).front().getTerminator()->getOperand(result_idx); for (Region ®ion : op->getRegions().drop_front()) { Value ret = region.front().getTerminator()->getOperand(result_idx); if (ret != ret0) {
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/quantization/tensorflow/passes/quantize.cc
for (const auto& indexed_regions : llvm::enumerate(quantizing_op->getRegions())) { IRMapping mapping; indexed_regions.value().cloneInto( &quantized_op->getRegion(indexed_regions.index()), mapping); } } for (const auto& output_index_pair : outputs_replaced) { Value output = output_index_pair.getFirst();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 23.6K bytes - Viewed (0)