Searched defs:batch_tensor1 (Results 1 – 2 of 2) sorted by relevance
114 SymIntArrayRef batch_tensor1(x1.sym_sizes().data(), dim1 - 2); in cdist_impl() local199 IntArrayRef batch_tensor1(_x1.sizes().data(), dim1 - 2); in _cdist_backward() local
2105 auto batch_tensor1 = tensor1.sizes().slice(0, std::max<int64_t>(dim_tensor1 - 2, 0LL)); in _matmul_impl() local