- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for getArgAttr (0.24 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/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)