Searched refs:check_nested_tensor_matrix_constraints (Results 1 – 1 of 1) sorted by relevance
15 inline void check_nested_tensor_matrix_constraints( in check_nested_tensor_matrix_constraints() function62 check_nested_tensor_matrix_constraints(input, weight, c10::string_view{"Linear"}); in nested_linear()76 check_nested_tensor_matrix_constraints(self, other, c10::string_view{"Matmul"}); in NestedTensor_matmul()