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()
|