Go to the source code of this file.
|
template<typename T > |
std::vector< SimpleTensor< T > > | unstack (const SimpleTensor< T > &input_tensor, std::vector< SimpleTensor< T >> &output_tensors, int axis) |
|
template std::vector< SimpleTensor< float > > | unstack (const SimpleTensor< float > &input_tensor, std::vector< SimpleTensor< float >> &output_tensors, int axis) |
|
template std::vector< SimpleTensor< half > > | unstack (const SimpleTensor< half > &input_tensor, std::vector< SimpleTensor< half >> &output_tensors, int axis) |
|
template std::vector< SimpleTensor< uint8_t > > | unstack (const SimpleTensor< uint8_t > &input_tensor, std::vector< SimpleTensor< uint8_t >> &output_tensors, int axis) |
|