Class Broadcast

Inheritance Relationships

Base Type

Class Documentation

class Broadcast : public mindspore::ops::PrimitiveC

Broadcasts the tensor to the whole group. Refer to Python API mindspore.ops.Broadcast for more details.

Public Functions

inline Broadcast()

Constructor.

~Broadcast() = default

Destructor.

void Init(const int64_t root_rank, const std::string &group = "hccl_world_group")

Init. Refer to the parameters of Python API mindspore.ops.Broadcast for the inputs.

void set_root_rank(const int64_t root_rank)

Set root_rank.

void set_group(const std::string &group)

Set group.

int64_t get_root_rank() const

Get root_rank.

Returns

root_rank.

std::string get_group() const

Get group.

Returns

group.