Searched refs:num_squeezed_dims (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dsqueeze.cc52 int num_squeezed_dims = 0; local
57 ++num_squeezed_dims;
66 if (!should_squeeze[current]) ++num_squeezed_dims;
72 TfLiteIntArrayCreate(input_num_dims - num_squeezed_dims);

Completed in 94 milliseconds