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

Removed warning from the file.

Browse Source 
Fixed warning from the file.
This commit is contained in:
Amit Srivastava 2019-03-20 09:11:06 +05:30
parent 36b42528ec
commit 4f2a3ae9e6
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/kernels/fully_connected_test.cc
View File

@ -142,7 +142,7 @@ class BaseFullyConnectedOpModel : public SingleOpModel {
FullyConnectedOptionsWeightsFormat_DEFAULT)
: batches_(batches), units_(units) {
int total_input_size = 1;
for (int i = 0; i < input.shape.size(); ++i) {
for (size_t i = 0; i < input.shape.size(); ++i) {
total_input_size *= input.shape[i];
}
input_size_ = total_input_size / batches_;
@ -278,7 +278,7 @@ class HybridFullyConnectedOpModel : public SingleOpModel {
const TensorData& output = {TensorType_FLOAT32})
: batches_(batches), units_(units) {
int total_input_size = 1;
for (int i = 0; i < input.shape.size(); ++i) {
for (size_t i = 0; i < input.shape.size(); ++i) {
total_input_size *= input.shape[i];
}
input_size_ = total_input_size / batches_;