Lines Matching refs:attr_info

89         attr_info_t attr_info;  in create()  local
91 attr_info.all_post_ops.copy_from(po); in create()
94 attr_info.with_binary = (binary_idx != -1); in create()
97 attr_info.eltwise_idx = po.find(primitive_kind::eltwise); in create()
98 attr_info.with_eltwise = (attr_info.eltwise_idx != -1); in create()
100 if (attr_info.with_eltwise) { in create()
101 auto &eltwise = po.entry_[attr_info.eltwise_idx].eltwise; in create()
102 attr_info.eltwise_alg = eltwise.alg; in create()
103 attr_info.eltwise_scale = eltwise.scale; in create()
104 attr_info.eltwise_alpha = eltwise.alpha; in create()
105 attr_info.eltwise_beta = eltwise.beta; in create()
107 attr_info.eltwise_alg = alg_kind::undef; in create()
108 attr_info.eltwise_scale = 1.0f; in create()
109 attr_info.eltwise_alpha = 1.0f; in create()
110 attr_info.eltwise_beta = 0.0f; in create()
114 attr_info.sum_idx = po.find(primitive_kind::sum); in create()
115 attr_info.sum_scale = (attr_info.sum_idx != -1 in create()
116 ? po.entry_[attr_info.sum_idx].sum.scale in create()
118 attr_info.sum_data_type = (attr_info.sum_idx != -1) in create()
119 ? po.entry_[attr_info.sum_idx].sum.dt in create()
121 attr_info.with_sum in create()
122 = (attr_info.sum_idx != -1) && (attr_info.sum_scale != 0.0f); in create()
125 attr_info.with_oscales = !attr->output_scales_.has_default_values(); in create()
128 attr_info.with_common_oscales in create()
129 = attr_info.with_oscales && (scales_mask == 0); in create()
130 attr_info.common_oscales = (attr_info.with_common_oscales in create()
134 attr_info.with_per_oc_oscales in create()
135 = attr_info.with_oscales && (scales_mask == (1 << 1)); in create()
137 attr_info.with_runtime_oscales = attr_info.with_per_oc_oscales in create()
141 attr_info.with_src0_scale = !src0_scales.has_default_values(); in create()
142 attr_info.src0_scale = *src0_scales.scales_; in create()
146 attr_info.with_src1_scale = !src1_scales.has_default_values(); in create()
147 attr_info.src1_scale = *src1_scales.scales_; in create()
152 attr_info.with_src_zpoints = !zp.has_default_values(DNNL_ARG_SRC); in create()
153 attr_info.with_dst_zpoints = !zp.has_default_values(DNNL_ARG_DST); in create()
155 attr_info.with_per_ic_src_zpoints = attr_info.with_src_zpoints in create()
157 attr_info.common_src_zpoint in create()
158 = attr_info.with_src_zpoints && zp.defined(DNNL_ARG_SRC) in create()
162 attr_info.with_per_oc_dst_zpoints = attr_info.with_dst_zpoints in create()
164 attr_info.common_dst_zpoint in create()
165 = attr_info.with_dst_zpoints && zp.defined(DNNL_ARG_DST) in create()
169 attr_info.initialized = true; in create()
170 return attr_info; in create()
283 attr_info_t attr_info; member
339 attr_info_t attr_info; member
350 attr_info_t attr_info; member
376 attr_info_t attr_info; member
559 attr_info_t attr_info; member
585 attr_info_t attr_info; member
665 attr_info_t attr_info; member
733 conf.attr_info = attr_info_t::create(&attr); in set_default_pool_conf()
794 conf.attr_info = attr_info_t::create(&attr); in set_default_conf()
1169 compute::kernel_ctx_t &kernel_ctx, const attr_info_t &attr_info) { in def_attr_info() argument
1170 assert(attr_info.initialized); in def_attr_info()
1172 kernel_ctx.define_int("WITH_POST_OP", attr_info.all_post_ops.len() > 0); in def_attr_info()
1174 kernel_ctx.define_int("WITH_ELTWISE", attr_info.with_eltwise); in def_attr_info()
1175 kernel_ctx.define_int("ELTWISE_IDX", attr_info.eltwise_idx); in def_attr_info()
1176 kernel_ctx.define_int("ELTWISE_ALG", attr_info.eltwise_alg); in def_attr_info()
1177 kernel_ctx.define_int("ELTWISE_ALPHA0", attr_info.eltwise_alpha == 0.0f); in def_attr_info()
1179 kernel_ctx.define_int("WITH_SUM", attr_info.with_sum); in def_attr_info()
1180 kernel_ctx.define_int("SUM_IDX", attr_info.sum_idx); in def_attr_info()
1181 kernel_ctx.define_int("SUM_SCALE", attr_info.sum_scale); in def_attr_info()
1182 kernel_ctx.define_int("SUM_SCALE1", attr_info.sum_scale == 1.0f); in def_attr_info()
1184 kernel_ctx.define_int("WITH_SRC0_SCALE", attr_info.with_src0_scale); in def_attr_info()
1185 kernel_ctx.define_int("WITH_SRC1_SCALE", attr_info.with_src1_scale); in def_attr_info()
1187 kernel_ctx.define_int("WITH_SCALES", attr_info.with_oscales); in def_attr_info()
1188 kernel_ctx.define_int("SCALES_PER_OC", attr_info.with_per_oc_oscales); in def_attr_info()
1189 kernel_ctx.define_int("SCALES_COMMON", attr_info.with_common_oscales); in def_attr_info()
1191 kernel_ctx.define_int("WITH_SRC_ZPOINTS", attr_info.with_src_zpoints); in def_attr_info()
1192 kernel_ctx.define_int("WITH_DST_ZPOINTS", attr_info.with_dst_zpoints); in def_attr_info()
1193 kernel_ctx.define_int("SRC_ZPOINT_COMMON", attr_info.common_src_zpoint); in def_attr_info()
1194 kernel_ctx.define_int("DST_ZPOINT_COMMON", attr_info.common_dst_zpoint); in def_attr_info()
1196 "WITH_SRC_ZPOINTS_PER_IC", attr_info.with_per_ic_src_zpoints); in def_attr_info()
1198 "WITH_DST_ZPOINTS_PER_OC", attr_info.with_per_oc_dst_zpoints); in def_attr_info()
1203 def_post_ops_cfg(kernel_ctx, attr_info.all_post_ops); in def_attr_info()