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

Change check to allow tensors with up to 6 dims.

Browse Source 
PiperOrigin-RevId: 298531979
Change-Id: I6b4d6196d68c32fb93c84c9fbd980197c118ebaa
This commit is contained in:
A. Unique TensorFlower 2020-03-02 22:55:00 -08:00 committed by TensorFlower Gardener
parent b3b8d1df64
commit 48393637f8
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/graph_transformations/propagate_fixed_sizes.cc
View File

@ -358,8 +358,8 @@ void ProcessOpWithShapeInput(Model* model, Operator* op) {
return;
}
CHECK(dims_array.data_type == ArrayDataType::kInt32) << "dims must be int32";
CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 4)
<< "dims vector can be no larger than 4 values";
CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 6)
<< "dims vector can be no larger than 6 values";
std::vector<int32> const& dims =
dims_array.GetBuffer<ArrayDataType::kInt32>().data;