Home
last modified time | relevance | path

Searched refs:compiled_fns (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/jit/
D_state.py110 def _set_jit_overload_cache(key, compiled_fns): argument
111 _jit_function_overload_caching[key] = [fn.qualified_name for fn in compiled_fns]
D_script.py1496 compiled_fns = []
1498 compiled_fns.append(
1503 compiled_fns = existing_compiled_fns + compiled_fns
1506 _set_jit_overload_cache(obj, compiled_fns)
1508 return compiled_fns
/external/pytorch/torch/csrc/jit/python/
Dpython_sugared_value.cpp1363 auto compiled_fns = py::cast<std::vector<StrongFunctionPtr>>(overloads); in toSugaredValue() local
1364 return std::make_shared<FunctionValue>(std::move(compiled_fns)); in toSugaredValue()