Searched defs:output_size_shape (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
H A D | resize_bilinear.h | 61 const RuntimeShape output_size_shape = in ResizeBilinear() local 149 const RuntimeShape output_size_shape = in ResizeBilinearInteger() local
|
H A D | resize_nearest_neighbor.h | 53 const RuntimeShape& output_size_shape, const int32_t* output_size_data, in ResizeNearestNeighbor()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
H A D | resize_nearest_neighbor_test.cc | 36 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestReferenceResizeNearestNeighbor() local 349 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestOptimizedResizeNearestNeighbor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
H A D | resize_bilinear.h | 1612 const RuntimeShape& output_size_shape, in ResizeBilinear() 1663 const RuntimeShape& output_size_shape, in ResizeBilinear()
|
H A D | optimized_ops.h | 5538 const RuntimeShape& output_size_shape, const int32* output_size_data, in ResizeNearestNeighbor()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
H A D | propagate_fixed_sizes.cc | 961 const auto& output_size_shape = output_size_array.shape(); in ProcessResizeBilinearOperator() local 989 const auto& output_size_shape = output_size_array.shape(); in ProcessResizeNearestNeighborOperator() local
|