Searched refs:convert_back_shape (Results 1 – 1 of 1) sorted by relevance
304 Shape convert_back_shape = reduce_window->shape(); in HandleBatchGroupCount() local305 convert_back_shape.set_element_type(activation->shape().element_type()); in HandleBatchGroupCount()309 HloInstruction::CreateConvert(convert_back_shape, reduce_window); in HandleBatchGroupCount()