MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ExtendShape

Function ExtendShape

tensorflow/lite/toco/tooling_util.cc:632–637  ·  view source on GitHub ↗

Note remaining raw-array extension in ProcessTensorFlowReshapeOperator().

Source from the content-addressed store, hash-verified

630
631// Note remaining raw-array extension in ProcessTensorFlowReshapeOperator().
632void ExtendShape(Shape* shape, int new_shape_size) {
633 CHECK_GE(new_shape_size, shape->dimensions_count());
634 const int size_increase = new_shape_size - shape->dimensions_count();
635 auto* shape_dims = shape->mutable_dims();
636 shape_dims->insert(shape_dims->begin(), size_increase, 1);
637}
638
639// TODO(b/62904716) Remove along with remaining uses.
640void UnextendShape(Shape* shape, int new_shape_size) {

Callers 3

ShuffleArrayTemplateFunction · 0.85
TransposeFunction · 0.85

Calls 4

dimensions_countMethod · 0.80
mutable_dimsMethod · 0.80
insertMethod · 0.45
beginMethod · 0.45

Tested by

no test coverage detected