You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
2.0 KiB
58 lines
2.0 KiB
/**
|
|
* Copyright 2019 Huawei Technologies Co., Ltd
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#ifndef PREDICT_COMMON_COMMON_H_
|
|
#define PREDICT_COMMON_COMMON_H_
|
|
|
|
#include <string>
|
|
#include "schema/inner/ms_generated.h"
|
|
|
|
namespace mindspore {
|
|
namespace predict {
|
|
enum NCHW_SHAPE { NCHW_N = 0, NCHW_C = 1, NCHW_H = 2, NCHW_W = 3 };
|
|
enum NHWC_SHAPE { NHWC_N = 0, NHWC_H = 1, NHWC_W = 2, NHWC_C = 3 };
|
|
enum HWCK_SHAPE { HWCK_H = 0, HWCK_W = 1, HWCK_C = 2, HWCK_K = 3 };
|
|
enum KCHW_SHAPE { KCHW_K = 0, KCHW_C = 1, KCHW_H = 2, KCHW_W = 3 };
|
|
enum CHW_SHAPE { CHW_C = 0, CHW_H = 1, CHW_W = 2 };
|
|
enum HWC_SHAPE { HWC_H = 0, HWC_W = 1, HWC_C = 2 };
|
|
|
|
static constexpr int TENSOR_MAX_REFCOUNT = 999;
|
|
|
|
static const char *DELIM_COLON = ":";
|
|
static const char *DELIM_COMMA = ",";
|
|
static const char *DELIM_SLASH = "/";
|
|
static const char *DELIM_DOUBLE_BACKSLASH = "\\";
|
|
|
|
// quantization relative
|
|
static const char QUANTIZED_UINT8[] = "QUANTIZED_UINT8";
|
|
static const char QUANTIZED_INT8[] = "QUANTIZED_INT8";
|
|
static const char QUANTIZED_INT16[] = "QUANTIZED_INT16";
|
|
static const char QUANTIZED_UINT16[] = "QUANTIZED_UINT16";
|
|
static const char QUANTIZED_FLOAT16[] = "FLOAT16";
|
|
static const char QUANTIZED_FLOAT32[] = "FLOAT32";
|
|
static const char QUANTIZATION_TYPE_DYNAMIC[] = "DYNAMIC";
|
|
static const char QUANTIZATION_TYPE_STATIC[] = "STATIC";
|
|
static const char CALIB_NORM[] = "NORM";
|
|
|
|
// dims
|
|
static const int32_t DIM_DEFAULT_SIZE = 4;
|
|
|
|
static const Format DEFAULT_FORMAT = Format_NCHW;
|
|
} // namespace predict
|
|
} // namespace mindspore
|
|
|
|
#endif // PREDICT_COMMON_COMMON_H_
|