Searched refs:_test_mlp_contiguous_relu_compile (Results 1 – 1 of 1) sorted by relevance
183 def _test_mlp_contiguous_relu_compile(backend, dense_input_shape): member in SparseSemiStructuredTensorCompileTest230 …SparseSemiStructuredTensorCompileTest._test_mlp_contiguous_relu_compile("cusparselt", dense_input_…240 …SparseSemiStructuredTensorCompileTest._test_mlp_contiguous_relu_compile("cutlass", dense_input_sha…