Searched refs:non_flatten_output_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils_test.cc | 286 RuntimeShape non_flatten_output_shape; in TEST() local
290 &non_flatten_output_shape, &non_flatten_params); in TEST()
293 EXPECT_EQ(non_flatten_output_shape, RuntimeShape({7, 5})); in TEST()
313 RuntimeShape non_flatten_output_shape; in TEST() local
317 &non_flatten_output_shape, &non_flatten_params); in TEST()
320 EXPECT_EQ(non_flatten_output_shape, RuntimeShape({7, 5, 9})); in TEST()
341 RuntimeShape non_flatten_output_shape; in TEST() local
345 &non_flatten_output_shape, &non_flatten_params); in TEST()
348 EXPECT_EQ(non_flatten_output_shape, RuntimeShape({9, 7})); in TEST()
|
| D | transpose_utils.cc | 123 RuntimeShape* non_flatten_output_shape, in Flatten() argument
140 non_flatten_output_shape->Resize(new_dims_cnt); in Flatten()
145 non_flatten_output_shape->SetDim(i - skip_dims_cnt, output_shape.Dims(i)); in Flatten()
|
| D | transpose_utils.h | 45 RuntimeShape* non_flatten_output_shape,
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 7897 RuntimeShape non_flatten_output_shape; in Transpose() local
7902 &non_flatten_input_shape, &non_flatten_output_shape, in Transpose()
7908 input_data + i, non_flatten_output_shape, in Transpose()
|