Searched refs:new_convolution (Results 1 – 1 of 1) sorted by relevance
240 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleBatchGroupCount() local251 expanded_filter_shape, HloOpcode::kSelect, filter_mask, new_convolution, in HandleBatchGroupCount()255 auto reduce_window_shape = new_convolution->shape(); in HandleBatchGroupCount()286 for (int64 i = 0; i < new_convolution->shape().dimensions_size(); ++i) { in HandleBatchGroupCount()380 auto new_convolution = HloInstruction::CreateConvolve( in HandleConvolution() local385 convolution, std::move(new_convolution))); in HandleConvolution()442 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleConvolution() local449 ShapeUtil::DeleteDimension(new_spatial_dim, new_convolution->shape()); in HandleConvolution()451 reshaped_convolution_shape, new_convolution); in HandleConvolution()520 auto new_convolution = add(HloInstruction::CreateConvolve( in HandleConvolution() local[all …]