Searched defs:arg_t (Results 1 – 7 of 7) sorted by relevance
61 using arg_t = std::tuple_element_t<arg_index, args_t>; in apply() typedef74 using arg_t = std::tuple_element_t<arg_index, args_t>; in apply() typedef370 using arg_t = typename traits::template arg<i>::type; in apply() typedef
252 using arg_t = typename binary_function_traits<func_t>::arg1_t; typedef288 using arg_t = at::opmath_type<scalar_t>; typedef343 using arg_t = typename std::decay<typename traits::template arg<0>::type>::type; typedef1138 using arg_t = typename traits::template arg<0>::type; in gpu_reduce_kernel() typedef1245 using arg_t = at::opmath_type<scalar_t>; in jitted_gpu_reduce_kernel() typedef
351 using arg_t = typename MinOps<scalar_t>::arg_t; typedef391 using arg_t = std::pair<scalar_t, int64_t>; in argmax_kernel_impl() typedef415 using arg_t = std::pair<scalar_t, int64_t>; in argmin_kernel_impl() typedef
23 FunctionBody::FunctionBody(const FunctionDef& f, DataTypeSlice arg_t, in FunctionBody()
151 FullTypeDef* arg_t = ret_type.add_args(); in UnaryContainerCreate() local
242 FullTypeDef* arg_t = t.mutable_args(j); in SubstituteGeneric() local
451 using arg_t = detail::pair<scalar_t, index_t>; member