Searched refs:output_axis (Results 1 – 1 of 1) sorted by relevance
1850 const int output_axis = is_conv2d_backprop_input ? 1 : 0; in ConvertConv2DHelper() local1851 const int noutput = weights.shape_.d[output_axis] * num_groups; in ConvertConv2DHelper()