Searched refs:tensor1_larger (Results 1 – 1 of 1) sorted by relevance
1928 const auto tensor1_larger = tensor1.dim() >= tensor2.dim(); in should_fold() local1933 const auto t1 = tensor1_larger ? MaybeOwned<Tensor>::borrowed(tensor1) in should_fold()1936 const auto dim_t2 = tensor1_larger ? tensor2.dim() in should_fold()1953 bool t2_requires_grad = tensor1_larger ? tensor2.requires_grad() : tensor1.requires_grad(); in should_fold()