Searched refs:in_channel_index (Results 1 – 1 of 1) sorted by relevance
599 int filter_x_index, filter_y_index, in_channel_index, out_channel_index; in ConvolutionDimensionsFromInputs() local603 in_channel_index = 2; in ConvolutionDimensionsFromInputs()609 in_channel_index = 1; in ConvolutionDimensionsFromInputs()627 if (iz != 1 && filter_shape.dim(in_channel_index).size() != 1) { in ConvolutionDimensionsFromInputs()628 CHECK_EQ(iz, filter_shape.dim(in_channel_index).size()); in ConvolutionDimensionsFromInputs()630 iz = std::max<int64>(iz, filter_shape.dim(in_channel_index).size()); in ConvolutionDimensionsFromInputs()