Searched refs:pad_matrix_dims (Results 1 – 1 of 1) sorted by relevance
46 auto pad_matrix_dims = [&pad_dim](Shape s) { in PadForTensorCores() local55 Shape new_lshape = pad_matrix_dims(lshape); in PadForTensorCores()56 Shape new_rshape = pad_matrix_dims(rshape); in PadForTensorCores()57 Shape new_result_shape = pad_matrix_dims(result_shape); in PadForTensorCores()