Searched refs:kPrimDropout (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 70 {kGPUDevice, OpLevel_0, prim::kPrimDropout}, in GetExpandOps()
201 {prim::kPrimDropout, std::make_shared<DropoutExpander>()}, in GetExpander()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 162 {prim::kPrimDropout, {InferImplDropout, nullptr, true}}, in GetPrimitiveToEvalImplMap()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | redundant_op_remove_pass.cc | 383 if (!is_train_model_ && CheckPrimitiveType(node, prim::kPrimDropout)) { in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | dropout_unify_mindir.cc | 327 return VectorRef({prim::kPrimDropout, X}); in DefinePattern()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 359 inline const PrimitivePtr kPrimDropout = std::make_shared<Primitive>(kDropout); variable
|