- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 157 for getUses (0.17 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/fuse_convolution_pass.cc
}); } // For dynamic case, the result of conv should be used by shape_of and mul. if (is_dynamic_broadcast) { auto conv_uses = (*conv_op.getODSResults(0).begin()).getUses(); if (std::distance(conv_uses.begin(), conv_uses.end()) != 2 || quant::FindUserOfType<shape::ShapeOfOp>(conv_op) == nullptr || quant::FindUserOfType<mhlo::MulOp>(conv_op) == nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 22:21:19 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_hashtables.cc
mlir::isa<IntegerType>(key_dtype) && mlir::cast<IntegerType>(key_dtype).getWidth() == 64))) { return false; } for (auto& use : hashtable->getUses()) { Operation* user = use.getOwner(); // Allow consuming hash table ops that can be covered by TensorFlow Lite // hash table kernels.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.6K 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) -
platforms/core-execution/workers/src/main/java/org/gradle/workers/internal/WorkerDaemonClientsManager.java
synchronized (lock) { List<WorkerDaemonClient> sortedClients = CollectionUtils.sort(idleClients, comparingInt(WorkerDaemonClient::getUses)); List<WorkerDaemonClient> clientsToStop = selectionFunction.transform(new ArrayList<>(sortedClients)); if (!clientsToStop.isEmpty()) { stopWorkers(clientsToStop); }
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Thu May 30 19:54:37 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc
// determine whether we have taken care of all uses and can remove // the arg. for (BlockArgument arg : func.getArguments()) { auto uses = arg.getUses(); use_count[arg.getArgNumber()] = std::distance(uses.begin(), uses.end()); argument_to_index.insert({arg, arg.getArgNumber()}); } // We're only considering return values that are the same (at this
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
llvm::SmallVector<Operation*, 4> candidates; for (Value value : values) { if (incoming) { candidates = {value.getDefiningOp()}; } else { candidates.assign(value.getUsers().begin(), value.getUsers().end()); } for (Operation* candidate_op : candidates) { if (cluster_ops.contains(candidate_op) || cluster_dependent_ops.contains(candidate_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_weight_param.cc
if (!type || !type.getElementType().isF32()) { return failure(); } return success( op->hasOneUse() && IsWeightQuantizableFunction(*op->getUses().begin(), type.getRank())); } // Checks if the operand is second operand of `tf.XlaCallModule` op for // `stablehlo.convolution` or `stablehlo.dot_general` with fully_quantizable // trait.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 05:56:10 UTC 2024 - 10.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/default_quant_params.cc
// bias is used immediately by the user. This assumption is always correct // after constant folding. bool UsedAsBias(Value value) { for (auto &use : value.getUses()) { auto biases = TFL::GetOpQuantSpec(use.getOwner())->biases_params; if (biases.find(use.getOperandNumber()) != biases.end()) return true; } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
if (!type || !type.getElementType().isF32()) return false; Value value = op.getResult(); // Check whether dynamic range quantization can be applied. for (auto& use : value.getUses()) { Operation* user = use.getOwner(); int operand_num = use.getOperandNumber(); std::unique_ptr<OpQuantSpec> spec = GetTFOpQuantSpec(user);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
values.getUses().begin().getUser()); values_slice_op.getResult().replaceAllUsesWith(top_k_op.getValues()); values_slice_op.erase(); } if (!indices.use_empty()) { auto indices_slice_op = llvm::dyn_cast_or_null<TFL::SliceOp>( indices.getUses().begin().getUser()); indices_slice_op.getResult().replaceAllUsesWith(top_k_op.getIndices());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0)