Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/runtime/
Dargument_spec.cpp8 void ArgumentSpecCreator::scan( in scan()
78 ArgumentSpecCreator::WrittenSlots& written_slots) { in scanWrittenSlots()
94 ArgumentSpecCreator::ArgumentSpecCreator(Graph& graph) in ArgumentSpecCreator() function in torch::jit::ArgumentSpecCreator
103 void ArgumentSpecCreator::dump() const { in dump()
132 ArgumentSpec ArgumentSpecCreator::create(bool with_grad, const Stack& input) in create()
195 void ArgumentSpecCreator::specializeTypes( in specializeTypes()
Dargument_spec.h173 struct TORCH_API ArgumentSpecCreator { struct
193 ArgumentSpecCreator(Graph& graph); argument
Dgraph_executor.cpp783 ArgumentSpecCreator arg_spec_creator_;
/external/pytorch/test/cpp/jit/
Dtest_argument_spec.cpp149 ArgumentSpecCreator arg_spec_creator(*graph); in TEST()
Dtest_autodiff.cpp257 ArgumentSpecCreator asc(*g); in TEST()
Dtest_misc.cpp2012 ArgumentSpecCreator arg_spec_creator(opt_graph); in TEST()
/external/pytorch/torch/csrc/jit/python/
Dinit.cpp553 ArgumentSpecCreator arg_spec_creator(*graph); in initJITBindings()