Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Data Fields
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
Nodenode = 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: