experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix std::uniform_int_distribution usage in random uniform op

Browse Source 
The standard does not specify support for char/int8 types, and compilation
fails with MSVC.

GitHub #47452 is related.

PiperOrigin-RevId: 360082481
Change-Id: I3c8614bc8c7e8de10a1807c4f610df7385a8902f
This commit is contained in:
Jaesung Chung 2021-02-28 17:13:56 -08:00 committed by TensorFlower Gardener
parent fc1bdb3a68
commit 358948ec8d
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/random_uniform.cc
View File

@ -140,7 +140,7 @@ TfLiteStatus EvalInt(TfLiteContext* context, TfLiteNode* node) {
size_t output_size = tflite::NumElements(output); size_t output_size = tflite::NumElements(output);
switch (output->type) { switch (output->type) {
case kTfLiteInt8: case kTfLiteInt8:
RandomUniformSample<int8_t, std::uniform_int_distribution<int8_t>>( RandomUniformSample<int8_t, std::uniform_int_distribution<int32_t>>(
params->rng, GetTensorData<int8_t>(output), output_size, min_value, params->rng, GetTensorData<int8_t>(output), output_size, min_value,
max_value); max_value);
break; break;

7
tensorflow/lite/kernels/random_uniform_test.cc
View File

@ -36,6 +36,11 @@ tflite::TensorType GetTTEnum<double>() {
return tflite::TensorType_FLOAT64; return tflite::TensorType_FLOAT64;
} }
template <>
tflite::TensorType GetTTEnum<int8_t>() {
return tflite::TensorType_INT8;
}
template <> template <>
tflite::TensorType GetTTEnum<int32_t>() { tflite::TensorType GetTTEnum<int32_t>() {
return tflite::TensorType_INT32; return tflite::TensorType_INT32;
@ -150,7 +155,7 @@ class RandomUniformIntTest : public ::testing::Test {
using Int = IntType; using Int = IntType;
}; };
using TestTypesInt = ::testing::Types<int32_t, int64_t>; using TestTypesInt = ::testing::Types<int8_t, int32_t, int64_t>;
TYPED_TEST_SUITE(RandomUniformIntTest, TestTypesInt); TYPED_TEST_SUITE(RandomUniformIntTest, TestTypesInt);