Searched refs:LshProjectionType (Results 1 – 16 of 16) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | lsh_projection.cc | 22 void LshProjection::Init(const LshProjectionType &type) { set_type(type); } in Init()
24 void LshProjection::set_type(const LshProjectionType &type) { in set_type()
29 LshProjectionType LshProjection::get_type() const { return LshProjectionType(GetValue<int64_t>(GetA… in get_type()
52 switch ((int64_t)LshProjectionType(GetValue<int64_t>(primitive->GetAttr(kType)))) { in LshProjectionInfer()
53 case (int64_t)LshProjectionType::SPARSE: in LshProjectionInfer()
56 case (int64_t)LshProjectionType::DENSE: in LshProjectionInfer()
|
| D | lsh_projection.h | 44 void Init(const LshProjectionType &type);
49 void set_type(const LshProjectionType &type);
54 LshProjectionType get_type() const;
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_lsh_projection.cc | 37 lsh_projection->Init(LshProjectionType::SPARSE); in TEST_F()
38 EXPECT_EQ(lsh_projection->get_type(), LshProjectionType::SPARSE); in TEST_F()
66 lsh_projection->Init(LshProjectionType::DENSE); in TEST_F()
67 EXPECT_EQ(lsh_projection->get_type(), LshProjectionType::DENSE); in TEST_F()
95 lsh_projection->Init(LshProjectionType::DENSE); in TEST_F()
96 EXPECT_EQ(lsh_projection->get_type(), LshProjectionType::DENSE); in TEST_F()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_lsh_projection_parser.cc | 41 prim->set_type(mindspore::LshProjectionType::SPARSE); in Parse()
44 prim->set_type(mindspore::LshProjectionType::DENSE); in Parse()
47 prim->set_type(mindspore::LshProjectionType::UNKNOWN); in Parse()
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_types_generated.h | 507 enum LshProjectionType : int8_t { enum
515 inline const LshProjectionType (&EnumValuesLshProjectionType())[3] { in EnumValuesLshProjectionType()
516 static const LshProjectionType values[] = { in EnumValuesLshProjectionType()
534 inline const char *EnumNameLshProjectionType(LshProjectionType e) { in EnumNameLshProjectionType()
|
| D | ops_types.fbs | 126 enum LshProjectionType : byte {
|
| D | ops_v0.fbs | 129 enum LshProjectionType : byte {
1054 type : LshProjectionType;
|
| D | ops.fbs | 651 type: LshProjectionType;
|
| D | ops_v0_generated.h | 1120 enum LshProjectionType : int8_t { enum
1128 inline const LshProjectionType (&EnumValuesLshProjectionType())[3] { in EnumValuesLshProjectionType()
1129 static const LshProjectionType values[] = { in EnumValuesLshProjectionType()
1147 inline const char *EnumNameLshProjectionType(LshProjectionType e) { in EnumNameLshProjectionType()
10857 mindspore::schema::v0::LshProjectionType type() const { in type()
10858 return static_cast<mindspore::schema::v0::LshProjectionType>(GetField<int8_t>(VT_TYPE, 0)); in type()
10871 void add_type(mindspore::schema::v0::LshProjectionType type) { in add_type()
10887 …mindspore::schema::v0::LshProjectionType type = mindspore::schema::v0::LshProjectionType_UNKNOWN) {
|
| D | ops_generated.h | 6714 mindspore::schema::LshProjectionType type() const {
6715 return static_cast<mindspore::schema::LshProjectionType>(GetField<int8_t>(VT_TYPE, 0));
6728 void add_type(mindspore::schema::LshProjectionType type) {
6744 mindspore::schema::LshProjectionType type = mindspore::schema::LshProjectionType_UNKNOWN) {
|
| /third_party/mindspore/mindspore/lite/mindir/include/ |
| D | mindir_types.h | 192 enum LshProjectionType : int8_t { enum
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | ops_types_generated.h | 516 enum LshProjectionType : int8_t { enum
524 inline const LshProjectionType (&EnumValuesLshProjectionType())[3] { in EnumValuesLshProjectionType()
525 static const LshProjectionType values[] = { in EnumValuesLshProjectionType()
543 inline const char *EnumNameLshProjectionType(LshProjectionType e) { in EnumNameLshProjectionType()
|
| D | ops_v0_generated.h | 1738 enum LshProjectionType : int8_t { enum
1746 inline const LshProjectionType (&EnumValuesLshProjectionType())[3] { in EnumValuesLshProjectionType()
1747 static const LshProjectionType values[] = { in EnumValuesLshProjectionType()
1765 inline const char *EnumNameLshProjectionType(LshProjectionType e) { in EnumNameLshProjectionType()
15237 mindspore::schema::v0::LshProjectionType type = mindspore::schema::v0::LshProjectionType_UNKNOWN;
15249 mindspore::schema::v0::LshProjectionType type() const { in type()
15250 return static_cast<mindspore::schema::v0::LshProjectionType>(GetField<int8_t>(VT_TYPE, 0)); in type()
15252 bool mutate_type(mindspore::schema::v0::LshProjectionType _type) { in mutate_type()
15269 void add_type(mindspore::schema::v0::LshProjectionType type) { in add_type()
15285 …mindspore::schema::v0::LshProjectionType type = mindspore::schema::v0::LshProjectionType_UNKNOWN) {
|
| D | ops_generated.h | 10738 mindspore::schema::LshProjectionType type = mindspore::schema::LshProjectionType_UNKNOWN;
10750 mindspore::schema::LshProjectionType type() const {
10751 return static_cast<mindspore::schema::LshProjectionType>(GetField<int8_t>(VT_TYPE, 0));
10753 bool mutate_type(mindspore::schema::LshProjectionType _type) {
10770 void add_type(mindspore::schema::LshProjectionType type) {
10786 mindspore::schema::LshProjectionType type = mindspore::schema::LshProjectionType_UNKNOWN) {
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 116 enum class LshProjectionType : int64_t { UNKNOWN = 0, SPARSE = 1, DENSE = 2 }; enum
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_def.cc | 651 OP_ATTR_ENUM(type, LshProjectionType)
|