- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 136 for getUsers (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_while_results.cc
return false; } } else if (!isMemoryEffectFree(candidate_op)) { return false; } // Don't prune if the body block argument has any other user. for (Operation* op : body_block_argument.getUsers()) { if (op != candidate_op) return false; } // Don't prune if the condition block argument has any user. if (!cond_block_argument.use_empty()) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc
llvm::SmallVectorImpl<Operation*>& erase_list, llvm::SmallPtrSetImpl<Operation*>& dead_ops) { for (mlir::Value res : op->getResults()) { for (Operation* user : res.getUsers()) { if (!dead_ops.insert(user).second) continue; RecursiveRemove(user, erase_list, dead_ops); } } erase_list.push_back(op); for (auto& use : op->getOpOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_saved_model_assets.cc
auto asset = LookupBoundInputOfType<AssetOp>(func, i, symbol_table); if (!asset) continue; auto arg = func.getArgument(i); bool arg_is_deletable = true; for (auto user : arg.getUsers()) { if (auto read_op = llvm::dyn_cast<TF::InitializeTableFromTextFileV2Op>(user)) { init_table_from_text_file_ops_to_erase.push_back(read_op); } else {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jan 30 01:12:09 UTC 2023 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/topological_sort.cc
while (!todo.empty()) { Value value = todo.front(); todo.pop(); // All operations that have all their inputs available are good to go. // Uses, not Users, in case getUsers ever dedups. for (OpOperand& operand : value.getUses()) { Operation* user = ancestor[operand.getOwner()]; remaining_incoming_data_edges[user]--; if (remaining_incoming_data_edges[user] == 0 &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 08 17:01:11 UTC 2022 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/cast_bf16_ops_to_f32.cc
value.setType(CloneTypeWithNewElementType(value.getType(), rewriter.getF32Type())); rewriter.setInsertionPointAfterValue(value); for (Operation* user : op->getUsers()) { for (int i = 0; i < user->getNumOperands(); i++) { if (user->getOperand(i) == value) { Value bf16_cast = rewriter.create<TF::CastOp>( user->getLoc(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Dec 10 05:52:02 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
head_outside_compiled_ops.insert(&cluster_op); } } for (auto head_outside_compiled_op : llvm::reverse(head_outside_compiled_ops)) { auto users = head_outside_compiled_op->getUsers(); if (users.empty() || HasOutsideCompilationAttribute(head_outside_compiled_op)) continue; bool should_expand_op_to_host_computation = true; for (auto consumer_op : users) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/guarantee_all_funcs_one_use.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/hoist_loop_invariant.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_formation.cc
for (Operation& op : region->front()) { for (Value v : op.getResults()) { // A value is live-out if any of its users are not inside value producer's // region. bool is_live_out = llvm::any_of(v.getUsers(), [&](Operation* user) { return !region->isAncestor(user->getParentRegion()); }); if (is_live_out) live_outs->emplace_back(v); } } }
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/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
LogicalResult matchAndRewrite(func::CallOp call_op, PatternRewriter& rewriter) const override { if (call_op.getNumResults() != 1) return failure(); auto users = call_op->getUsers(); for (auto user : users) { if (!llvm::isa<mlir::stablehlo::UniformDequantizeOp>(user)) { return failure(); } } auto func_name = call_op.getCallee();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0)