Home
last modified time | relevance | path

Searched refs:HcomBroadcast (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dhcom_ops_declare.cc29 INPUT_MAP(HcomBroadcast) = EMPTY_INPUT_MAP;
30 DYN_INPUT_MAP(HcomBroadcast) = {{1, DYN_INPUT_DESC(x)}};
31 DYN_OUTPUT_MAP(HcomBroadcast) = {{0, DYN_OUTPUT_DESC(y)}};
32 ATTR_MAP(HcomBroadcast) = {{"root_rank", ATTR_DESC(root_rank, AnyTraits<int64_t>())},
34 REG_ADPT_DESC(HcomBroadcast, kNameBroadcast, ADPT_DESC(HcomBroadcast))
Dhcom_ops_declare.h29 DECLARE_OP_ADAPTER(HcomBroadcast)
30 DECLARE_OP_USE_DYN_INPUT(HcomBroadcast)
31 DECLARE_OP_USE_DYN_OUTPUT(HcomBroadcast)
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dconvert.h52 using HcomBroadcast = ge::op::HcomBroadcast; variable
118 …void SetupBroadcast(const std::shared_ptr<HcomBroadcast> &broadcast, const std::vector<GeTensorDes…
Dconvert.cc304 void DfGraphConvertor::SetupBroadcast(const std::shared_ptr<HcomBroadcast> &broadcast, in SetupBroadcast()
496 auto broadcast = std::make_shared<HcomBroadcast>("broadcast_parameter"); in GenerateBroadcastGraph()