Searched refs:t0_axis_labels (Results 1 – 1 of 1) sorted by relevance
284 def _einsum_reduction(t0, t0_axis_labels, t1, t1_axis_labels, axes_to_sum): argument310 if len(t0_axis_labels) != len(t0.get_shape()):313 (len(t0.get_shape()), len(t0_axis_labels)))333 assert all(a in t0_axis_labels and a in t1_axis_labels for a in axes_to_sum)334 preserved_axes = (set(t0_axis_labels) & set(t1_axis_labels)) - axes_to_sum336 for i, sym_list in enumerate([t0_axis_labels, t1_axis_labels]):352 axis_labels = [t0_axis_labels, t1_axis_labels]