Searched refs:GetExpandedFilterMask (Results 1 – 1 of 1) sorted by relevance
163 HloInstruction* GetExpandedFilterMask( in GetExpandedFilterMask() function235 GetExpandedFilterMask(convolution->shape(), output_batch_dimension, in HandleBatchGroupCount()343 GetExpandedFilterMask(filter->shape(), kernel_input_feature_dim, in HandleConvolution()