Searched refs:out_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() local604 out_channel_index = 3; in ConvolutionDimensionsFromInputs()610 out_channel_index = 0; in ConvolutionDimensionsFromInputs()624 int64 oz = filter_shape.dim(out_channel_index).size(); in ConvolutionDimensionsFromInputs()