Searched refs:diff_dst_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 63 memory::dims diff_dst_dims; member
72 memory::dims diff_dst_dims, memory::dims strides, in MklConvBwdInputParams()
78 diff_dst_dims(diff_dst_dims), in MklConvBwdInputParams()
193 {convBwdInputDims.diff_dst_dims}, MklDnnType<T>(), user_data_fmt)); in Setup()
280 key_creator.AddAsKey(convBwdInputDims.diff_dst_dims); in CreateKey()
370 memory::dims diff_dst_dims, fwd_src_dims, fwd_filter_dims; in Compute() local
408 conv_util.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()
414 : memory::desc(diff_dst_dims, MklDnnType<T>(), mkl_fmt_tag); in Compute()
420 fwd_src_dims, fwd_filter_dims, diff_dst_dims, strides, tf_fmt, in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 59 memory::dims diff_dst_dims; member
69 memory::dims diff_dst_dims, memory::dims strides, in MklConvBwdFilterParams()
76 diff_dst_dims(diff_dst_dims), in MklConvBwdFilterParams()
219 {convBwdFilterDims.diff_dst_dims}, MklDnnType<T>(), user_data_fmt)); in Setup()
335 key_creator.AddAsKey(convBwdFilterDims.diff_dst_dims); in CreateKey()
418 memory::dims diff_dst_dims, fwd_src_dims, fwd_filter_dims; in Compute() local
445 conv_util.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()
451 : memory::desc(diff_dst_dims, MklDnnType<T>(), mkl_fmt_tag); in Compute()
467 fwd_src_dims, fwd_filter_dims, diff_bias_dims, diff_dst_dims, strides, in Compute()
|
| D | mkl_avgpooling_op.cc | 216 memory::dims diff_dst_dims = in Compute() local
237 : memory::desc(diff_dst_dims, MklDnnType<T>(), in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 379 memory::dims diff_dst_dims; member
387 MklBatchNormBwdParams(memory::dims src_dims, memory::dims diff_dst_dims, in MklBatchNormBwdParams()
391 diff_dst_dims(diff_dst_dims), in MklBatchNormBwdParams()
613 key_creator.AddAsKey(bwdParams.diff_dst_dims); in CreateKey()
1165 memory::dims diff_dst_dims = in Compute() local
1179 : memory::desc(diff_dst_dims, MklDnnType<T>(), dnn_fmt); in Compute()
1220 MklBatchNormBwdParams bwdParams(src_dims, diff_dst_dims, depth_, epsilon_, in Compute()
|
| D | mkl_maxpooling_op.cc | 272 memory::dims diff_dst_dims = in Compute() local
294 : memory::desc(diff_dst_dims, MklDnnType<T>(), in Compute()
|
| D | mkl_relu_op.cc | 638 auto diff_dst_dims = TFShapeToMklDnnDimsInNCHW(diff_dst_tensor.shape(), in Compute() local
641 diff_dst_dims, MklDnnType<T>(), in Compute()
|