Searched refs:kPrimOnesLike (Results 1 – 5 of 5) sorted by relevance
32 OnesLike() : PrimitiveC(prim::kPrimOnesLike->name()) {} in OnesLike()
52 REGISTER_PRIMITIVE_EVAL_IMPL(OnesLike, prim::kPrimOnesLike, OnesLikeInfer, nullptr, true);
96 {kGPUDevice, OpLevel_0, prim::kPrimOnesLike}, in GetExpandOps()
328 {prim::kPrimOnesLike->name(), OnesLike<T>}, in LaunchKernel()
377 inline const PrimitivePtr kPrimOnesLike = std::make_shared<Primitive>(kOnesLike); variable