|
|
@@ -2137,7 +2137,10 @@ void DavinciModel::GetModelAttr(vector<string> &out_shape_info) const { |
|
|
|
} |
|
|
|
|
|
|
|
void DavinciModel::SetInputDimsInfo(const vector<int64_t> &input_dims, Format &format, ShapeDescription &shape_info) { |
|
|
|
uint32_t n, c, h, w; |
|
|
|
uint32_t n; |
|
|
|
uint32_t c; |
|
|
|
uint32_t h; |
|
|
|
uint32_t w; |
|
|
|
n = format == FORMAT_NHWC ? NHWC_DIM_N : NCHW_DIM_N; |
|
|
|
c = format == FORMAT_NHWC ? NHWC_DIM_C : NCHW_DIM_C; |
|
|
|
h = format == FORMAT_NHWC ? NHWC_DIM_H : NCHW_DIM_H; |
|
|
|