Commit 47c59f96 authored by Scott Cyphers's avatar Scott Cyphers

Add doc for broadcast.

parent 2e420183
......@@ -101,6 +101,12 @@ namespace ngraph
class BroadcastCall : public BuiltinCall
{
public:
/**
** /param arg The tensor view to be broadcast.
** /param shape The shape of the result
** /param broadcast_axes The axis positions (0-based) in the result that are being broadcast.
** the remaining axes in shape must be the same as the shape of arg.
**/
BroadcastCall(const Node::ptr& arg, const Shape& shape, std::vector<size_t> broadcast_axes)
: BuiltinCall(s_op, {arg})
, m_shape(shape)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment