Home
last modified time | relevance | path

Searched refs:setTensorExprDynamicShapeFusionEnabled (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.h21 TORCH_API void setTensorExprDynamicShapeFusionEnabled(bool val);
H A Dtensorexpr_fuser.cpp173 void setTensorExprDynamicShapeFusionEnabled(bool val) { in setTensorExprDynamicShapeFusionEnabled() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/static/
H A Dfusion.cpp339 setTensorExprDynamicShapeFusionEnabled(true); in performTensorExprFusion()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp971 &setTensorExprDynamicShapeFusionEnabled) in initJITBindings()