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

Updated resolve_svdf.cc

Browse Source 
Fixed warning in the file
This commit is contained in:
Amit 2019-02-26 08:31:20 +05:30 committed by GitHub
parent db1ce36dba
commit 310d35fb50
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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/toco/tensorflow_graph_matching/resolve_svdf.cc
View File

@ -186,7 +186,7 @@ void SvdfCluster::MaybeMergeConstNodes(
allocated_tensor->mutable_tensor_shape();
auto tensor_shape_dim0 = allocated_tensor_shape->add_dim();
int allocated_content_flat_size = 0;
for (int i = 0; i < const_node_parts.size(); i++) {
for (size_t i = 0; i < const_node_parts.size(); i++) {
const auto& value_attr = const_node_parts[i]->attr().at("value");
const tensorflow::TensorProto& tensor = value_attr.tensor();
if (i == 0) {
@ -214,7 +214,7 @@ void SvdfCluster::MaybeMergeConstNodes(
std::unique_ptr<char[]> allocated_content(
new char[allocated_content_flat_size]);
char* content_ptr = allocated_content.get();
for (int i = 0; i < const_node_parts.size(); i++) {
for (size_t i = 0; i < const_node_parts.size(); i++) {
const auto& value_attr = const_node_parts[i]->attr().at("value");
const tensorflow::TensorProto& tensor = value_attr.tensor();
port::CopyToBuffer(tensor.tensor_content(), content_ptr);