- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 124 for getUsers (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
for (auto& use : uses_optional.value()) { if (llvm::isa<TF::WhileOp>(use.getUser())) { if (*while_op != nullptr) { use.getUser()->emitOpError() << "multiple users of function."; return LogicalResult::failure(); } else { *while_op = llvm::cast<TF::WhileOp>(use.getUser()); } } else { use.getUser()->emitOpError() << "non while use of function.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
// terminator. bool checkIfAnyUserIsConnectedToTermiantor(BlockArgument op) const { for (const auto& user : op.getUsers()) { if (user->template hasTrait<OpTrait::IsTerminator>()) return true; if (auto next_user = dyn_cast_or_null<TF::IdentityOp>(user)) { return (*(next_user->getResult(0).getUsers().begin())) ->template hasTrait<OpTrait::IsTerminator>(); } } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc
// Bypass Transpose nodes for all results. for (OpResult result : op->getResults()) { result.setType( cast<TransposeOp>(*result.getUsers().begin()).getY().getType()); for (Operation* transpose : result.getUsers()) { transpose->getResult(0).replaceAllUsesWith(result); } } // Maybe add a Transpose node for all operands (or reuse existing transposes).
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types.cc
// Check if an op result value is consumed by qint -> int TF Cast OP. bool IsQintValueQintToIntCast(Value v) { if (!IsIllegalType(v.getType())) { return true; } if (v.getUsers().empty()) { return false; } return llvm::all_of(v.getUsers(), [&](OpOperand operand) { return IsQintToIntCast(operand.getOwner()); }); } // Check if an op operand value is defined by int -> qint TF Cast OP.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/check_control_dependencies.cc
worklist.push(source_op); while (!worklist.empty()) { IslandOp curr_op = worklist.front(); worklist.pop(); if (curr_op == target_op) break; for (Operation* user : curr_op.getControl().getUsers()) { auto user_island = dyn_cast<IslandOp>(user); if (!user_island) continue; // We have labeled `user_island` before so it also must have been added to // `worklist` before.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
if (member.source.is<BlockArgument *>()) continue; Operation *op = member.source.dyn_cast<Operation *>(); assert(op && "member operation must be not null"); for (Operation *user : op->getUsers()) { // Skip users in other blocks. if (user->getBlock() != op->getBlock()) continue; // Skip users is in the `dst_root` or `src_root` clusters, if we'll merge
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc
// ops in first_if and the ops in second_if, it's not safe to merge // them. std::vector<Operation*> dependencies; for (auto* user : first_if.getOperation()->getUsers()) { if (!downstream_if_ops.contains(user)) { dependencies.push_back(user); } } for (auto* successor : side_effect_analysis.DirectControlSuccessors(first_if.getOperation())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
min -= kNearZeroTolerance; max += kNearZeroTolerance; } } }; template <typename VerifierT> bool UsedBy(Operation* op) { for (Operation* user : op->getUsers()) { if (llvm::isa_and_nonnull<VerifierT>(user)) return true; } return false; } template <typename VerifierT> void CreateVerifier(Operation* quantizing_op, Operation* quantized_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc
queue.push(root_op); while (!queue.empty()) { Operation* op = queue.front(); queue.pop(); if (llvm::isa<tf_device::ReturnOp>(op)) continue; ops_to_move.insert(op); for (Operation* user : op->getUsers()) { if (!ops_to_move.contains(user)) { queue.push(user); } } } return ops_to_move; } tf_device::LaunchOp CreateLaunchForBlock(OpBuilder* builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.4K bytes - Viewed (0)