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

Document vector/tensor copy functions

Browse Source
This commit is contained in:
Reuben Morais 2019-06-17 08:42:18 -03:00
parent 4b305d2f5e
commit ea1422d47b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
native_client/tflitemodelstate.cc
View File

@ -170,6 +170,8 @@ TFLiteModelState::init(const char* model_path,
return DS_ERR_OK;
}
// Copy contents of vec into the tensor with index tensor_idx.
// If vec.size() < num_elements, set the remainder of the tensor values to zero.
void
TFLiteModelState::copy_vector_to_tensor(const vector<float>& vec,
int tensor_idx,
@ -185,6 +187,7 @@ TFLiteModelState::copy_vector_to_tensor(const vector<float>& vec,
}
}
// Copy num_elements elements from the tensor with index tensor_idx into vec
void
TFLiteModelState::copy_tensor_to_vector(int tensor_idx,
int num_elements,