- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for takeBody (0.85 sec)
-
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_types, operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); while_op.replaceAllUsesWith( new_while_op.getResults().take_front(while_op.getNumResults())); while_op.erase(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
} } auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_result_types, new_operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); for (auto i = 0; i < n; ++i) { if (!while_op.getResult(i).use_empty()) { auto j = id_map[i];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc
replicate_op->getLoc(), new_result_types, replicate_op->getOperands(), replicate_op->getAttrs()); // Move region to the new op. new_replicate_op.getRegion().takeBody(replicate_op.getRegion()); // Replace all old uses with new op results. int old_num_results = replicate_op->getNumResults(); replicate_op->replaceAllUsesWith(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
int new_idx = child_idx >= cluster_idx ? child_idx + num_cores_per_replica : child_idx; new_parallel_execute->getRegions()[new_idx].takeBody( old_parallel_execute.getRegions()[old_idx]); } return cluster_idx; } mlir::tf_device::LaunchOp WrapOpInLaunch(mlir::OpBuilder* builder, mlir::Location loc,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/raise_custom_ops.cc
inner_op->setOperand(idx_args.index(), idx_args.value()); } custom_op->setAttrs(inner_op->getAttrs()); builder.create<YieldOp>(loc, inner_op->getResults()); custom_op.getBody().takeBody(region); op->replaceAllUsesWith(custom_op); op->erase(); } } } // namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateRaiseCustomOpsPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto new_while = builder.create<TF::WhileRegionOp>( while_op.getLoc(), body_region.front().getTerminator()->getOperandTypes(), new_while_operands, while_op->getAttrs()); new_while.getBody().takeBody(body_region); new_while.getCond().takeBody(cond_region); for (const auto& entry : output_buffer_to_size) { (*buffer_to_size)[new_while.getResult(std::get<0>(entry))] = { new_while.getResult(std::get<1>(entry)), std::get<2>(entry)};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
parallel_execute.getLoc(), num_regions, new_result_types); for (auto region : llvm::zip(new_parallel_execute.getRegions(), parallel_execute.getRegions())) std::get<0>(region).takeBody(std::get<1>(region)); for (auto result : llvm::zip(results_to_remap, new_parallel_execute.getResults())) std::get<0>(result).replaceAllUsesWith(std::get<1>(result)); parallel_execute.erase();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 06 04:46:18 UTC 2022 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
tf_device::LaunchOp launch_op = builder->create<tf_device::LaunchOp>( builder->getUnknownLoc(), builder->getStringAttr(c.target), live_out_types); // Attach the region to launch_op. launch_op.getBody().takeBody(region); // Replace any external uses of live-out values with return values of launch // op. So live-out values no longer escape the region. ReplaceLiveOutExternalUses(live_outs, launch_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Oct 05 13:30:21 UTC 2023 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
auto new_while_op = GetUpdatedWhileOp( while_op, while_op.getCond().getArgumentTypes(), args_to_erase); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); llvm::BitVector erase_indices; UpdateTerminatorArguments(new_while_op.getBody(), args_to_erase, erase_indices);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
region_if.getPred()); // Move all regions from the old `mhlo.if` op to its replacement. new_if.getTrueBranch().takeBody(region_if.getTrueBranch()); new_if.getFalseBranch().takeBody(region_if.getFalseBranch()); // Forward result from old `mhlo.if` with replacement. SmallVector<Value> old_if_results = region_if.getResults();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0)