Searched refs:output_inflated (Results 1 – 1 of 1) sorted by relevance
678 Tensor output_inflated; in Compute() local682 true /* should_inflate */, &output_inflated)); in Compute()683 if (output_inflated.dims() > contraction_output.dims()) { in Compute()713 ctx, output_inflated, output_permutation, &output)); in Compute()