Searched refs:convert_shape (Results 1 – 1 of 1) sorted by relevance
262 Shape convert_shape = new_filter->shape(); in HandleBatchGroupCount() local263 convert_shape.set_element_type(F32); in HandleBatchGroupCount()265 add(HloInstruction::CreateConvert(convert_shape, new_filter)); in HandleBatchGroupCount()