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

Merge pull request #24648 from siju-samuel:patch-9

Browse Source 
PiperOrigin-RevId: 249840868
This commit is contained in:
TensorFlower Gardener 2019-05-24 09:00:27 -07:00
commit 8ecc723c27
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/kernels/expand_dims.cc
View File

@ -14,7 +14,9 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#include <string.h>
#include <vector>
#include "tensorflow/lite/c/builtin_op_data.h"
#include "tensorflow/lite/c/c_api_internal.h"
#include "tensorflow/lite/kernels/internal/reference/reference_ops.h"
@ -27,7 +29,6 @@ namespace builtin {
namespace expand_dims {
constexpr int kInput = 0;
constexpr int kAxis = 1;
constexpr int kOutput = 0;
namespace {
TfLiteStatus ExpandTensorDim(TfLiteContext* context, const TfLiteTensor& input,