torch::jit::TreeToken Struct Reference

Public Member Functions
	operator bool ()
std::vector< Node * >	removeTransposesAndGatherMatmuls ()

Static Public Member Functions
static TreeToken	mm (Node *mm)
static TreeToken	transpose (Node *t, TreeToken &inp_token)
static TreeToken	add (Node *add, TreeToken &l, TreeToken &r)

Data Fields
uint64_t	tree_size = 0
Node *	node = nullptr
bool	is_root = false

Detailed Description

Definition at line 133 of file batch_mm.cpp.

---

The documentation for this struct was generated from the following file:

• torch/csrc/jit/passes/batch_mm.cpp