Searched refs:key_creator (Results 1 – 12 of 12) sorted by relevance
334 FactoryKeyCreator key_creator; in CreateKey() local335 key_creator.AddAsKey(prefix); in CreateKey()336 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.src_dims); in CreateKey()337 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.weight_dims); in CreateKey()338 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.bias_dims); in CreateKey()339 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.dst_dims); in CreateKey()340 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.dtypes); in CreateKey()341 key_creator.AddAsKey(mkldnn_matmul_fwd_dims.weight_format); in CreateKey()349 key_creator.AddAsKey(post_op_param.name); in CreateKey()350 key_creator.AddAsKey(post_op_param.param[0]); in CreateKey()[all …]
187 FactoryKeyCreator key_creator; in CreateKey() local188 key_creator.AddAsKey(prefix); in CreateKey()189 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()190 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()191 key_creator.AddAsKey(fwdParams.filter_dims); in CreateKey()192 key_creator.AddAsKey(fwdParams.strides); in CreateKey()193 key_creator.AddAsKey(fwdParams.padding_left); in CreateKey()194 key_creator.AddAsKey(fwdParams.padding_right); in CreateKey()195 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.alg_kind)); in CreateKey()196 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.prop_kind)); in CreateKey()[all …]
274 FactoryKeyCreator key_creator; in CreateKey() local288 key_creator.AddAsKey(prefix); in CreateKey()289 key_creator.AddAsKey(static_cast<int>(from_desc.data_type)); in CreateKey()290 key_creator.AddAsKey(from_dims); in CreateKey()291 key_creator.AddAsKey(from_strides_outer_blocks); in CreateKey()292 key_creator.AddAsKey(static_cast<int>(to_desc.data_type)); in CreateKey()293 key_creator.AddAsKey(to_dims); in CreateKey()294 key_creator.AddAsKey(to_strides_outer_blocks); in CreateKey()295 key_creator.AddAsKey(sliceParams.begin_dims); in CreateKey()296 key_creator.AddAsKey(sliceParams.size_dims); in CreateKey()[all …]
276 FactoryKeyCreator key_creator; in CreateKey() local277 key_creator.AddAsKey(prefix); in CreateKey()278 key_creator.AddAsKey(convBwdInputDims.diff_src_dims); in CreateKey()279 key_creator.AddAsKey(convBwdInputDims.filter_dims); in CreateKey()280 key_creator.AddAsKey(convBwdInputDims.diff_dst_dims); in CreateKey()281 key_creator.AddAsKey(convBwdInputDims.strides); in CreateKey()282 key_creator.AddAsKey(convBwdInputDims.dilations); in CreateKey()283 key_creator.AddAsKey(convBwdInputDims.padding_left); in CreateKey()284 key_creator.AddAsKey(convBwdInputDims.padding_right); in CreateKey()286 key_creator.AddAsKey(convBwdInputDims.tf_fmt); in CreateKey()[all …]
171 FactoryKeyCreator key_creator; in CreateKey() local172 key_creator.AddAsKey(prefix); in CreateKey()173 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()174 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.src_fmt)); in CreateKey()175 key_creator.AddAsKey<int>(fwdParams.axis); in CreateKey()177 return key_creator.GetKey(); in CreateKey()
330 FactoryKeyCreator key_creator; in CreateKey() local331 key_creator.AddAsKey(prefix); in CreateKey()332 key_creator.AddAsKey(convBwdFilterDims.src_dims); in CreateKey()333 key_creator.AddAsKey(convBwdFilterDims.diff_filter_dims); in CreateKey()334 key_creator.AddAsKey(convBwdFilterDims.diff_bias_dims); in CreateKey()335 key_creator.AddAsKey(convBwdFilterDims.diff_dst_dims); in CreateKey()336 key_creator.AddAsKey(convBwdFilterDims.strides); in CreateKey()337 key_creator.AddAsKey(convBwdFilterDims.dilations); in CreateKey()338 key_creator.AddAsKey(convBwdFilterDims.padding_left); in CreateKey()339 key_creator.AddAsKey(convBwdFilterDims.padding_right); in CreateKey()[all …]
353 FactoryKeyCreator key_creator; in CreateKey() local354 key_creator.AddAsKey(prefix); in CreateKey()355 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()356 key_creator.AddAsKey<int>(fwdParams.depth); in CreateKey()357 key_creator.AddAsKey<float>(fwdParams.eps); in CreateKey()358 key_creator.AddAsKey<bool>(fwdParams.training); in CreateKey()359 key_creator.AddAsKey<FusedBNActivationMode>(fwdParams.activation_mode); in CreateKey()360 key_creator.AddAsKey(typeid(T).name()); in CreateKey()361 key_creator.AddAsKey(typeid(U).name()); in CreateKey()362 return key_creator.GetKey(); in CreateKey()[all …]
195 FactoryKeyCreator key_creator; in CreateKey() local196 key_creator.AddAsKey(prefix); in CreateKey()197 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()198 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.alg_kind)); in CreateKey()199 key_creator.AddAsKey<float>(static_cast<float>(fwdParams.alpha)); in CreateKey()200 key_creator.AddAsKey<float>(static_cast<float>(fwdParams.beta)); in CreateKey()201 return key_creator.GetKey(); in CreateKey()407 FactoryKeyCreator key_creator; in CreateKey() local408 key_creator.AddAsKey(prefix); in CreateKey()409 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()[all …]
184 FactoryKeyCreator key_creator; in CreateKey() local185 key_creator.AddAsKey(MklReorderPrimitiveFactory<T>::CreateKey(from, to)); in CreateKey()189 key_creator.AddAsKey(fwdParams.post_op_params.name); in CreateKey()190 key_creator.AddAsKey(fwdParams.post_op_params.param[0]); in CreateKey()195 return key_creator.GetKey(); in CreateKey()
363 FactoryKeyCreator key_creator; in CreateKey() local364 key_creator.AddAsKey(prefix); in CreateKey()365 key_creator.AddAsKey(convFwdDims.src_dims); in CreateKey()366 key_creator.AddAsKey(convFwdDims.filter_dims); in CreateKey()367 key_creator.AddAsKey(convFwdDims.bias_dims); in CreateKey()368 key_creator.AddAsKey(convFwdDims.dst_dims); in CreateKey()369 key_creator.AddAsKey(convFwdDims.strides); in CreateKey()370 key_creator.AddAsKey(convFwdDims.dilations); in CreateKey()371 key_creator.AddAsKey(convFwdDims.padding_left); in CreateKey()372 key_creator.AddAsKey(convFwdDims.padding_right); in CreateKey()[all …]
418 FactoryKeyCreator key_creator; in CreateKey() local419 key_creator.AddAsKey(prefix); in CreateKey()421 key_creator.AddAsKey(concat_fwd_dims.src_dims[k]); in CreateKey()423 key_creator.AddAsKey(concat_fwd_dims.concat_dims); in CreateKey()424 return key_creator.GetKey(); in CreateKey()
1923 FactoryKeyCreator key_creator; in CreateKey() local1951 key_creator.AddAsKey(prefix); in CreateKey()1952 key_creator.AddAsKey(static_cast<int>(from_desc.extra.flags)); in CreateKey()1953 key_creator.AddAsKey(static_cast<int>(from_inner_nblks)); in CreateKey()1954 key_creator.AddAsKey(from_inner_blks_1); in CreateKey()1955 key_creator.AddAsKey(from_inner_idxs_1); in CreateKey()1956 key_creator.AddAsKey(static_cast<int>(from_desc.data_type)); in CreateKey()1957 key_creator.AddAsKey(from_dims); in CreateKey()1958 key_creator.AddAsKey(from_strides_outer_blocks); in CreateKey()1959 key_creator.AddAsKey(static_cast<int>(to_desc.extra.flags)); in CreateKey()[all …]