Searched refs:nchw_input (Results 1 – 1 of 1) sorted by relevance
127 Layout nchw_input, nchw_filter, nchw_output; in XlaConvLayoutsToStreamExecutorLayouts() local128 std::tie(nchw_input, nchw_filter, nchw_output) = in XlaConvLayoutsToStreamExecutorLayouts()142 if (LayoutUtil::Equal(input, nchw_input)) { in XlaConvLayoutsToStreamExecutorLayouts()