Searched refs:stack_layer (Results 1 – 6 of 6) sorted by relevance
41 SimpleTensor<T> stack_layer(const std::vector<SimpleTensor<T>> &in, const TensorShape &output_shape… in stack_layer() function119 template SimpleTensor<int> stack_layer(const std::vector<SimpleTensor<int>> &in, const TensorShape …120 template SimpleTensor<short> stack_layer(const std::vector<SimpleTensor<short>> &in, const TensorSh…121 template SimpleTensor<char> stack_layer(const std::vector<SimpleTensor<char>> &in, const TensorShap…
39 SimpleTensor<T> stack_layer(const std::vector<SimpleTensor<T>> &in, const TensorShape &output_shape…
43 for stack_layer in self.context_stack.iter().rev() { in lookup()44 match stack_layer { in lookup()85 for stack_layer in self.context_stack.iter().rev() { in lookup_index()86 match stack_layer { in lookup_index()
129 return reference::stack_layer<T>(src, shape_dst, data_type, axis_u); in compute_reference()
91 __kernel void stack_layer(
1557 __kernel void stack_layer(