Searched refs:new_shape_size (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 231 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() argument
233 return RuntimeShape(new_shape_size, shape, 1); in ExtendedShape()
259 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape() argument
261 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
262 TFLITE_CHECK_LE(new_shape_size, kMaxSmallSize); in RuntimeShape()
263 Resize(new_shape_size); in RuntimeShape()
264 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 246 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() argument
248 return RuntimeShape(new_shape_size, shape, 1); in ExtendedShape()
274 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape() argument
278 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
279 Resize(new_shape_size); in RuntimeShape()
280 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 114 void ExtendShape(Shape* shape, int new_shape_size);
117 void UnextendShape(Shape* shape, int new_shape_size);
|
| D | tooling_util.cc | 610 void ExtendShape(Shape* shape, int new_shape_size) { in ExtendShape() argument
611 CHECK_GE(new_shape_size, shape->dimensions_count()); in ExtendShape()
612 const int size_increase = new_shape_size - shape->dimensions_count(); in ExtendShape()
618 void UnextendShape(Shape* shape, int new_shape_size) { in UnextendShape() argument
619 CHECK_LE(new_shape_size, shape->dimensions_count()); in UnextendShape()
620 const int size_reduction = shape->dimensions_count() - new_shape_size; in UnextendShape()
|