Searched refs:key_creator (Results 1 – 8 of 8) sorted by relevance
186 FactoryKeyCreator key_creator; in CreateKey() local187 key_creator.AddAsKey(prefix); in CreateKey()188 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()189 key_creator.AddAsKey(fwdParams.dst_dims); in CreateKey()190 key_creator.AddAsKey(fwdParams.filter_dims); in CreateKey()191 key_creator.AddAsKey(fwdParams.strides); in CreateKey()192 key_creator.AddAsKey(fwdParams.padding_left); in CreateKey()193 key_creator.AddAsKey(fwdParams.padding_right); in CreateKey()194 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.alg_kind)); in CreateKey()195 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.prop_kind)); in CreateKey()[all …]
266 FactoryKeyCreator key_creator; in CreateKey() local279 key_creator.AddAsKey(prefix); in CreateKey()280 key_creator.AddAsKey(static_cast<int>(from_desc.format)); in CreateKey()281 key_creator.AddAsKey(static_cast<int>(from_desc.data_type)); in CreateKey()282 key_creator.AddAsKey(from_dims); in CreateKey()283 key_creator.AddAsKey(from_strides); in CreateKey()284 key_creator.AddAsKey(static_cast<int>(to_desc.format)); in CreateKey()285 key_creator.AddAsKey(static_cast<int>(to_desc.data_type)); in CreateKey()286 key_creator.AddAsKey(to_dims); in CreateKey()287 key_creator.AddAsKey(to_strides); in CreateKey()[all …]
274 FactoryKeyCreator key_creator; in CreateKey() local275 key_creator.AddAsKey(prefix); in CreateKey()276 key_creator.AddAsKey(convBwdInputDims.diff_src_dims); in CreateKey()277 key_creator.AddAsKey(convBwdInputDims.filter_dims); in CreateKey()278 key_creator.AddAsKey(convBwdInputDims.diff_dst_dims); in CreateKey()279 key_creator.AddAsKey(convBwdInputDims.strides); in CreateKey()280 key_creator.AddAsKey(convBwdInputDims.dilations); in CreateKey()281 key_creator.AddAsKey(convBwdInputDims.padding_left); in CreateKey()282 key_creator.AddAsKey(convBwdInputDims.padding_right); in CreateKey()283 return key_creator.GetKey(); in CreateKey()
334 FactoryKeyCreator key_creator; in CreateKey() local335 key_creator.AddAsKey(prefix); in CreateKey()336 key_creator.AddAsKey(convBwdFilterDims.src_dims); in CreateKey()337 key_creator.AddAsKey(convBwdFilterDims.diff_filter_dims); in CreateKey()338 key_creator.AddAsKey(convBwdFilterDims.diff_bias_dims); in CreateKey()339 key_creator.AddAsKey(convBwdFilterDims.diff_dst_dims); in CreateKey()340 key_creator.AddAsKey(convBwdFilterDims.strides); in CreateKey()341 key_creator.AddAsKey(convBwdFilterDims.dilations); in CreateKey()342 key_creator.AddAsKey(convBwdFilterDims.padding_left); in CreateKey()343 key_creator.AddAsKey(convBwdFilterDims.padding_right); in CreateKey()[all …]
200 FactoryKeyCreator key_creator; in CreateKey() local201 key_creator.AddAsKey(prefix); in CreateKey()202 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()203 key_creator.AddAsKey<int>(static_cast<int>(fwdParams.alg_kind)); in CreateKey()204 key_creator.AddAsKey<float>(static_cast<float>(fwdParams.alpha)); in CreateKey()205 key_creator.AddAsKey<float>(static_cast<float>(fwdParams.beta)); in CreateKey()206 key_creator.AddAsKey<int>(static_cast<int>(src_fmt)); in CreateKey()207 return key_creator.GetKey(); in CreateKey()419 FactoryKeyCreator key_creator; in CreateKey() local420 key_creator.AddAsKey(prefix); in CreateKey()[all …]
250 FactoryKeyCreator key_creator; in CreateKey() local251 key_creator.AddAsKey(prefix); in CreateKey()252 key_creator.AddAsKey(fwdParams.src_dims); in CreateKey()253 key_creator.AddAsKey<int>(fwdParams.depth); in CreateKey()254 key_creator.AddAsKey<float>(fwdParams.eps); in CreateKey()255 key_creator.AddAsKey<bool>(fwdParams.training); in CreateKey()256 return key_creator.GetKey(); in CreateKey()473 FactoryKeyCreator key_creator; in CreateKey() local474 key_creator.AddAsKey(prefix); in CreateKey()475 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()[all …]
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 …]
2319 FactoryKeyCreator key_creator; in CreateKey() local2332 key_creator.AddAsKey(prefix); in CreateKey()2333 key_creator.AddAsKey(static_cast<int>(from_desc.format)); in CreateKey()2334 key_creator.AddAsKey(static_cast<int>(from_desc.data_type)); in CreateKey()2335 key_creator.AddAsKey(from_dims); in CreateKey()2336 key_creator.AddAsKey(from_strides); in CreateKey()2337 key_creator.AddAsKey(static_cast<int>(to_desc.format)); in CreateKey()2338 key_creator.AddAsKey(static_cast<int>(to_desc.data_type)); in CreateKey()2339 key_creator.AddAsKey(to_dims); in CreateKey()2340 key_creator.AddAsKey(to_strides); in CreateKey()[all …]