Class Broadcast
Defined in File broadcast.h
Inheritance Relationships
Base Type
public mindspore::ops::PrimitiveC
(Class PrimitiveC)
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.
-
inline Broadcast()