OneHot.h File Reference

#include "cker/Shape.h"

Go to the source code of this file.

Namespaces
namespace	nnfw
namespace	nnfw::cker

Functions
template<typename T , typename TI >
void	nnfw::cker::OneHot (const int32_t depth, const T on_value, const T off_value, int32_t axis, const Shape &indices_shape, const TI *indices_data, const Shape &, T *output_data)