ngraph.utils.broadcasting

Module Attributes

Functions

get_broadcast_axes(output_shape, input_shape)

Generate a list of broadcast axes for ngraph++ broadcast.

get_dtype(ngraph_type)

Return a numpy.dtype for an ngraph element type.

make_constant_node(value[, dtype])

Return an ngraph Constant node with the specified value.

Classes

AxisSet

ngraph.impl.AxisSet wraps ngraph::AxisSet

Node

ngraph.impl.Node wraps ngraph::Node

Exceptions

Modules