container::einsum_utils::BCast Struct Reference

Collaboration diagram for container::einsum_utils::BCast:

Static Public Member Functions
static void	reverse (std::vector< int64_t > &vec)

Public Attributes
bool	valid = true
bool	requires_broadcast = false
int64_t	x_batch_size = 1
int64_t	y_batch_size = 1
int64_t	z_batch_size = 1
std::vector< int64_t >	x_batch_shape = {}
std::vector< int64_t >	y_batch_shape = {}
std::vector< int64_t >	z_batch_shape = {}
std::vector< int64_t >	x_bcast_shape = {}
std::vector< int64_t >	y_bcast_shape = {}

Member Function Documentation

reverse()

static void container::einsum_utils::BCast::reverse ( std::vector< int64_t > &  vec )

inlinestatic

Member Data Documentation

requires_broadcast

bool container::einsum_utils::BCast::requires_broadcast = false

valid

bool container::einsum_utils::BCast::valid = true

x_batch_shape

std::vector<int64_t> container::einsum_utils::BCast::x_batch_shape = {}

x_batch_size

int64_t container::einsum_utils::BCast::x_batch_size = 1

x_bcast_shape

std::vector<int64_t> container::einsum_utils::BCast::x_bcast_shape = {}

y_batch_shape

std::vector<int64_t> container::einsum_utils::BCast::y_batch_shape = {}

y_batch_size

int64_t container::einsum_utils::BCast::y_batch_size = 1

y_bcast_shape

std::vector<int64_t> container::einsum_utils::BCast::y_bcast_shape = {}

z_batch_shape

std::vector<int64_t> container::einsum_utils::BCast::z_batch_shape = {}

z_batch_size

int64_t container::einsum_utils::BCast::z_batch_size = 1

---

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

• /home/runner/work/abacus-develop/abacus-develop/source/source_base/module_container/ATen/ops/einsum_op.cpp