Searched defs:ntensors2 (Results 1 – 2 of 2) sorted by relevance
25 int64_t ntensors2 = mat2.is_nested() ? get_nested_tensor_impl(mat2)->size(0) : mat2.size(0); in bmm_nested() local250 int64_t ntensors = self_ptr->size(0), ntensors2 = mat2_ptr->size(0); in matmul_nested() local
292 int64_t ntensors = self_ptr->size(0), ntensors2 = mat2_ptr->size(0); in bmm_nested_cuda() local