- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 106 for getUsers (0.13 sec)
-
tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc
Type ref_qtype = nullptr; for (auto *var_handle_user : var_handle_op.getResult().getUsers()) { auto read_variable_op = dyn_cast_or_null<ReadVariableOp>(var_handle_user); if (!read_variable_op) continue; for (auto *read_variable_user : read_variable_op.getResult().getUsers()) { auto q_op = dyn_cast_or_null<QuantizeOp>(read_variable_user); if (!q_op || ref_qtype) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/propagate_quantize_type.cc
if (!function_name.starts_with(kDequantizeFunctionName)) return failure(); llvm::SmallVector<Operation*> users(op->getUsers().begin(), op->getUsers().end()); bool changed = false; for (auto& use : op->getUses()) { Operation* user_op = use.getOwner(); int user_idx = use.getOperandNumber(); if (!IsOpWithInt8TypeOperand(user_op)) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc
if (!result.hasOneUse()) return resource; Operation* result_user = *result.getUsers().begin(); auto assign_var = dyn_cast<TF::AssignVariableOp>(result_user); if (!assign_var) return resource; auto handle = assign_var.getResource(); // Skip result if cluster writes to the same variable via multiple results. for (Operation* handle_user : handle.getUsers()) { if (handle_user == assign_var) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_composite_resource_ops.cc
resource_users.emplace_back(op); }; llvm::SmallVector<Operation*, 4> resource_users_to_visit; for (auto composite_arguments : replicate.GetPackedBlockArguments()) { for (auto resource_user : composite_arguments.getUsers()) resource_users_to_visit.emplace_back(resource_user); } while (!resource_users_to_visit.empty()) { llvm::SmallVector<Operation*, 4> new_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/readonly_references_to_resources.cc
if (!classes_attr) { // Attempt to parse "_class" from the IdentityOp that follows VariableV2. // For read-only reference variables, IdentityOp should be the only user of // VariableV2. auto identity_op = op->getUsers().begin(); classes_attr = identity_op->getAttrOfType<ArrayAttr>(kClassAttr); if (!classes_attr) { op->emitOpError() << "has no '_class' and 'shared_name' attributes"; return StringRef(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/freeze_global_tensors.cc
freezeable[val] = false; continue; } freezeable[val] = true; // Verify users are supported kind. for (Operation *user : val.getUsers()) { if (!(isa<TF::ReadVariableOp>(user) || isa<CallOpInterface>(user))) { freezeable[val] = false; // Error out early if possible. if (!allow_mutable_tensors) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/device_index_selector.cc
// future. OpBuilder b(op); RankedTensorType type = RankedTensorType::get({}, b.getIntegerType(32)); int index = op.getDeviceNames().size(); for (auto use : op.getOperation()->getUsers()) { // Skip if it doesn't feed into case. Alternatively this could always // return the CPU device index if it exists. if (!isa<TF::CaseOp>(use)) return; } DenseElementsAttr attr =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/remove_identity_op_pattern.cc
namespace mlir { namespace quant { LogicalResult RemoveIdentity::matchAndRewrite(TF::IdentityOp identity, PatternRewriter &rewriter) const { for (Operation *user : identity->getUsers()) { // Replace the op with the input if output is only used by TF ops. // Currently this is more on the conservative side since we need to ensure // every consumer op to be a TF op before applying this pattern. We can
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 15 06:13:49 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc
return mlir::failure(); } for (auto result : old_parallel_execute.getExecuteOutputs()) { for (mlir::Operation* user : llvm::make_early_inc_range(result.getUsers())) { if (llvm::isa<mlir::TF::TPUPartitionedOutputV2Op>(user)) { assert(user->use_empty()); user->erase(); } } } for (auto operand : cluster.getOperands()) {
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/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)