Searched refs:out_el_shape (Results 1 – 1 of 1) sorted by relevance
71 def out_el_shape(out_shape, num_classes): function82 output_shape = out_el_shape(y_shape, n_classes)85 out_el_shape(v, n_classes[k]