Home
last modified time | relevance | path

Searched refs:ArgTypeTestKernel (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration_test.cpp773 struct ArgTypeTestKernel final : OperatorKernel { struct
774 …explicit ArgTypeTestKernel(InputType input, std::function<void(const InputType&)> inputExpectation… in ArgTypeTestKernel() argument
789 …::my_op" + schema, c10::RegisterOperators::options().catchAllKernel<ArgTypeTestKernel>(input, inpu… in test()
822 ArgTypeTestKernel<InputType, OutputType>::test( in test()
830 ArgTypeTestKernel<InputType, OutputType>::test( in test()
838 ArgTypeTestKernel<InputType, std::tuple<>>::test( in test()
843 ArgTypeTestKernel<InputType, std::tuple<int64_t, OutputType>>::test( in test()