- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 20 for setRegion (1.02 sec)
-
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) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
ASSERT_TRUE(module_op); func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); auto if_op = FindOperationOfType<mlir::stablehlo::IfOp>(main_fn); Block& block = if_op->getRegion(0).front(); Operation& add_op = *absl::c_find_if(block, [](Operation& entry) { return dyn_cast_or_null<::mlir::stablehlo::AddOp>(&entry); }); EXPECT_TRUE(IsInStableHloOpRegion(&add_op)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
pilot/pkg/xds/ads.go
// can still help provide some client-side Envoy context when load balancing based on location. proxy.Locality = &core.Locality{ Region: proxy.XdsNode.Locality.GetRegion(), Zone: proxy.XdsNode.Locality.GetZone(), SubZone: proxy.XdsNode.Locality.GetSubZone(), } } // add topology labels to proxy labels proxy.Labels = labelutil.AugmentLabels( proxy.Labels,
Registered: Fri Jun 14 15:00:06 UTC 2024 - Last Modified: Mon Jun 03 08:29:05 UTC 2024 - 23.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
Region* region = state.addRegion(); region->push_back(new Block); } state.addTypes(output_types); } Block& ParallelExecuteOp::GetRegionBlockWithIndex(unsigned index) { return getOperation()->getRegion(index).front(); } Operation::result_range ParallelExecuteOp::GetRegionOutputs( unsigned region_index) { int num_region_results = GetRegionBlockWithIndex(region_index).getTerminator()->getNumOperands();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
if (quantizing_op->getNumRegions() != 0) { for (const auto& indexed_regions : llvm::enumerate(quantizing_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: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0)