Searched refs:broadcast_desc (Results 1 – 2 of 2) sorted by relevance
305 const std::vector<GeTensorDesc> &broadcast_desc, in SetupBroadcast() argument308 if (broadcast_desc.size() != broadcast_input.size()) { in SetupBroadcast()312 (void)broadcast->create_dynamic_output_y(static_cast<unsigned int>(broadcast_desc.size())); in SetupBroadcast()315 (void)broadcast->update_dynamic_output_desc_y(i, broadcast_desc[i]); in SetupBroadcast()495 std::vector<GeTensorDesc> broadcast_desc; in GenerateBroadcastGraph() local523 broadcast_desc.push_back(*desc); in GenerateBroadcastGraph()532 DfGraphConvertor::SetupBroadcast(broadcast, broadcast_desc, broadcast_graph, broadcast_input); in GenerateBroadcastGraph()
118 …t(const std::shared_ptr<HcomBroadcast> &broadcast, const std::vector<GeTensorDesc> &broadcast_desc,