ONE - On-device Neural Engine
•All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Namespaces | Functions
StatelessRandomUniform.h File Reference
#include "cker/Types.h"
#include "cker/Shape.h"
#include "cker/Utils.h"
#include "cker/eigen/EigenSupport.h"
#include "cker/operation/Helper/Tensor.h"
#include "cker/operation/Helper/PhiloxRandom.h"
#include "cker/operation/Helper/RandomOpCpu.h"
#include "cker/operation/Helper/RandomDistributions.h"

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)