- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 38 for setInsertionPointToEnd (0.62 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
auto launch = builder->create<mlir::tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new mlir::Block); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<mlir::tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster. op->moveBefore(launch.GetBody().getTerminator());
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
op->getOperands()); Region region; Block *new_block = new Block; region.push_back(new_block); builder.setInsertionPointToEnd(®ion.front()); Operation *inner_op = builder.clone(*op); new_block->addArguments(op->getOperandTypes(), SmallVector<Location>(op->getNumOperands(), loc));
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/tf2xla/internal/passes/xla_cluster_formation.cc
cluster.getBody().push_back(new Block); call_op->replaceAllUsesWith(cluster.getResults()); call_op->moveBefore(&cluster.GetBody(), cluster.GetBody().end()); builder.setInsertionPointToEnd(&cluster.GetBody()); builder.create<mlir::tf_device::ReturnOp>(call_op->getLoc(), call_op->getResults());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/split_into_island_per_op_pass.cc
loc, result_types, control_type, mlir::ValueRange{}); island.getBody().push_back(new Block); Block* block = &island.getBody().back(); OpBuilder island_builder(original_island); island_builder.setInsertionPointToEnd(block); sub_op.replaceAllUsesWith(island.getOutputs()); sub_op.moveBefore(block, block->begin()); island_builder.create<tf_executor::YieldOp>(loc, sub_op.getResults()); return island; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
GetLiveOuts(®ion, &live_outs); // Build a `tf_device.return` op at end of region, with all live-out values // as operand. OpBuilder return_builder(builder->getContext()); return_builder.setInsertionPointToEnd(block); return_builder.create<tf_device::ReturnOp>(return_builder.getUnknownLoc(), live_outs); llvm::SmallVector<Type, 4> live_out_types;
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/extract_tpu_copy_with_dynamic_shape_op.cc
builder.getStringAttr(device_str), /*result_types=*/tpu_copy_with_dynamic_shape_op->getResultTypes()); new_device_launch_op.getBody().push_back(new Block); builder.setInsertionPointToEnd(&new_device_launch_op.GetBody()); auto* return_op = builder .create<tf_device::ReturnOp>( tpu_copy_with_dynamic_shape_op->getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/pin_ops_with_side_effects.cc
loc, op->getResultTypes(), ControlType::get(op->getContext()), control_tokens); Region region; Block *new_block = new Block; region.push_back(new_block); builder.setInsertionPointToEnd(®ion.front()); Operation *inner_op = builder.clone(*op); builder.create<YieldOp>(loc, inner_op->getResults()); outer_op.getBody().takeBody(region);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
} OpBuilder builder(module.getContext()); OwningOpRef<ModuleOp> stablehlo_module = builder.create<ModuleOp>(op.getLoc()); builder.setInsertionPointToEnd(stablehlo_module->getBody()); // Copy all referenced StableHLO functions to the new module. WalkResult result = WalkReachableFunctions( entry_func, [&](func::FuncOp f) -> WalkResult {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
} // Creates and returns a new `FuncOp` named "tf_quant__save". The resulting // `FuncOp`'s body has no ops. func::FuncOp CreateEmptySaveFunc(ModuleOp module_op) { OpBuilder builder(module_op); builder.setInsertionPointToEnd(&module_op.getBodyRegion().front()); auto filename_input_type = RankedTensorType::get( /*shape=*/{}, /*elementType=*/builder.getType<TF::StringType>());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
auto launch = builder->create<tf_device::LaunchOp>( loc, builder->getStringAttr(device), op->getResultTypes()); launch.getBody().push_back(new Block); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside launch. op->moveBefore(launch.GetBody().getTerminator());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0)