Model::EvaluationSchema Struct Reference

Schema for the traced forward operators. More...

Detailed Description

Schema for the traced forward operators.

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

#include <Model.h>

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

Public Attributes
std::vector< Size >	dynamic_dims
std::vector< TensorShape >	intmd_shapes
at::DispatchKey	dispatch_key

Member Function Documentation

operator!=()

bool operator!=

(

const EvaluationSchema &

other

)

const

operator<()

bool operator<

(

const EvaluationSchema &

other

)

const

operator==()

bool operator==

(

const EvaluationSchema &

other

)

const

Member Data Documentation

dispatch_key

at::DispatchKey dispatch_key

dynamic_dims

std::vector<Size> dynamic_dims

intmd_shapes

std::vector<TensorShape> intmd_shapes