Searched refs:user_data_fmt (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 182 memory::format_tag user_data_fmt; in Setup() local
184 user_data_fmt = in Setup()
190 user_data_fmt = memory::format_tag::any; in Setup()
193 {convBwdInputDims.diff_dst_dims}, MklDnnType<T>(), user_data_fmt)); in Setup()
195 {convBwdInputDims.diff_src_dims}, MklDnnType<T>(), user_data_fmt)); in Setup()
|
| D | mkl_conv_grad_filter_ops.cc | 205 memory::format_tag user_data_fmt; in Setup() local
207 user_data_fmt = in Setup()
213 user_data_fmt = memory::format_tag::any; in Setup()
216 MklDnnType<T>(), user_data_fmt)); in Setup()
219 {convBwdFilterDims.diff_dst_dims}, MklDnnType<T>(), user_data_fmt)); in Setup()
|
| D | mkl_conv_ops.cc | 210 memory::format_tag user_data_fmt; in Setup() local
212 user_data_fmt = MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()
215 user_data_fmt = memory::format_tag::any; in Setup()
218 {convFwdDims.src_dims}, MklDnnType<Tinput>(), user_data_fmt)); in Setup()
225 {convFwdDims.dst_dims}, MklDnnType<Toutput>(), user_data_fmt)); in Setup()
|