| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | pooling_parameter.h | 21 typedef enum PoolMode { PoolMode_No, PoolMode_MaxPool, PoolMode_AvgPool } PoolMode; enum
28 PoolMode pool_mode_;
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | pooling_grad.h | 35 …void Init(const PoolMode &pool_mode, const std::vector<int64_t> &window, const std::vector<int64_t…
38 void set_pool_mode(const PoolMode &pool_mode);
50 PoolMode get_pool_mode() const;
|
| D | pooling_grad.cc | 22 void PoolingGrad::Init(const PoolMode &pool_mode, const std::vector<int64_t> &window, in Init()
36 void PoolingGrad::set_pool_mode(const PoolMode &pool_mode) { in set_pool_mode()
41 PoolMode PoolingGrad::get_pool_mode() const { in get_pool_mode()
43 return PoolMode(GetValue<int64_t>(value_ptr)); in get_pool_mode()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/ |
| D | pooling_fp16_grad.h | 25 using mindspore::schema::PoolMode;
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | pooling_grad.h | 25 using mindspore::schema::PoolMode;
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_types_generated.h | 342 enum PoolMode : int8_t { enum
349 inline const PoolMode (&EnumValuesPoolMode())[2] { in EnumValuesPoolMode()
350 static const PoolMode values[] = { in EnumValuesPoolMode()
366 inline const char *EnumNamePoolMode(PoolMode e) { in EnumNamePoolMode()
|
| D | ops_types.fbs | 96 enum PoolMode : byte {
|
| D | ops_v0.fbs | 97 enum PoolMode : byte {
304 poolingMode: PoolMode;
350 poolingMode: PoolMode;
|
| D | ops_v0_generated.h | 948 enum PoolMode : int8_t { enum
955 inline const PoolMode (&EnumValuesPoolMode())[2] { in EnumValuesPoolMode()
956 static const PoolMode values[] = { in EnumValuesPoolMode()
972 inline const char *EnumNamePoolMode(PoolMode e) { in EnumNamePoolMode()
3067 mindspore::schema::v0::PoolMode poolingMode() const { in poolingMode()
3068 return static_cast<mindspore::schema::v0::PoolMode>(GetField<int8_t>(VT_POOLINGMODE, 0)); in poolingMode()
3129 void add_poolingMode(mindspore::schema::v0::PoolMode poolingMode) { in add_poolingMode()
3179 mindspore::schema::v0::PoolMode poolingMode = mindspore::schema::v0::PoolMode_MAX_POOLING,
3543 mindspore::schema::v0::PoolMode poolingMode() const { in poolingMode()
3544 return static_cast<mindspore::schema::v0::PoolMode>(GetField<int8_t>(VT_POOLINGMODE, 0)); in poolingMode()
[all …]
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_x86/src/ |
| D | net.c | 90 const PoolingParameter pooling_parameter = {{ "", false, 92, g_thread_num, 0}, (PoolMode)(1), (Roun… in Inference()
119 const PoolingParameter pooling_parameter = {{ "", false, 92, g_thread_num, 0}, (PoolMode)(1), (Roun… in Inference()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_stream_utils.h | 38 std::ostream &operator<<(std::ostream &code, PoolMode pool_mode);
|
| D | nnacl_stream_utils.cc | 50 std::ostream &operator<<(std::ostream &code, PoolMode pool_mode) { in operator <<()
|
| /third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/ |
| D | pooling_tests.cc | 25 OpParameter *CreateParameter(PoolMode pool_mode, int window_h, int window_w, int stride_h, int stri… in CreateParameter()
|
| /third_party/mindspore/mindspore/lite/mindir/include/ |
| D | mindir_types.h | 162 enum PoolMode : int8_t { enum
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | ops_types_generated.h | 351 enum PoolMode : int8_t { enum
358 inline const PoolMode (&EnumValuesPoolMode())[2] { in EnumValuesPoolMode()
359 static const PoolMode values[] = { in EnumValuesPoolMode()
375 inline const char *EnumNamePoolMode(PoolMode e) { in EnumNamePoolMode()
|
| D | ops_v0_generated.h | 1566 enum PoolMode : int8_t { enum
1573 inline const PoolMode (&EnumValuesPoolMode())[2] { in EnumValuesPoolMode()
1574 static const PoolMode values[] = { in EnumValuesPoolMode()
1590 inline const char *EnumNamePoolMode(PoolMode e) { in EnumNamePoolMode()
4356 mindspore::schema::v0::PoolMode poolingMode = mindspore::schema::v0::PoolMode_MAX_POOLING;
4397 mindspore::schema::v0::PoolMode poolingMode() const { in poolingMode()
4398 return static_cast<mindspore::schema::v0::PoolMode>(GetField<int8_t>(VT_POOLINGMODE, 0)); in poolingMode()
4400 bool mutate_poolingMode(mindspore::schema::v0::PoolMode _poolingMode) { in mutate_poolingMode()
4498 void add_poolingMode(mindspore::schema::v0::PoolMode poolingMode) { in add_poolingMode()
4548 mindspore::schema::v0::PoolMode poolingMode = mindspore::schema::v0::PoolMode_MAX_POOLING,
[all …]
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.h | 109 enum PoolMode : int64_t { enum
|