@@ -103,18 +103,19 @@ class OpRegistrar : public Registrar {
103103template <typename PlaceType, bool at_end, size_t I, typename ... KernelType>
104104struct OpKernelRegistrarFunctor ;
105105
106- template <typename PlaceType, size_t I, typename ... KernelType>
107- struct OpKernelRegistrarFunctor <PlaceType, false , I, KernelType...> {
108- using KT = typename std::tuple_element<I, std::tuple<KernelType...>>::type;
106+ template <typename PlaceType, size_t I, typename ... KernelTypes>
107+ struct OpKernelRegistrarFunctor <PlaceType, false , I, KernelTypes...> {
108+ using KERNEL_TYPE =
109+ typename std::tuple_element<I, std::tuple<KernelTypes...>>::type;
109110
110111 void operator ()(const char * op_type) const {
111- using T = typename KT ::ELEMENT_TYPE;
112+ using T = typename KERNEL_TYPE ::ELEMENT_TYPE;
112113 OperatorWithKernel::OpKernelKey key (ToDataType (std::type_index (typeid (T))),
113114 PlaceType ());
114- OperatorWithKernel::AllOpKernels ()[op_type][key].reset (new KT );
115+ OperatorWithKernel::AllOpKernels ()[op_type][key].reset (new KERNEL_TYPE );
115116
116- constexpr auto size = std::tuple_size<std::tuple<KernelType ...>>::value;
117- OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1 , KernelType ...>
117+ constexpr auto size = std::tuple_size<std::tuple<KernelTypes ...>>::value;
118+ OpKernelRegistrarFunctor<PlaceType, I + 1 == size, I + 1 , KernelTypes ...>
118119 func;
119120 func (op_type);
120121 }
0 commit comments