Searched refs:dropout_schema (Results 1 – 1 of 1) sorted by relevance
116 static const c10::FunctionSchema dropout_schema = torch::jit::parseSchema( in is_nondeterministic() local118 if (dropout_schema == schema_ && value_map_.count("train") && in is_nondeterministic()