Searched refs:is_ndhwc (Results 1 – 1 of 1) sorted by relevance
3254 const bool is_ndhwc = (data_format == "NDHWC"); // Or NCDHW 01234 - > 02341 in ConvertConv3DHelper() local3255 const int d_index = is_ndhwc ? 1 : 2; in ConvertConv3DHelper()3256 const int h_index = is_ndhwc ? 2 : 3; in ConvertConv3DHelper()3257 const int w_index = is_ndhwc ? 3 : 4; in ConvertConv3DHelper()3258 const int c_index = is_ndhwc ? 4 : 1; in ConvertConv3DHelper()3335 const bool need_transpose = is_ndhwc; in ConvertConv3DHelper()3439 const bool is_ndhwc = (data_format == "NDHWC"); in ConvertPool3D() local3440 const int c_index = is_ndhwc ? 4 : 1; in ConvertPool3D()3441 const int d_index = is_ndhwc ? 1 : 2; in ConvertPool3D()3442 const int h_index = is_ndhwc ? 2 : 3; in ConvertPool3D()[all …]