Searched refs:ArgTypeTestKernel (Results 1 – 1 of 1) sorted by relevance
773 struct ArgTypeTestKernel final : OperatorKernel { struct774 …explicit ArgTypeTestKernel(InputType input, std::function<void(const InputType&)> inputExpectation… in ArgTypeTestKernel() function789 …::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()