Searched refs:tf_fmt (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_grad_input_ops.cc | 65 MklTensorFormat tf_fmt; member
73 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdInputParams()
80 tf_fmt(tf_fmt), in MklConvBwdInputParams()
185 MklTensorFormatToMklDnnDataFormat(convBwdInputDims.tf_fmt); in Setup()
286 key_creator.AddAsKey(convBwdInputDims.tf_fmt); in CreateKey()
388 auto tf_fmt = is_conv2d in Compute() local
392 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
420 fwd_src_dims, fwd_filter_dims, diff_dst_dims, strides, tf_fmt, in Compute()
440 auto bwd_diff_src_format = GetOutputFormat(tf_fmt); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 61 MklTensorFormat tf_fmt; member
70 MklTensorFormat tf_fmt, bool native_format, in MklConvBwdFilterParams()
78 tf_fmt(tf_fmt), in MklConvBwdFilterParams()
208 MklTensorFormatToMklDnnDataFormat(convBwdFilterDims.tf_fmt); in Setup()
341 key_creator.AddAsKey(convBwdFilterDims.tf_fmt); in CreateKey()
433 auto tf_fmt = is_conv2d in Compute() local
436 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
468 tf_fmt, native_format, dilations, padding_left, padding_right); in Compute()
|
| D | mkl_conv_ops.cc | 67 MklTensorFormat tf_fmt; member
82 MklTensorFormat tf_fmt, bool native_format) in MklConvFwdParams()
91 tf_fmt(tf_fmt), in MklConvFwdParams()
212 user_data_fmt = MklTensorFormatToMklDnnDataFormat(convFwdDims.tf_fmt); in Setup()
375 key_creator.AddAsKey(convFwdDims.tf_fmt); in CreateKey()
596 auto tf_fmt = is_conv2d ? TFDataFormatToMklDnnDataFormat(data_format_) in Compute() local
599 auto mkl_fmt_tag = MklTensorFormatToMklDnnDataFormat(tf_fmt); in Compute()
656 tf_fmt, native_format); in Compute()
666 AllocateOutputTensor(context, *conv_fwd_pd, dst_dims_mkl_order, tf_fmt, in Compute()
|