Searched refs:contraction_output (Results 1 – 1 of 1) sorted by relevance
669 Tensor contraction_output; in Compute() local672 &contraction_output)); in Compute()681 ctx, contraction_output, result_labels, output_label_counts, in Compute()683 if (output_inflated.dims() > contraction_output.dims()) { in Compute()