Searched refs:new_shape_size (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 278 inline static RuntimeShape ExtendedShape(int new_shape_size, in ExtendedShape() argument
280 return RuntimeShape(new_shape_size, shape, 1); in ExtendedShape()
304 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value) in RuntimeShape() argument
308 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount()); in RuntimeShape()
309 Resize(new_shape_size); in RuntimeShape()
310 const int size_increase = new_shape_size - shape.DimensionsCount(); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 441 inline static RuntimeShape ExtendedShape(int new_shape_size,
443 return RuntimeShape(new_shape_size, shape, 1);
467 RuntimeShape(int new_shape_size, const RuntimeShape& shape, int pad_value)
471 TFLITE_CHECK_GE(new_shape_size, shape.DimensionsCount());
472 Resize(new_shape_size);
473 const int size_increase = new_shape_size - shape.DimensionsCount();
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 111 void ExtendShape(Shape* shape, int new_shape_size);
114 void UnextendShape(Shape* shape, int new_shape_size);
|
| D | tooling_util.cc | 638 void ExtendShape(Shape* shape, int new_shape_size) { in ExtendShape() argument
639 CHECK_GE(new_shape_size, shape->dimensions_count()); in ExtendShape()
640 const int size_increase = new_shape_size - shape->dimensions_count(); in ExtendShape()
646 void UnextendShape(Shape* shape, int new_shape_size) { in UnextendShape() argument
647 CHECK_LE(new_shape_size, shape->dimensions_count()); in UnextendShape()
648 const int size_reduction = shape->dimensions_count() - new_shape_size; in UnextendShape()
|