Searched refs:MKL_TENSOR_FORMAT_NHWC (Results 1 – 3 of 3) sorted by relevance
97 #define MKL_TENSOR_FORMAT_NHWC MklTensorFormat::FORMAT_NHWC macro119 #define TENSOR_FORMAT_NHWC MKL_TENSOR_FORMAT_NHWC
47 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, MKL_TENSOR_FORMAT_NHWC); in TEST()
72 mkl_shape.SetTfLayout(4, {1, 2, 2, 2}, MKL_TENSOR_FORMAT_NHWC); in CreateMklInput()