Searched refs:t1_axis_labels (Results 1 – 1 of 1) sorted by relevance
901 def _einsum_v1_reduction(t0, t0_axis_labels, t1, t1_axis_labels, axes_to_sum): argument931 if len(t1_axis_labels) != len(t1.shape):934 (len(t1.shape), len(t1_axis_labels)))950 assert all(a in t0_axis_labels and a in t1_axis_labels for a in axes_to_sum)951 preserved_axes = (set(t0_axis_labels) & set(t1_axis_labels)) - axes_to_sum953 for i, sym_list in enumerate([t0_axis_labels, t1_axis_labels]):969 axis_labels = [t0_axis_labels, t1_axis_labels]