Model::TraceSchema Struct Reference

Schema for the traced forward operators. More...

Detailed Description

Schema for the traced forward operators.

The schema is determined by the batch dimensions of all input variables and model parameters.

#include <Model.h>

Public Member Functions
bool	operator== (const TraceSchema &other) const
bool	operator< (const TraceSchema &other) const

Public Attributes
std::vector< Size >	batch_dims
at::DispatchKey	dispatch_key

Member Function Documentation

operator<()

bool operator<

(

const TraceSchema &

other

)

const

operator==()

bool operator==

(

const TraceSchema &

other

)

const

Member Data Documentation

batch_dims

std::vector<Size> batch_dims

dispatch_key

at::DispatchKey dispatch_key