Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dmkl_types.h93 #define MKL_TENSOR_FORMAT_NC MklTensorFormat::FORMAT_NC
Dmkl_util.h167 FORMAT_NC = 5, enumerator
208 } else if (format == MklTensorFormat::FORMAT_NC) {
1015 if (format == MklTensorFormat::FORMAT_NC) return memory::format_tag::nc; in MklTensorFormatToMklDnnDataFormat()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_softmax_op.cc239 layout_type = MklTensorFormat::FORMAT_NC; in Compute()
Dmkl_matmul_op_fused.cc163 MklTensorFormatToMklDnnDataFormat(MklTensorFormat::FORMAT_NC); in Compute()
Dmkl_qmatmul_op.cc203 auto input_output_fmt_mkldnn = MklTensorFormat::FORMAT_NC; in Compute()