Searched refs:kExpandDimsOpName (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 70 Register(kExpandDimsOpName, {1}); in ConstInputToAttrInfoRegistry()
|
| D | helper.cc | 267 static std::unordered_set<std::string> nop_nodes = {prim::kPrimReshape->name(), kExpandDimsOpName, in IsNopNode()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 96 constexpr auto kExpandDimsOpName = "ExpandDims"; variable
685 kCastOpName, kExpandDimsOpName, kReshapeOpName, kEmbeddingLookupOpName, kReduceMinOpName,
689 …kCastOpName, kExpandDimsOpName, kReshapeOpName, kEmbeddingLookupOpName, kTransposeOpName, k…
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | sparse_softmax_cross_entropy_with_logits_unify_mindir.cc | 210 auto expand_dims_primitive = std::make_shared<Primitive>(kExpandDimsOpName); in CreateExpandDims()
233 auto expand_dims_primitive = std::make_shared<Primitive>(kExpandDimsOpName); in CreateExpandDimsPynative()
|