Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for IsTypeLegalizedWithMlir (0.33 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc

    }
    
    bool IsOpLegalizedWithMlir(Operation& op) {
      auto abstractOp = op.getRegisteredInfo();
      if (!abstractOp) return false;
      return IsTypeLegalizedWithMlir(abstractOp->getTypeID());
    }
    
    bool IsTypeLegalizedWithMlir(const TypeID& type_id) {
      return MlirAlwaysOps().contains(type_id);
    }
    
    bool IsOpAllowedTf2xlaFallback(const TypeID& type_id) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 04:08:35 UTC 2024
    - 21.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf.cc

        // If the pattern does not have a specific operation, always include it,
        // If the pattern is in include_ops then include it.
        bool include =
            !pat_op_name ||
            IsTypeLegalizedWithMlir(pat_op_name->getRegisteredInfo()->getTypeID());
        if (include) to.add(std::move(pattern));
      }
    
      // Don't filter PDLPatterns.
      to.add(std::move(from.getPDLPatterns()));
    
      return to;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 10.2K bytes
    - Viewed (0)
Back to top