Poplar and PopLibs
RandomSeed.hpp
1// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
2#ifndef poplar_RandomSeed_hpp
3#define poplar_RandomSeed_hpp
4
5#include <poplar/Graph.hpp>
6#include <string>
7
8namespace poplar {
9
21 const DebugContext &debugContext = {});
22
32void setHwSeeds(Graph &graph, const Tensor &hwSeeds, program::Sequence &prog,
33 const DebugContext &debugContext = {});
34
35} // end namespace poplar
36
37#endif // poplar_RandomSeed_hpp
DebugContext gathers the common external parameters of the context of an operation.
Definition: DebugContext.hpp:221
This class represents a graph program to be executed on the IPU.
Definition: Graph.hpp:52
A reference to a subset of tensor elements.
Definition: Tensor.hpp:38
Program that executes a sequence of programs.
Definition: Program.hpp:77
Poplar classes and functions.
Definition: ArrayRef.hpp:14
void setHwSeeds(Graph &graph, const Tensor &hwSeeds, program::Sequence &prog, const DebugContext &debugContext={})
Sets the hw seeds for each worker in a device from a snapshot of the seeds.
Tensor getHwSeeds(Graph &graph, program::Sequence &prog, const DebugContext &debugContext={})
Gets a snapshot of the h/w seeds for each worker in device.