Searched defs:target_param_layout (Results 1 – 1 of 1) sorted by relevance
1036 const std::shared_ptr<TensorLayout> &target_param_layout, in ReplaceAdaSumStridedSliceValue()1050 RankList GetRankListByLayout(const std::shared_ptr<TensorLayout> &target_param_layout) { in GetRankListByLayout()1120 …dleAdaSumReshape(const CNodePtr &reshape_cnode, const std::shared_ptr<TensorLayout> &target_param_… in HandleAdaSumReshape()1184 …asumSlice(const AnfNodePtr &stridedslice_node1, const std::shared_ptr<TensorLayout> &target_param_… in HandleAdasumSlice()1293 auto target_param_layout = (*adasum_param_tensor_layout_map)[target_param]; in HandleAdaSum() local1381 auto target_param_layout = (*adasum_param_tensor_layout_map)[target_param]; in HandleMirrorInAdaSum() local