Searched refs:kPrimBpropCut (Results 1 – 4 of 4) sorted by relevance
48 prim::kPrimMakeTuple, prim::kPrimBpropCut};56 … prim::kPrimBpropCut, prim::kPrimSwitchLayer}; in GetMsNonlinearOps()
640 if (prim->name() == prim::kPrimBpropCut->name()) { in IsCut()
161 {prim::kPrimBpropCut, {InferImplBpropCut, nullptr, true}}, in GetPrimitiveToEvalImplMap()
378 inline const PrimitivePtr kPrimBpropCut = std::make_shared<Primitive>("bprop_cut"); variable