Searched refs:dst_dims_mkl_order (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.cc | 942 memory::dims dst_dims_tf_order, dst_dims_mkl_order; in Compute() local
966 &dilations, &dst_dims_tf_order, &dst_dims_mkl_order, &padding_left, in Compute()
1074 dst_dims_mkl_order, strides, dilations, padding_left, padding_right); in Compute()
1085 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
|