Searched refs:new_filter (Results 1 – 1 of 1) sorted by relevance
250 auto new_filter = add(HloInstruction::CreateTernary( in HandleBatchGroupCount() local254 PrimitiveType reduce_type = new_filter->shape().element_type(); in HandleBatchGroupCount()262 Shape convert_shape = new_filter->shape(); in HandleBatchGroupCount()264 new_filter = in HandleBatchGroupCount()265 add(HloInstruction::CreateConvert(convert_shape, new_filter)); in HandleBatchGroupCount()301 reduce_window_shape, new_filter, zero_scalar, window, in HandleBatchGroupCount()376 auto new_filter = add(HloInstruction::CreateTernary( in HandleConvolution() local381 convolution->shape(), convolution->mutable_operand(0), new_filter, in HandleConvolution()