Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_distributed_ops.cpp62 auto& functionSchema = cuPtr->get_function(qualifiedName).getSchema(); in prepare_and_call_rpc_op() local
71 userCallableStack.reserve(functionSchema.arguments().size()); in prepare_and_call_rpc_op()
82 i < functionSchema.arguments().size(); in prepare_and_call_rpc_op()
84 const auto& arg = functionSchema.arguments()[i]; in prepare_and_call_rpc_op()
93 functionSchema.name(), in prepare_and_call_rpc_op()
97 functionSchema)); in prepare_and_call_rpc_op()
108 throw std::runtime_error(functionSchema.findErrorInKwargs(names)); in prepare_and_call_rpc_op()
129 functionSchema, in prepare_and_call_rpc_op()
140 functionSchema, in prepare_and_call_rpc_op()
157 functionSchema, in prepare_and_call_rpc_op()
/aosp_15_r20/external/pytorch/torch/csrc/distributed/rpc/
H A Dtorchscript_functions.cpp18 const c10::FunctionSchema& functionSchema, in rpcTorchscript() argument
50 auto returns = functionSchema.returns(); in rpcTorchscript()
84 const c10::FunctionSchema& functionSchema, in remoteTorchscript() argument
93 auto returns = functionSchema.returns(); in remoteTorchscript()
H A Dpython_functions.cpp250 auto functionSchema = PythonRpcHandler::getInstance() in pyRpcTorchscript() local
259 functionSchema, in pyRpcTorchscript()
268 functionSchema, in pyRpcTorchscript()
400 auto functionSchema = PythonRpcHandler::getInstance() in pyRemoteTorchscript() local
409 functionSchema, args, kwargs, std::nullopt); in pyRemoteTorchscript()
415 functionSchema, in pyRemoteTorchscript()
H A Dtorchscript_functions.h24 const c10::FunctionSchema& functionSchema,
32 const c10::FunctionSchema& functionSchema,
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dir_emitter.cpp4326 const auto& functionSchema = callablePtr->getSchema(); in emitRpcExpr() local
4354 matchSchema(functionSchema, loc, *graphPtr, args, kwargs); in emitRpcExpr()
4377 const std::vector<Argument>& returns = functionSchema.returns(); in emitRpcExpr()