Searched refs:dst_dims_mkl_order (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_qmatmul_op.cc | 178 memory::dims dst_dims_tf_order, dst_dims_mkl_order; in Compute() local
190 dst_dims_mkl_order = {static_cast<int>(src_tf_shape.dim_size(0)), in Compute()
228 dst_dims_mkl_order); in Compute()
241 this->AllocateOutputTensor(context, *matmul_fwd_pd, dst_dims_mkl_order, in Compute()
|
| D | mkl_conv_ops.cc | 516 memory::dims dst_dims_tf_order, dst_dims_mkl_order; in Compute() local
542 &dilations, &dst_dims_tf_order, &dst_dims_mkl_order, &padding_left, in Compute()
655 dst_dims_mkl_order, strides, dilations, padding_left, padding_right, in Compute()
666 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
|