- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 45 for getTerminator (1.06 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/resource_op_lifting_cleanup.cc
cloned_cond.eraseArgument(idx); cloned_body.front().getTerminator()->eraseOperand(idx); cloned_body.eraseArgument(idx); } // Patch up branch function types. for (func::FuncOp func : {cloned_cond, cloned_body}) { func.setType( FunctionType::get(func.getContext(), func.front().getArgumentTypes(), func.front().getTerminator()->getOperandTypes())); }
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/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
} mlir::tf_device::LaunchOp launch = CreateLaunchForBlock( builder, cluster, /*before=*/true, launch_block, host_device); for (auto result : llvm::zip(launch.GetBody().getTerminator()->getOperands(), launch.getResults())) replaceAllUsesInRegionWith(std::get<0>(result), std::get<1>(result), cluster.getBody()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
for (OpOperand& operand : reduced_launch_op.GetBody().getTerminator()->getOpOperands()) { reduced_return_operand_map.insert( {operand.get(), operand.getOperandNumber()}); } for (int i = 0; i < original_launch_op->getNumResults(); i++) { Value operand = original_launch_op.GetBody().getTerminator()->getOperand(i); auto defining_op = operand.getDefiningOp();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K 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/tf2xla/internal/passes/extract_outside_compilation.cc
{host_if.getThenBranch().front().getTerminator()}, core_to_compilation_key, core_to_device_ordinal, /*control_above=*/true, is_map_oc, communication_key_index))) return WalkResult::interrupt(); if (failed(MoveToHostMultiCluster( device_cluster, &if_op.getElseBranch().front(), {host_if.getElseBranch().front().getTerminator()},
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
RegionRange regions) { for (Region* region : regions) { assert(llvm::hasSingleElement(*region) && "Expected single block region"); Block& front = region->front(); auto old_return = front.getTerminator(); assert(old_return->getNumOperands() == op_->getNumResults()); auto new_return_operands = llvm::to_vector<4>(old_return->getOperands()); new_return_operands.resize(num_new_results_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K 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/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
BlockArgument arg = std::get<1>(sharding_and_arg); if (failed(VerifySharding(arg.getType(), sharding))) return mlir::failure(); } Operation* terminator = function_block.getTerminator(); for (auto sharding_and_retval : llvm::zip(sharding_for_rets, terminator->getOpOperands())) { const auto& sharding = std::get<0>(sharding_and_retval);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
FunctionType type; if (passthru_extra_args) { type = FunctionType::get(context, types, types); } else { SmallVector<Type, 4> result_types; auto operands = region.front().getTerminator()->getOperandTypes(); result_types.append(operands.begin(), operands.end()); type = FunctionType::get(context, types, result_types); } auto outlined_func = builder.create<func::FuncOp>(loc, name, type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0)