Searched defs:TreeToken (Results 1 – 1 of 1) sorted by relevance
171 struct TreeToken { struct172 uint64_t tree_size = 0; // NOTE: measured in number of leaves i.e. mm ops173 Node* node = nullptr;174 bool is_root = false;176 static TreeToken mm(Node* mm) { in mm()186 static TreeToken transpose(Node* t, TreeToken& inp_token) { in transpose()201 static TreeToken add(Node* add, TreeToken& l, TreeToken& r) { in add()214 explicit operator bool() { in operator bool()218 std::vector<Node*> removeTransposesAndGatherMatmuls() { in removeTransposesAndGatherMatmuls()