Searched refs:tf_fmt (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_grad_filter_ops.cc | 438 auto tf_fmt = is_conv2d in Compute() local
445 : memory::desc(fwd_src_dims, MklDnnType<T>(), tf_fmt); in Compute()
453 : memory::desc(diff_dst_dims, MklDnnType<T>(), tf_fmt); in Compute()
562 if (GetOutputFormat(tf_fmt) != in Compute()
565 diff_filter.SetUsrMem(bwd_output_dims, GetOutputFormat(tf_fmt), in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 377 auto tf_fmt = is_conv2d in Compute() local
398 : memory::desc(diff_dst_dims, MklDnnType<T>(), tf_fmt); in Compute()
423 auto bwd_diff_src_format = GetOutputFormat(tf_fmt); in Compute()
|
| D | mkl_conv_ops.cc | 1020 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute() local
1032 : memory::desc(src_dims, MklDnnType<Tinput>(), tf_fmt); in Compute()
1085 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
|