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

Merge pull request #26120 from amitsrivastava78:patch-25

Browse Source 
PiperOrigin-RevId: 235997867
This commit is contained in:
TensorFlower Gardener 2019-02-27 15:52:20 -08:00
commit 434bf84d50
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(); allocated_tensor->mutable_tensor_shape();
auto tensor_shape_dim0 = allocated_tensor_shape->add_dim(); auto tensor_shape_dim0 = allocated_tensor_shape->add_dim();
int allocated_content_flat_size = 0; 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 auto& value_attr = const_node_parts[i]->attr().at("value");
const tensorflow::TensorProto& tensor = value_attr.tensor(); const tensorflow::TensorProto& tensor = value_attr.tensor();
if (i == 0) { if (i == 0) {
@ -214,7 +214,7 @@ void SvdfCluster::MaybeMergeConstNodes(
std::unique_ptr<char[]> allocated_content( std::unique_ptr<char[]> allocated_content(
new char[allocated_content_flat_size]); new char[allocated_content_flat_size]);
char* content_ptr = allocated_content.get(); 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 auto& value_attr = const_node_parts[i]->attr().at("value");
const tensorflow::TensorProto& tensor = value_attr.tensor(); const tensorflow::TensorProto& tensor = value_attr.tensor();
port::CopyToBuffer(tensor.tensor_content(), content_ptr); port::CopyToBuffer(tensor.tensor_content(), content_ptr);