Searched refs:nchw_output (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()164 if (LayoutUtil::Equal(output, nchw_output)) { in XlaConvLayoutsToStreamExecutorLayouts()