Searched refs:contraction_output_reshaped (Results 1 – 1 of 1) sorted by relevance
638 Tensor contraction_output_reshaped; in Compute() local641 &contraction_output_reshaped)); in Compute()645 TensorShape result_shape = contraction_output_reshaped.shape(); in Compute()671 ctx, EinsumHelper::CopyFrom(contraction_output_reshaped, result_shape, in Compute()