Home
last modified time | relevance | path

Searched refs:new_dims_cnt (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils.cc82 int new_dims_cnt = 0; in RemoveOneSizeDimensions() local
87 input_shape->SetDim(new_dims_cnt, input_shape->Dims(i)); in RemoveOneSizeDimensions()
88 ++new_dims_cnt; in RemoveOneSizeDimensions()
90 input_shape->Resize(new_dims_cnt); in RemoveOneSizeDimensions()
94 new_dims_cnt = 0; in RemoveOneSizeDimensions()
99 new_params.perm[new_dims_cnt] = params->perm[i]; in RemoveOneSizeDimensions()
100 output_shape->SetDim(new_dims_cnt, output_shape->Dims(i)); in RemoveOneSizeDimensions()
101 ++new_dims_cnt; in RemoveOneSizeDimensions()
103 output_shape->Resize(new_dims_cnt); in RemoveOneSizeDimensions()
104 new_params.perm_count = new_dims_cnt; in RemoveOneSizeDimensions()
[all …]