1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
/*******************************************************************************
* Copyright 2017-2018 Intel Corporation
*
* 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 _DNN_TYPES_HPP
#define _DNN_TYPES_HPP
#include <stdlib.h>
#include <stddef.h>
#include <string.h>
#include "common.hpp"
#include "mkldnn_types.h"
enum dir_t {
DIR_UNDEF = 0,
FLAG_DAT = 1, FLAG_WEI = 2, FLAG_BIA = 4,
FLAG_FWD = 32, FLAG_BWD = 64,
FLAG_INF = 128,
FWD_D = FLAG_FWD + FLAG_DAT,
FWD_I = FLAG_FWD + FLAG_DAT + FLAG_INF,
FWD_B = FLAG_FWD + FLAG_DAT + FLAG_BIA,
BWD_D = FLAG_BWD + FLAG_DAT,
BWD_DW = FLAG_BWD + FLAG_DAT + FLAG_WEI,
BWD_W = FLAG_BWD + FLAG_WEI,
BWD_WB = FLAG_BWD + FLAG_WEI + FLAG_BIA,
};
dir_t str2dir(const char *str);
const char *dir2str(dir_t dir);
typedef int data_kind_t;
enum {
SRC = 0, WEI, BIA, DST, ACC,
DATA, MEAN, VAR, SS,
GWEI,
DAT_TOTAL };
const char *data_kind2str(data_kind_t kind);
data_kind_t fmt2data_kind(mkldnn_memory_format_t fmt);
struct attr_t {
enum round_mode_t {
NEAREST = (int)mkldnn_round_nearest,
DOWN = (int)mkldnn_round_down,
};
struct scale_t {
enum policy_t { NONE = 0, COMMON, PER_OC, POLICY_TOTAL };
static policy_t str2policy(const char *str);
static const char *policy2str(policy_t policy);
int str2scale(const char *str, const char **end_s);
void scale2str(char *buffer, char **end_b) const;
bool is_def() const { return this->policy == NONE; }
policy_t policy = NONE;
float scale = 1.;
};
struct post_ops_t {
enum kind_t { SUM, RELU, KIND_TOTAL };
static kind_t str2kind(const char *str);
static const char *kind2str(kind_t kind);
struct entry_t {
kind_t kind;
union {
struct { float scale; } sum;
struct {
// eltwise algorithm in future
float scale, alpha, beta; // unused now
} eltwise;
};
};
post_ops_t(): len(0) {}
int from_str(const char *str, const char **end_s);
void to_str(char *buffer, char **end_b) const;
bool is_def() const { return len == 0; }
enum { capacity = 4 };
int len;
entry_t entry[4];
};
round_mode_t irmode = NEAREST;
scale_t oscale;
post_ops_t post_ops;
bool is_def() const;
};
const size_t max_attr_len = 128;
int str2attr(attr_t *attr, const char *str);
void attr2str(const attr_t *attr, char *buffer);
mkldnn_memory_format_t get_default_format(int ndims, data_kind_t kind);
mkldnn_primitive_attr_t create_mkldnn_attr(const attr_t &attr, int scale_cnt,
int scale_mask, const float *scales);
inline mkldnn_primitive_attr_t create_mkldnn_attr(const attr_t &attr,
int scale_cnt, const float *scales)
{ return create_mkldnn_attr(attr, scale_cnt, -1, scales); }
#endif
|
