Searched refs:squeeze_prim (Results 1 – 6 of 6) sorted by relevance
27 auto squeeze_prim = primitive->value_as_Squeeze(); in PopulateSqueezeParameter() local28 if (squeeze_prim == nullptr) { in PopulateSqueezeParameter()39 auto axis = squeeze_prim->axis(); in PopulateSqueezeParameter()
95 auto squeeze_prim = utils::cast<std::shared_ptr<mindspore::ops::Squeeze>>(squeeze_primitive); in Process() local97 MS_ASSERT(squeeze_prim != nullptr); in Process()98 if (squeeze_prim->get_axis() == unsqueeze_prim->get_axis()) { in Process()
715 auto squeeze_prim = std::make_shared<ops::Squeeze>(); in CreateSqueezeNode() local716 MS_CHECK_TRUE_RET(squeeze_prim != nullptr, nullptr); in CreateSqueezeNode()720 squeeze_prim->set_axis(axis_vec); in CreateSqueezeNode()721 auto squeeze_cnode = func_graph->NewCNode(squeeze_prim, {input_node}); in CreateSqueezeNode()
723 auto squeeze_prim = std::make_shared<ops::Squeeze>(); in GetPostProcessNode() local724 MS_CHECK_TRUE_RET(squeeze_prim != nullptr, nullptr); in GetPostProcessNode()725 squeeze_prim->set_axis(std::vector<int64_t>{1}); in GetPostProcessNode()726 auto squeeze_value_node = NewValueNode(squeeze_prim); in GetPostProcessNode()
54 auto squeeze_prim = std::make_shared<ops::Squeeze>(); in NewSqueezeOpNode() local55 MS_CHECK_TRUE_MSG(squeeze_prim != nullptr, nullptr, "create squeeze failed."); in NewSqueezeOpNode()56 squeeze_prim->set_attr("axis", MakeValue(axis)); in NewSqueezeOpNode()57 ValueNodePtr value_node = NewValueNode(squeeze_prim); in NewSqueezeOpNode()
26 auto squeeze_prim = primitive->value_as_Squeeze(); in Init() local27 auto axis = squeeze_prim->axis(); in Init()