Searched refs:kPrimOneHot (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | one_hot.h | 32 OneHot() : PrimitiveC(prim::kPrimOneHot->name()) { in OneHot()
|
| D | one_hot.cc | 95 REGISTER_PRIMITIVE_EVAL_IMPL(OneHot, prim::kPrimOneHot, OneHotInfer, nullptr, true);
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | const_input_to_attr_registry.cc | 49 Register(prim::kPrimOneHot->name(), {1}); in ConstInputToAttrInfoRegistry()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | branch_culling.cc | 68 {prim::kPrimOneHot, {2}}, in InConvertWhiteList()
89 {prim::kPrimTranspose, {2}}, {prim::kPrimOneHot, {2}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | selection_ops_declare.cc | 72 REG_ADPT_DESC(OneHot, prim::kPrimOneHot->name(), ADPT_DESC(OneHot))
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 43 const auto kOneHot = prim::kPrimOneHot->name(); in GetDependsFormMap()
|
| /third_party/mindspore/tests/ut/cpp/transform/ |
| D | convert_test.cc | 497 auto prim = prim::kPrimOneHot; in TEST_F()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 362 inline const PrimitivePtr kPrimOneHot = std::make_shared<Primitive>("OneHot"); variable
|