Searched refs:nhwc_input (Results 1 – 2 of 2) sorted by relevance
177 const bool nhwc_input = (flags & XNN_FLAG_INPUT_NHWC) != 0; in xnn_create_convolution2d_nchw_f32() local178 if (is_1x1 && !any_padding && !nhwc_input && groups == 1) { in xnn_create_convolution2d_nchw_f32()182 nhwc_input && groups == 1) in xnn_create_convolution2d_nchw_f32()187 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()193 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()199 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()205 !nhwc_input && group_input_channels == 1 && group_output_channels == 1) in xnn_create_convolution2d_nchw_f32()
134 Layout nhwc_input, nhwc_filter, nhwc_output; in XlaConvLayoutsToStreamExecutorLayouts() local135 std::tie(nhwc_input, nhwc_filter, nhwc_output) = in XlaConvLayoutsToStreamExecutorLayouts()144 } else if (LayoutUtil::Equal(input, nhwc_input)) { in XlaConvLayoutsToStreamExecutorLayouts()