ONE - On-device Neural Engine
Loading...
Searching...
No Matches
StatelessRandomUniform.h File Reference

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

void nnfw::cker::GenerateKey (Tensor seed, random::PhiloxRandom::Key *out_key, random::PhiloxRandom::ResultType *out_counter)
 
template<typename Device , class Distribution >
void nnfw::cker::Fill (random::PhiloxRandom random, Tensor *output)
 
void nnfw::cker::StatelessRandomUniform (const Shape &shape_shape, const int32_t *shape_data, const Shape &seed_shape, const int32_t *seed_data, const Shape &output_shape, float *output_data)