Searched refs:functionSchema (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | register_distributed_ops.cpp | 62 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 D | torchscript_functions.cpp | 18 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 D | python_functions.cpp | 250 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 D | torchscript_functions.h | 24 const c10::FunctionSchema& functionSchema,
32 const c10::FunctionSchema& functionSchema,
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/ |
| H A D | ir_emitter.cpp | 4326 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()
|