ONE - On-device Neural Engine
|
#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) |