Searched refs:out_channel_index (Results 1 – 1 of 1) sorted by relevance
917 int filter_x_index, filter_y_index, in_major_channel_index, out_channel_index, in ConvolutionDimensionsFromInputs() local923 out_channel_index = 3; in ConvolutionDimensionsFromInputs()925 out_channel_index = 0; in ConvolutionDimensionsFromInputs()932 out_channel_index = 0; in ConvolutionDimensionsFromInputs()966 int64 oz = filter_shape.dim(out_channel_index).size(); in ConvolutionDimensionsFromInputs()