Searched defs:ArgTypeTestKernel (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/ |
| H A D | op_registration_test.cpp | 773 struct ArgTypeTestKernel final : OperatorKernel { struct
774 …explicit ArgTypeTestKernel(InputType input, std::function<void(const InputType&)> inputExpectation… in ArgTypeTestKernel() function
777 OutputType operator()(InputType input) const { in operator ()()
782 …Expectation, OutputType output, std::function<void(const c10::Stack&)> outputExpectation, const st… in test()
787 …on, OutputType output, std::function<void(const c10::Stack&)> outputExpectation, const std::string… in test()
793 …on, OutputType output, std::function<void(const c10::Stack&)> outputExpectation, const std::string… in test()
803 …n<void(const InputType&)> inputExpectation, OutputType output, std::function<void(const c10::Stack… in test_()
811 InputType input_;
812 std::function<void(const InputType&)> inputExpectation_;
813 OutputType output_;
[all …]
|