Home
last modified time | relevance | path

Searched refs:MklDnnDataFormatToTFDataFormat (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_aggregate_ops.cc173 tf_data_format = MklDnnDataFormatToTFDataFormat(mkl_data_format); in Compute()
Dmkl_slice_op.cc406 auto input_tf_format = MklDnnDataFormatToTFDataFormat(input_mkl_format); in ComputeMklSlice()
Dmkl_relu_op.cc637 MklDnnDataFormatToTFDataFormat(src_mkl_data_format); in Compute()
650 MklDnnDataFormatToTFDataFormat(diff_dst_mkl_data_format); in Compute()
Dmkl_concat_op.cc673 dst_dims, MklDnnDataFormatToTFDataFormat(orig_tf_format)); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h177 TensorFormat MklDnnDataFormatToTFDataFormat(MklTensorFormat format);
520 TensorFormat data_format = MklDnnDataFormatToTFDataFormat(format); in SetTfDimOrder()
1048 inline TensorFormat MklDnnDataFormatToTFDataFormat(MklTensorFormat format) { in MklDnnDataFormatToTFDataFormat() function