Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for SubTypeInfo (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h

      DataType imported_dtype;
    
      // Node "shape" attribute value.
      TensorShapeProto shape;
    };
    
    struct ArrayInfo : public ArrayInfoBase {
      using SubTypeInfo = ArrayInfoBase;
      // DT_RESOURCE and DT_VARIANT have subtypes
      std::vector<SubTypeInfo> subtypes;
    };
    
    struct GraphImportConfig {
      // Returns string representation of config.
      std::string str() const;
    
      using InputArrays =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 04:56:10 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.cc

        TF_RET_CHECK(absl::SimpleAtoi(dim_str, &size));
        dims.push_back(size);
      }
      return dims;
    }
    
    static Status HandleSubtype(absl::string_view subtype,
                                ArrayInfo::SubTypeInfo* result) {
      std::vector<std::string> shape_and_type = absl::StrSplit(subtype, ':');
    
      std::vector<int> dims;
      if (shape_and_type.size() > 2) {
        return errors::FailedPrecondition("Invalid argument: '", subtype,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 10.4K bytes
    - Viewed (0)
Back to top