Searched defs:MeshAttrStorage (Results 1 – 1 of 1) sorted by relevance
29 struct MeshAttrStorage : public AttributeStorage { struct30 using Mesh = tensorflow::dtensor::Mesh;31 using KeyTy = Mesh;33 explicit MeshAttrStorage(Mesh mesh) : mesh(std::move(mesh)) {} in MeshAttrStorage() function35 bool operator==(const KeyTy& key) const { return key == KeyTy(mesh); } in operator ==()37 static llvm::hash_code hashKey(const KeyTy& key) { in hashKey()42 static MeshAttrStorage* construct(mlir::AttributeStorageAllocator& allocator, in construct()46 Mesh mesh;