Searched refs:getOpWithStack (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/csrc/distributed/rpc/ |
| D | python_functions.cpp | 86 opWithStack = torch::jit::getOpWithStack(c10OpsForSymbol, args, kwargs); in matchBuiltinOp()
88 opWithStack = torch::jit::getOpWithStack(ops, args, kwargs); in matchBuiltinOp()
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | pybind_utils.cpp | 749 std::pair<std::shared_ptr<Operator>, Stack> getOpWithStack( in getOpWithStack() function
807 auto [found_op, stack] = getOpWithStack(operations, args, kwargs); in invokeOperatorFromPython()
|
| D | pybind_utils.h | 1247 TORCH_PYTHON_API std::pair<std::shared_ptr<Operator>, Stack> getOpWithStack(
|
| D | init.cpp | 1725 auto opWithStack = getOpWithStack(overloads, std::move(args), kwargs); in initJITBindings()
|