Home
last modified time | relevance | path

Searched refs:AddAsKey (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_pooling_ops_common.h187 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()
339 key_creator.AddAsKey(prefix); in CreateKey()
[all …]
Dmkl_slice_op.cc279 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()
288 key_creator.AddAsKey(sliceParams.begin_dims); in CreateKey()
[all …]
Dmkl_conv_grad_input_ops.cc275 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()
Dmkl_conv_grad_filter_ops.cc335 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()
Dmkl_relu_op.cc201 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()
420 key_creator.AddAsKey(prefix); in CreateKey()
421 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()
422 key_creator.AddAsKey(static_cast<int>(bwdParams.alg_kind)); in CreateKey()
423 key_creator.AddAsKey(static_cast<float>(bwdParams.alpha)); in CreateKey()
[all …]
Dmkl_fused_batch_norm_op.cc251 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()
474 key_creator.AddAsKey(prefix); in CreateKey()
475 key_creator.AddAsKey(bwdParams.src_dims); in CreateKey()
476 key_creator.AddAsKey(bwdParams.diff_dst_dims); in CreateKey()
477 key_creator.AddAsKey<int>(bwdParams.depth); in CreateKey()
478 key_creator.AddAsKey<float>(bwdParams.eps); in CreateKey()
[all …]
Dmkl_conv_ops.cc364 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()
373 key_creator.AddAsKey(convFwdDims.dtypes); in CreateKey()
[all …]
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h2216 void AddAsKey(const string& str) { Append(str); } in AddAsKey() function
2218 void AddAsKey(const mkldnn::memory::dims& dims) { in AddAsKey() function
2220 AddAsKey<int>(dims[i]); in AddAsKey()
2225 void AddAsKey(const T data) { in AddAsKey() function
2332 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()
[all …]