Home
last modified time | relevance | path

Searched refs:createStackForSchema (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dthroughput_benchmark.cpp82 ScriptModuleInput stack = jit::createStackForSchema( in runOnce()
105 jit::Stack stack = jit::createStackForSchema( in addInput()
H A Dpython_dispatch.cpp227 auto stack = torch::jit::createStackForSchema( in ophandle_call_boxed()
274 auto stack = torch::jit::createStackForSchema( in initDispatchBindings()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dpython_functions.cpp258 stack = torch::jit::createStackForSchema( in pyRpcTorchscript()
408 stack = torch::jit::createStackForSchema( in pyRemoteTorchscript()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpybind_utils.cpp757 stack = createStackForSchema(op->schema(), args, kwargs, std::nullopt); in getOpWithStack()
765 stack = createStackForSchema(op->schema(), args, kwargs, std::nullopt); in getOpWithStack()
H A Dpybind_utils.h984 inline Stack createStackForSchema( in createStackForSchema() function
1109 createStackForSchema(callee.getSchema(), args, kwargs, std::move(self)); in runAndInsertCall()