Searched defs:ExtractVariables (Results 1 – 1 of 1) sorted by relevance
246 struct ExtractVariables : IterArgs<ExtractVariables> { struct251 ExtractVariables(std::vector<bool>& is_var, variable_list& list) in ExtractVariables() function253 void operator()(const std::optional<at::Tensor>& x) { in operator()282 ExtractVariables(is_var, list).apply(std::forward<Args>(args)...); in extract_vars() argument