Searched refs:stack_serial_kernel_impl (Results 1 – 3 of 3) sorted by relevance
16 detail::stack_serial_kernel_impl<scalar_t, TensorList>(result, tensors, dim); in stack_serial_kernel()
29 void stack_serial_kernel_impl(Tensor& result, TensorListType tensors, int64_t dim) { in stack_serial_kernel_impl() function
782 stack_serial_kernel_impl<scalar_t, ProcessedNodeInputWrapper>( in varStackSerialOut()