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

Build fix

Browse Source 
PiperOrigin-RevId: 239089368
This commit is contained in:
A. Unique TensorFlower 2019-03-18 16:51:32 -07:00 committed by TensorFlower Gardener
parent 7737fbf246
commit 17a8ee5006
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -112,7 +112,7 @@ void FillDiagHelper(const TfLiteTensor* input, TfLiteTensor* output) {
return FillDiag<uint8_t>(input, output, batch_size, row_size, col_size);
}
default:
return FillDiag<float_t>(input, output, batch_size, row_size, col_size);
return FillDiag<float>(input, output, batch_size, row_size, col_size);
}
}

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

@ -57,7 +57,7 @@ template <typename T>
class MatrixDiagOpTest : public ::testing::Test {};
using TypesUnderTest =
::testing::Types<TypeUnion<int32_t>, TypeUnion<float_t>, TypeUnion<int16_t>,
::testing::Types<TypeUnion<int32_t>, TypeUnion<float>, TypeUnion<int16_t>,
TypeUnion<int8_t>, TypeUnion<uint8_t>>;
TYPED_TEST_SUITE(MatrixDiagOpTest, TypesUnderTest);

4
tensorflow/lite/kernels/test_util.h
View File

@ -533,11 +533,11 @@ template <typename T>
struct TypeUnion;
template <>
struct TypeUnion<float_t> {
struct TypeUnion<float> {
public:
static const TensorType tensor_type = TensorType::TensorType_FLOAT32;
static const TfLiteType tflite_type = TfLiteType::kTfLiteFloat32;
typedef float_t ScalarType;
typedef float ScalarType;
};
template <>