- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for getArgAttr (1.22 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} return success(); } static bool HasAnyTfSavedModelArgAttr(func::FuncOp func) { for (int i = 0, e = func.getNumArguments(); i < e; i++) { if (func.getArgAttr(i, kTfSavedModelIndexPathAttr) || func.getArgAttr(i, "tf_saved_model.bound_input")) { return true; } } for (int i = 0, e = func.getNumResults(); i < e; i++) { if (func.getResultAttr(i, kTfSavedModelIndexPathAttr) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_dataflow.cc
IsComposite result; if (auto barg = mlir::dyn_cast<BlockArgument>(value)) { if (func::FuncOp func = dyn_cast<func::FuncOp>(barg.getOwner()->getParentOp())) { if (func.getArgAttr(barg.getArgNumber(), kCompositeDevice)) { result.is_on_composite_device = true; } return result; } } return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/optimize_global_tensors.cc
for (auto func : module.getOps<func::FuncOp>()) { llvm::BitVector args_to_erase(func.getNumArguments()); for (int i = 0, e = func.getNumArguments(); i < e; i++) { if (func.getArgAttr(i, "tf_saved_model.bound_input") && func.getArgument(i).use_empty()) { args_to_erase.set(i); } } func.eraseArguments(args_to_erase); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/decompose.cc
arg.index(), kAttrArgumentNameAttr); auto attribute = op->getAttr(attr_name.getValue()); if (!attribute) { attribute = compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr); } if (!attribute && attr_name.getValue() == "out_type") { auto type = op->getResult(0).getType(); if (mlir::isa<TensorType>(type)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
if (operand.index() >= call_op.getNumOperands()) { auto attr_name = signature.getArgAttrOfType<StringAttr>( operand.index(), kAttrArgumentNameAttr); auto attr_value = signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr); arg_attrs->push_back( rewriter.getNamedAttr(attr_name.getValue(), attr_value)); continue; } // The index is valid for the call_op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
// alias. const bool has_arg_unique_id_attrs = llvm::any_of(func_op.getArguments(), [&](const BlockArgument& arg) { return func_op.getArgAttr(arg.getArgNumber(), kResourceArgUniqueIdAttr); }); if (has_arg_unique_id_attrs) { // Resource arguments have IDs attached (via `kResourceArgUniqueIdAttr`)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0)