Home
last modified time | relevance | path

Searched defs:ShapeAttrStorage (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_attributes.cc24 struct ShapeAttrStorage : public AttributeStorage { struct
25 using KeyTy = std::pair<ArrayRef<int64_t>, bool>;
27 explicit ShapeAttrStorage(ArrayRef<int64_t> shape, bool unranked = false) in ShapeAttrStorage() argument
30 bool operator==(const KeyTy& key) const { in operator ==()
33 static unsigned hashKey(const KeyTy& key) { in hashKey()
38 static ShapeAttrStorage* construct(mlir::AttributeStorageAllocator& allocator, in construct()
44 ArrayRef<int64_t> shape;
45 bool unranked = false;