Searched defs:mat1_sizes (Results 1 – 9 of 9) sorted by relevance
26 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1); in check_matmul_args() local188 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1_W_packed); in add_matmul_optimized_node() local
24 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1); in check_q_8w_linear_args() local164 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1_W_packed); in add_q_8w_linear_optimized_node() local
33 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1); in check_addmm_args() local173 std::vector<int64_t> mat1_sizes = graph.sizes_of(mat1_W_packed); in add_addmm_optimized_node() local
102 auto mat1_sizes = mat1.sizes(); in cublasCommonArgs() local269 IntArrayRef mat1_sizes = mat1.sizes(); in addmm_out_cuda_impl() local1049 IntArrayRef mat1_sizes = mat1.sizes(); in _scaled_mm_out_cuda() local
242 IntArrayRef mat1_sizes = mat1.sizes(); in sparse_sampled_addmm_check_inputs() local
757 std::vector<int64_t> mat1_sizes = {M, K}; in TEST_F() local
2850 std::vector<int64_t> mat1_sizes = {3, 4, 5, 6}; in TEST() local
1691 const auto mat1_sizes = mat1.sizes(); in baddbmm_with_gemm_() local
1436 at::SymIntArrayRef mat1_sizes, in mm_mat1_backward()