Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetTFOpName (0.25 sec)

  1. 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)
  2. tensorflow/compiler/mlir/tfr/utils/utils.cc

          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) {
        if (compose_func_name[i] >= 'A' && compose_func_name[i] <= 'Z') {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 06 03:08:33 UTC 2023
    - 6.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc

        const llvm::StringMap<Attribute>& derived_attrs) const {
      // Create the new op
      Location loc = call_op.getLoc();
      rewriter.setInsertionPointAfter(call_op);
      std::string tf_op_name = GetTFOpName(call_op.getCallee());
      OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list);
      Operation* new_op = rewriter.create(new_state);
      if (materialize_derived_attrs_) {
    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