Searched refs:stack_func (Results 1 – 1 of 1) sorted by relevance
49 void _test_stack(TensorList inputs, int64_t dim, StackFunc stack_func) { in _test_stack() argument52 auto res = stack_func(inputs, dim); in _test_stack()53 auto res_neg = stack_func(inputs, dim - x.dim() - 1); in _test_stack()