- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for getTerminator (0.44 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
Operation* tpu_copy_with_dynamic_shape_op) { llvm::SmallSetVector<Value, 4> new_launch_op_results; new_launch_op_results.insert( old_launch_op->GetBody().getTerminator()->getOperands().begin(), old_launch_op->GetBody().getTerminator()->getOperands().end()); for (Value operand : tpu_copy_with_dynamic_shape_op->getOperands()) { if (GetOpOfValue(operand)->getParentRegion() ==
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/tensorflow/transforms/convert_tf_control_flow_to_scf.cc
// If `destination_block` isn't empty, erase its terminator to ensure that it // never contains two terminator-like ops after merging. if (!destination_block->empty()) rewriter.eraseOp(destination_block->getTerminator()); destination_block->addArguments( block_arguments_type, SmallVector<Location>(block_arguments_type.size(), source_block->getParent()->getLoc()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_parallel_execute_sink_resource_write.cc
auto results = parallel_execute.GetRegionOutputs(i); auto& results_to_remove = results_to_remove_by_region[i]; results_to_remove.reserve(results.size()); Operation* terminator = block.getTerminator(); for (auto result : llvm::enumerate(results)) { TF::AssignVariableOp assign_var = GetSingleUseResourceWrite(parallel_execute, result.value()); if (!assign_var) {
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/tpu_annotate_dynamic_shape_inputs.cc
GetOpOfValue(cluster_func_operand.value())); if (!device_launch_op) continue; for (auto result : llvm::zip( device_launch_op.getResults(), device_launch_op.GetBody().getTerminator()->getOperands())) { if (std::get<0>(result) == cluster_func_operand.value() && llvm::isa<TF::TPUAnnotateTensorsWithDynamicShapeOp>( std::get<1>(result).getDefiningOp())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_functional_to_executor.cc
std::next(copy_range.begin()) == copy_range.end() && isa<tf_executor::GraphOp>(*copy_range.begin())) { // Already a graph. return; } auto return_op = dyn_cast<func::ReturnOp>(body.getTerminator()); if (!return_op) { LLVM_DEBUG(llvm::dbgs() << "Expect function to end with return\n"); return; } // Build GraphOp. OpBuilder builder(&body, body.begin());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc
void RewriteQuantizedIOPass::runOnOperation() { ModuleOp module = getOperation(); OpBuilder builder(module); module.walk([&](func::FuncOp func) { Block& block = func.front(); Operation* terminator = block.getTerminator(); // Replace input_arg(tensor<quant_type>) -> tfr.cast // with input_arg(tensor<storage_type>) -> tfr.cast for (BlockArgument arg : block.getArguments()) { Type arg_type = arg.getType();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<mlir::tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster. op->moveBefore(launch.GetBody().getTerminator()); builder->restoreInsertionPoint(insert_point); return launch; }
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/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
op->replaceAllUsesWith(launch); builder->setInsertionPointToEnd(&launch.GetBody()); builder->create<tf_device::ReturnOp>(loc, op->getResults()); // Move op inside cluster. op->moveBefore(launch.GetBody().getTerminator()); } llvm::SmallVector<Operation*, 4> GetResourceOpsUsingCompositeArgsInReplicate( tf_device::ReplicateOp replicate) { llvm::SmallVector<Operation*, 4> resource_users;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 16 17:41:12 UTC 2023 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_replicate_invariant_resource_writes.cc
llvm::ArrayRef<TF::AssignVariableOp> tail_assign_variable_ops) { const auto num_replicas = replicate_op.getN(); auto return_op = llvm::dyn_cast<tf_device::ReturnOp>( replicate_op.getRegion().front().getTerminator()); // Get the new result types. // TODO(prakalps): Do not add a value to returned values if it is already // returned. auto new_result_types = llvm::to_vector<4>(replicate_op->getResultTypes());
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/transforms/remove_unused_while_results.cc
// Don't prune if result is used. if (!result.use_empty()) return false; Block& body_block = while_op.getBody().front(); Block& cond_block = while_op.getCond().front(); Operation* body_yield_op = body_block.getTerminator(); // The body yield operand, body block argument, condition block argument, and // result all correspond to each other (see definition of `WhileRegionOp`). int idx = result.getResultNumber();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 5K bytes - Viewed (0)