Lines Matching defs:backend
183 def _test_mlp_contiguous_relu_compile(backend, dense_input_shape): argument
271 def test_to_sparse_semi_structured(self, dtype, backend): argument
286 def test_mm_sparse_first_NN(self, dense_input_shape, dtype, device, backend): argument
313 def test_mm_sparse_first_NT(self, dense_input_shape, dtype, device, backend): argument
349 def test_mm_sparse_first_TN(self, dtype, dense_input_shape, device, backend): argument
370 def test_mm_sparse_second_NT(self, dense_input_shape, dtype, device, backend): argument
395 def test_mm_sparse_second_NN(self, dense_input_shape, dtype, device, backend): argument
416 def test_linear(self, dense_input_shape, inference_mode, device, backend): argument
444 def test_mlp(self, device, dense_input_shape, backend): argument
474 def test_values(self, backend): argument
484 def test_indices(self, backend): argument
494 def test_min_sparse_shape(self, dtype, device, backend): argument
512 def test_unsupported_shape(self, dtype, device, backend): argument
522 def test_unsupported_dtype(self, dtype, device, backend): argument
535 def test_unsupported_dim(self, device, backend): argument
618 def test_pruning_algo_largest_abs_values_greedy(self, dtype, backend) -> None: argument
654 def test_pack_both_ways_meta_correctness(self, dtype, backend) -> None: argument
920 def test_sparse_semi_structured_ops_cutlass(self, device, dtype, backend): argument