Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/
H A D_ops.py504 current_mode_stack_pre_dispatch = mode_stack_state_for_pre_dispatch()
515 mode_stack_state_for_pre_dispatch().set(0, None)
519 mode_stack_state_for_pre_dispatch().set(1, None)
522 current_mode = mode_stack_state_for_pre_dispatch()._schema_check_mode
523 mode_stack_state_for_pre_dispatch()._schema_check_mode = None
555 current_mode = mode_stack_state_for_pre_dispatch()._schema_check_mode
560 mode_stack_state_for_pre_dispatch()._schema_check_mode = mode
562 current_mode = mode_stack_state_for_pre_dispatch().get(1)
564 mode_stack_state_for_pre_dispatch().set(1, mode)
566 current_mode = mode_stack_state_for_pre_dispatch().get(0)
[all …]