Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for kAttrArgumentNameAttr (0.26 sec)

  1. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        // default value of the operand needs to be used.
        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;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/passes/decompose.cc

            new_operands.push_back(build_list_op.getOut());
          } else {
            auto attr_name = compose_func.getArgAttrOfType<StringAttr>(
                arg.index(), kAttrArgumentNameAttr);
            auto attribute = op->getAttr(attr_name.getValue());
            if (!attribute) {
              attribute =
                  compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr);
            }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

            all_attrs.insert(attr.getName().strref());
          }
          for (const auto& operand : llvm::enumerate(getArgumentTypes())) {
            if (auto attr_name = getArgAttrOfType<StringAttr>(
                operand.index(), kAttrArgumentNameAttr)) {
              all_attrs.insert(attr_name.getValue());
            }
          }
          return all_attrs;
        }
      }];
    
      let hasVerifier = 1;
      let hasCustomAssemblyFormat = 1;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc

        }
    
        if (!arg_type.isa<TensorType>()) {
          if (first_attr == -1) {
            first_attr = arg.index();
          }
          auto name =
              func.getArgAttrOfType<StringAttr>(arg.index(), kAttrArgumentNameAttr);
          if (!name) {
            func.emitError(
                llvm::Twine(arg.index()) +
                " attribute argument doesn't have a tfr.name attribute.");
            return failure();
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
Back to top