Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for ArrayInfo (0.12 sec)

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

        int size;
        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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/translate/mlir_roundtrip_flags.h

      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 =
          llvm::MapVector<std::string, ArrayInfo, llvm::StringMap<unsigned>>;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 04:56:10 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

    }
    
    // Mapping from node name to feed (index and ArrayInfo). Node name must outlive
    // this map.
    using FeedsByNode = absl::flat_hash_map<
        absl::string_view,
        absl::flat_hash_map<int, const std::pair<std::string, ArrayInfo>*>>;
    
    // Creates from a `GraphImportConfig::InputArrays` a mapping from a feeds output
    // tensor name to index and ArrayInfo. Keys and values are backed by
    // `GraphImportConfig::InputArrays`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top