- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ArrayInfo (0.12 sec)
-
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) -
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) -
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)