Searched defs:OSSOpKernel (Results 1 – 1 of 1) sorted by relevance
50 struct OSSOpKernel { struct51 OSSOpKernel(std::string target, c10::OperatorHandle op_handle) in OSSOpKernel() argument54 std::string target_;55 c10::OperatorHandle op_handle_;56 std::vector<OSSDynamicArg> dynamic_args_;57 std::vector<OSSDynamicArg> outputs_;58 std::vector<c10::IValue> stack_;60 int num_output_tensors() const { in num_output_tensors()