Home
last modified time | relevance | path

Searched defs:ArgumentStash (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dtracer.h103 struct ArgumentStash { struct
104 struct IntArrayRefTrace : std::vector<Value*> {
108 static bool empty() { in empty()
118 static bool hasIntArrayRef(const std::string& arg_name) { in hasIntArrayRef()
122 static IntArrayRefTrace popIntArrayRef(const std::string& arg_name) { in popIntArrayRef()
148 static thread_local ArgumentStash stash; argument
149 std::unordered_map<std::string, IntArrayRefTrace> intlists;
150 std::unordered_map<std::string, Value*> values;
H A Dtracer.cpp607 using ArgumentStash = jit::tracer::ArgumentStash; in addInputs() typedef
621 using ArgumentStash = jit::tracer::ArgumentStash; in addInputs() typedef
646 using ArgumentStash = jit::tracer::ArgumentStash; in addInputs() typedef
789 using ArgumentStash = jit::tracer::ArgumentStash; in addInputs() typedef