diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-02 08:00:03 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-02 08:03:31 -0700 |
| commit | be571938196fb191f260a2c45176d406e6c19a13 (patch) | |
| tree | 31dd2158201796c4eb994dab8cae70d397df1bb1 /tensorflow/contrib/lite/toco/model.h | |
| parent | 1fbad5034a8ea531e496b0ecbf9e2c3839b62311 (diff) | |
Adding support for RandomUniform. Basic support for op import/export of RandomUniform, and constant resolution with static seeds.
PiperOrigin-RevId: 191293897
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 5199e292e1..64269d369d 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -60,6 +60,7 @@ enum class OperatorType { kMaxPool, kFakeQuant, kMul, + kRandomUniform, kRange, kRank, kRelu, @@ -946,6 +947,13 @@ struct FloorModOperator : Operator { FloorModOperator() : Operator(OperatorType::kFloorMod) {} }; +struct RandomUniformOperator : Operator { + RandomUniformOperator() : Operator(OperatorType::kRandomUniform) {} + ArrayDataType dtype = ArrayDataType::kNone; + int64 seed; + int64 seed2; +}; + // Creates a sequence of numbers that begins at start and extends by increments // of delta up to but not including limit. // |