Searched refs:DATA_FORMAT_NCHW (Results 1 – 2 of 2) sorted by relevance
71 DATA_FORMAT_NCHW = 'NCHW' variable110 if data_format not in (DATA_FORMAT_NCHW, DATA_FORMAT_NHWC):263 if data_format not in (DATA_FORMAT_NCHW, DATA_FORMAT_NHWC):282 if data_format == DATA_FORMAT_NCHW:599 if data_format not in (DATA_FORMAT_NCHW, DATA_FORMAT_NHWC):615 axis = 1 if data_format == DATA_FORMAT_NCHW else -1689 if data_format == DATA_FORMAT_NCHW:773 if data_format == DATA_FORMAT_NCHW:780 if data_format == DATA_FORMAT_NCHW:823 if data_format == DATA_FORMAT_NCHW:[all …]
37 DATA_FORMAT_NCHW = 'NCHW' variable99 if data_format not in (DATA_FORMAT_NCHW, DATA_FORMAT_NHWC):104 if data_format == DATA_FORMAT_NCHW: