Searched defs:ArgumentSpecCreator (Results 1 – 2 of 2) sorted by relevance
173 struct TORCH_API ArgumentSpecCreator { struct193 ArgumentSpecCreator(Graph& graph); argument197 using WrittenSlots = std::unordered_set<std::string>;204 size_t num_inputs_;205 size_t num_tensors_ = 0;206 size_t num_optionals_ = 0;207 std::vector<Inst> instructions_;
94 ArgumentSpecCreator::ArgumentSpecCreator(Graph& graph) in ArgumentSpecCreator() function in torch::jit::ArgumentSpecCreator