Searched refs:kPrimBCEWithLogitsLoss (Results 1 – 3 of 3) sorted by relevance
35 NewValueNode(std::make_shared<Primitive>(prim::kPrimBCEWithLogitsLoss->name()))}; in AddReduceNode()70 return VectorRef({prim::kPrimBCEWithLogitsLoss, Xs}); in DefinePattern()
37 NewValueNode(std::make_shared<Primitive>(prim::kPrimBCEWithLogitsLoss->name()))}; in AddReduceNode()77 return VectorRef({prim::kPrimBCEWithLogitsLoss, Xs}); in DefinePattern()
392 inline const PrimitivePtr kPrimBCEWithLogitsLoss = std::make_shared<Primitive>("BCEWithLogitsLoss"); variable