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

Function DimsToShape

tensorflow/lite/kernels/internal/types.h:318–321  ·  view source on GitHub ↗

TODO(b/80418076): Move to legacy ops file, update invocations.

Source from the content-addressed store, hash-verified

316
317// TODO(b/80418076): Move to legacy ops file, update invocations.
318inline RuntimeShape DimsToShape(const tflite::Dims<4>& dims) {
319 return RuntimeShape(
320 {dims.sizes[3], dims.sizes[2], dims.sizes[1], dims.sizes[0]});
321}
322
323// Gets next index to iterate through a multidimensional array.
324inline bool NextIndex(const int num_dims, const int* dims, int* current) {

Callers 15

DepthwiseConvFunction · 0.85
FullyConnectedFunction · 0.85
ShuffledFullyConnectedFunction · 0.85
DilatedIm2colFunction · 0.85
Im2colFunction · 0.85
ConvFunction · 0.85
HybridConvFunction · 0.85
TransposeConvFunction · 0.85
TransposeIm2colFunction · 0.85
LstmCellFunction · 0.85

Calls 1

RuntimeShapeClass · 0.85

Tested by

no test coverage detected