Searched refs:kPrimBroadcastShape (Results 1 – 4 of 4) sorted by relevance
180 FuncGraphPtr dg = Kprim(NewValueNode(prim::kPrimBroadcastShape), resourcePtr); in TEST_F()
111 {prim::kPrimBroadcastShape, {InferImplBroadCastShape, nullptr, true}}, in GetPrimitiveToEvalImplMap()
268 ASSERT_EQ(prim->name(), kPrimBroadcastShape->name()); in TEST_F()
170 inline const PrimitivePtr kPrimBroadcastShape = std::make_shared<Primitive>("broadcast_shape"); variable