Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for compose_func_name (0.17 sec)

  1. tensorflow/compiler/mlir/tfr/utils/utils.cc

          compose_func_name.push_back('_');
          compose_func_name.push_back(tf_op_name[i] + 'a' - 'A');
        } else {
          compose_func_name.push_back(tf_op_name[i]);
        }
      }
      return compose_func_name;
    }
    
    std::string GetTFOpName(StringRef compose_func_name) {
      std::string tf_op_name;
      bool after_underscore = false;
      for (int i = 0; i < compose_func_name.size(); ++i) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 6.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfr/passes/decompose.cc

        // bridge.
        if (op->isRegistered()) {
          return WalkResult::advance();
        }
    
        // Find out the compose function
        auto compose_func_name = GetComposeFuncName(op->getName().getStringRef());
        auto compose_func = table.lookup<TFRFuncOp>(compose_func_name);
        if (!compose_func || compose_func.isExternal()) {
          // There are no decomposition methods defined for this op, skip.
          return WalkResult::advance();
    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/utils/utils.h

    // This is a hardcoded rule for mapping a TFR function op name to the
    // corresponding TF opname. Examples:
    //   tf__pack -> tf.Pack
    //   tf__concat_v2 => tf.ConcatV2
    std::string GetTFOpName(StringRef compose_func_name);
    
    // Validate the attributes of 'src' is either contained in the registered
    // attribute sets or in the allowed list.
    LogicalResult ValidateAttrs(Operation* src, const StringSet<>& registered);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 2.6K bytes
    - Viewed (0)
Back to top