Searched refs:tensorExprFuserEnabled (Results 1 – 5 of 5) sorted by relevance
20 TORCH_API bool tensorExprFuserEnabled();
158 bool tensorExprFuserEnabled() { in tensorExprFuserEnabled() function
438 if (tensorExprFuserEnabled()) { in runNoGradOptimizations()640 if (!remaining_bailout_depth_.has_value() || !tensorExprFuserEnabled()) { in getOptimizedPlanFor()
950 if (tensorExprFuserEnabled()) { in runNondiffOptimization()
965 .def("_jit_texpr_fuser_enabled", &tensorExprFuserEnabled) in initJITBindings()