Searched refs:non_flatten_input_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 285 RuntimeShape non_flatten_input_shape; in TEST() local
289 input_shape, output_shape, params, &non_flatten_input_shape, in TEST()
292 EXPECT_EQ(non_flatten_input_shape, RuntimeShape({5, 7})); in TEST()
312 RuntimeShape non_flatten_input_shape; in TEST() local
316 input_shape, output_shape, params, &non_flatten_input_shape, in TEST()
319 EXPECT_EQ(non_flatten_input_shape, RuntimeShape({5, 7, 9})); in TEST()
340 RuntimeShape non_flatten_input_shape; in TEST() local
344 input_shape, output_shape, params, &non_flatten_input_shape, in TEST()
347 EXPECT_EQ(non_flatten_input_shape, RuntimeShape({7, 9})); in TEST()
|
| D | transpose_utils.cc | 122 RuntimeShape* non_flatten_input_shape, in Flatten() argument
139 non_flatten_input_shape->Resize(new_dims_cnt); in Flatten()
144 non_flatten_input_shape->SetDim(i - skip_dims_cnt, input_shape.Dims(i)); in Flatten()
|
| D | transpose_utils.h | 44 RuntimeShape* non_flatten_input_shape,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7896 RuntimeShape non_flatten_input_shape; in Transpose() local
7902 &non_flatten_input_shape, &non_flatten_output_shape, in Transpose()
7907 TransposeImpl<T, N>(non_flatten_params, non_flatten_input_shape, in Transpose()
|