Searched refs:new_shape_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 254 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() argument
256 return RuntimeShape(new_shape_size, shape, 1); in ExtendedShape()
280 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape() argument
284 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
285 Resize(new_shape_size); in RuntimeShape()
286 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 113 void ExtendShape(Shape* shape, int new_shape_size);
116 void UnextendShape(Shape* shape, int new_shape_size);
|
| D | tooling_util.cc | 635 void ExtendShape(Shape* shape, int new_shape_size) { in ExtendShape() argument
636 CHECK_GE(new_shape_size, shape->dimensions_count()); in ExtendShape()
637 const int size_increase = new_shape_size - shape->dimensions_count(); in ExtendShape()
643 void UnextendShape(Shape* shape, int new_shape_size) { in UnextendShape() argument
644 CHECK_LE(new_shape_size, shape->dimensions_count()); in UnextendShape()
645 const int size_reduction = shape->dimensions_count() - new_shape_size; in UnextendShape()
|