torch.fx.experimental.optimization.use_mkl_length#

torch.fx.experimental.optimization.use_mkl_length(graph)[source]#

This is a heuristic that can be passed into optimize_for_inference that determines whether a subgraph should be run in MKL by checking if there are more than 2 nodes in it