Home
last modified time | relevance | path

Searched refs:kPrimLayerNorm (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dlayer_norm.cc149 REGISTER_PRIMITIVE_EVAL_IMPL(LayerNorm, prim::kPrimLayerNorm, LayerNormInfer, nullptr, true);
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/
Dnn_norm_ops_declare.cc97 REG_ADPT_DESC(LayerNorm, prim::kPrimLayerNorm->name(), ADPT_DESC(LayerNorm))
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dgraph_kernel_expander.cc76 {kGPUDevice, OpLevel_1, prim::kPrimLayerNorm}, in GetExpandOps()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dkernel_info_setter.cc374 if (kernel_name == prim::kPrimLayerNorm->name()) { in CheckSupportFormatTransform()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h348 inline const PrimitivePtr kPrimLayerNorm = std::make_shared<Primitive>(kLayerNorm); variable