Searched defs:mat2_sizes (Results 1 – 9 of 9) sorted by relevance
103 auto mat2_sizes = mat2.sizes(); in cublasCommonArgs() local270 IntArrayRef mat2_sizes = mat2.sizes(); in addmm_out_cuda_impl() local1050 IntArrayRef mat2_sizes = mat2.sizes(); in _scaled_mm_out_cuda() local
89 std::vector<IntArrayRef> mat2_sizes = NestedTensor_get_sizes(mat2_ptr); in matmul_with_bmm_nested() local260 const auto& mat2_sizes = mat2_ptr->get_nested_sizes(); in matmul_nested() local
243 IntArrayRef mat2_sizes = mat2.sizes(); in sparse_sampled_addmm_check_inputs() local
27 std::vector<int64_t> mat2_sizes = graph.sizes_of(mat2_data); in check_matmul_args() local
34 std::vector<int64_t> mat2_sizes = graph.sizes_of(mat2_data); in check_addmm_args() local
758 std::vector<int64_t> mat2_sizes = {N, K}; in TEST_F() local
2851 std::vector<int64_t> mat2_sizes = {6, 5, 4, 3}; in TEST() local
1692 const auto mat2_sizes = mat2.sizes(); in baddbmm_with_gemm_() local
1455 at::SymIntArrayRef mat2_sizes, in mm_mat2_backward()