Searched defs:batch_tensor2 (Results 1 – 2 of 2) sorted by relevance
115 SymIntArrayRef batch_tensor2(x2.sym_sizes().data(), dim2 - 2); in cdist_impl() local200 IntArrayRef batch_tensor2(_x2.sizes().data(), dim2 - 2); in _cdist_backward() local
2108 const IntArrayRef batch_tensor2(tensor2.sizes().data(), in _matmul_impl() local