Searched refs:MKL_TENSOR_FORMAT_NCHW (Results 1 – 2 of 2) sorted by relevance
35 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, MKL_TENSOR_FORMAT_NCHW); in TEST()
94 #define MKL_TENSOR_FORMAT_NCHW MklTensorFormat::FORMAT_NCHW macro