Searched refs:dim_constraint (Results 1 – 1 of 1) sorted by relevance
40 int64_t dim_constraint = (caller == "Linear") ? 1 : 0; in check_nested_tensor_matrix_constraints() local41 auto dense_size = dense_matrix.size(dim_constraint); in check_nested_tensor_matrix_constraints()47 dim_constraint, in check_nested_tensor_matrix_constraints()52 dim_constraint, in check_nested_tensor_matrix_constraints()