Searched refs:output_size_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 35 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestReferenceResizeNearestNeighbor() local
39 op_params, input_shape, input_data.data(), output_size_shape, in TestReferenceResizeNearestNeighbor()
156 RuntimeShape output_size_shape({1, 1, 1, 2}); in TestOptimizedResizeNearestNeighbor() local
172 op_params, input_shape, input_data.data(), output_size_shape, in TestOptimizedResizeNearestNeighbor()
175 op_params, input_shape, input_data.data(), output_size_shape, in TestOptimizedResizeNearestNeighbor()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fixed_sizes.cc | 948 const auto& output_size_shape = output_size_array.shape(); in ProcessResizeBilinearOperator() local
949 CHECK_EQ(output_size_shape.dimensions_count(), 1); in ProcessResizeBilinearOperator()
950 CHECK_EQ(output_size_shape.dims(0), 2); in ProcessResizeBilinearOperator()
976 const auto& output_size_shape = output_size_array.shape(); in ProcessResizeNearestNeighborOperator() local
977 CHECK_EQ(output_size_shape.dimensions_count(), 1); in ProcessResizeNearestNeighborOperator()
978 CHECK_EQ(output_size_shape.dims(0), 2); in ProcessResizeNearestNeighborOperator()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 3022 const RuntimeShape output_size_shape = in ResizeBilinear() local
3032 TFLITE_DCHECK_EQ(output_size_shape.Dims(0), 1); in ResizeBilinear()
3033 TFLITE_DCHECK_EQ(output_size_shape.Dims(1), 1); in ResizeBilinear()
3034 TFLITE_DCHECK_EQ(output_size_shape.Dims(2), 1); in ResizeBilinear()
3035 TFLITE_DCHECK_EQ(output_size_shape.Dims(3), 2); in ResizeBilinear()
3036 int32 output_height = output_size_data[Offset(output_size_shape, 0, 0, 0, 0)]; in ResizeBilinear()
3037 int32 output_width = output_size_data[Offset(output_size_shape, 0, 0, 0, 1)]; in ResizeBilinear()
4550 const RuntimeShape& output_size_shape, const int32* output_size_data, in ResizeNearestNeighbor() argument
4569 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeNearestNeighbor()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5671 const RuntimeShape& output_size_shape, in ResizeBilinear() argument
5688 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeBilinear()
5720 const RuntimeShape& output_size_shape, in ResizeBilinear() argument
5737 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeBilinear()
6371 const RuntimeShape& output_size_shape, const int32* output_size_data, in ResizeNearestNeighbor() argument
6390 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeNearestNeighbor()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5999 const RuntimeShape& output_size_shape, in ResizeBilinear() argument
6016 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeBilinear()
6048 const RuntimeShape& output_size_shape, in ResizeBilinear() argument
6065 TFLITE_DCHECK_EQ(output_size_shape.FlatSize(), 2); in ResizeBilinear()
|