Searched refs:kPrimLogSoftmax (Results 1 – 5 of 5) sorted by relevance
71 REGISTER_PRIMITIVE_EVAL_IMPL(LogSoftmax, prim::kPrimLogSoftmax, LogSoftmaxInfer, nullptr, true);
89 REG_ADPT_DESC(LogSoftmaxV2, prim::kPrimLogSoftmax->name(), ADPT_DESC(LogSoftmaxV2))
78 {kGPUDevice, OpLevel_0, prim::kPrimLogSoftmax}, in GetExpandOps()
442 auto prim = prim::kPrimLogSoftmax; in TEST_F()
267 inline const PrimitivePtr kPrimLogSoftmax = std::make_shared<Primitive>("LogSoftmax"); variable