Searched defs:expand_batch_portion (Results 1 – 4 of 4) sorted by relevance
116 std::vector<SymInt> expand_batch_portion = infer_size_symint(batch_tensor1, batch_tensor2); in cdist_impl() local201 std::vector<int64_t> expand_batch_portion = infer_size(batch_tensor1, batch_tensor2); in _cdist_backward() local
318 std::vector<int64_t> expand_batch_portion = infer_size(arg1_batch_sizes, arg2_batch_sizes); in _linalg_broadcast_batch_dims() local
267 std::vector<int64_t> expand_batch_portion = at::infer_size(A_batch_sizes, B_batch_sizes); in apply_gels_batched() local
2641 std::vector<int64_t> expand_batch_portion = at::infer_size(A_batch_sizes, B_batch_sizes); in linalg_lstsq_gels() local