Searched defs:LayoutAttrStorage (Results 1 – 1 of 1) sorted by relevance
58 struct LayoutAttrStorage : public AttributeStorage { struct59 using Layout = tensorflow::dtensor::Layout;60 using KeyTy = Layout;62 explicit LayoutAttrStorage(Layout layout) : layout(std::move(layout)) {} in LayoutAttrStorage() argument64 bool operator==(const KeyTy& key) const { return key == KeyTy(layout); } in operator ==()66 static llvm::hash_code hashKey(const KeyTy& key) { in hashKey()71 static LayoutAttrStorage* construct( in construct()77 Layout layout;