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

RuntimeShapes class: minor tweak to fix builds.

Browse Source 
PiperOrigin-RevId: 198755870
This commit is contained in:
A. Unique TensorFlower 2018-05-31 11:52:43 -07:00 committed by TensorFlower Gardener
parent 10fa513e15
commit fdf4d0813d
1 changed files with 3 additions and 3 deletions

6
tensorflow/contrib/lite/kernels/internal/types.h
View File

@ -71,8 +71,8 @@ class RuntimeShape {
}
}
inline const int32 DimensionsCount() const { return size_; }
inline const int32 Dims(int i) const {
inline int32 DimensionsCount() const { return size_; }
inline int32 Dims(int i) const {
TFLITE_DCHECK_GE(i, 0);
TFLITE_DCHECK_LT(i, size_);
return size_ > kMaxSmallSize ? dims_pointer_[i] : dims_[i];
@ -123,7 +123,7 @@ class RuntimeShape {
// Returns the total count of elements, that is the size when flattened into a
// vector.
inline const int FlatSize() const {
inline int FlatSize() const {
int buffer_size = 1;
const int* dims_data = DimsData();
for (int i = 0; i < size_; i++) {