- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for kAttrArgumentDefaultAttr (0.25 sec)
-
tensorflow/compiler/mlir/tfr/ir/tfr_ops.h
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project namespace mlir { namespace TFR { constexpr char kAttrArgumentNameAttr[] = "tfr.name"; constexpr char kAttrArgumentDefaultAttr[] = "tfr.default"; constexpr char kAttrArgumentTypeAttr[] = "tfr.type"; class TFRDialect : public Dialect { public: explicit TFRDialect(MLIRContext *context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 01 14:00:36 UTC 2023 - 2.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
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)