Searched refs:capture_pre_autograd_graph_using_training_ir (Results 1 – 5 of 5) sorted by relevance
112 from torch._utils_internal import capture_pre_autograd_graph_using_training_ir128 if capture_pre_autograd_graph_using_training_ir():
9 from torch._utils_internal import capture_pre_autograd_graph_using_training_ir262 is_training_ir_flag = capture_pre_autograd_graph_using_training_ir()1041 assert capture_pre_autograd_graph_using_training_ir()1132 assert capture_pre_autograd_graph_using_training_ir()
7 from torch._utils_internal import capture_pre_autograd_graph_using_training_ir686 if capture_pre_autograd_graph_using_training_ir():
7 from torch._utils_internal import capture_pre_autograd_graph_using_training_ir1866 if capture_pre_autograd_graph_using_training_ir():
158 def capture_pre_autograd_graph_using_training_ir() -> bool: function