Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for kAttrArgumentDefaultAttr (0.29 sec)

  1. 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)
  2. 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)
Back to top