Searched defs:inference_mode (Results 1 – 5 of 5) sorted by relevance
206 class inference_mode(_DecoratorContextManager): class
155 bool inference_mode = c10::InferenceMode::is_enabled(); in TensorImpl() local
30 void assert_TLS_states(bool inference_mode) { in assert_TLS_states()
416 def test_linear(self, dense_input_shape, inference_mode, device, backend): argument
2054 inference_mode as inference_mode, unknown