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

Function MatchingDim

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

Get common shape dim, DCHECKing that they all agree.

Source from the content-addressed store, hash-verified

430
431// Get common shape dim, DCHECKing that they all agree.
432inline int MatchingDim(const RuntimeShape& shape1, int index1,
433 const RuntimeShape& shape2, int index2) {
434 TFLITE_DCHECK_EQ(shape1.Dims(index1), shape2.Dims(index2));
435 return std::min(shape1.Dims(index1), shape2.Dims(index2));
436}
437
438template <typename... Args>
439int MatchingDim(const RuntimeShape& shape1, int index1,

Callers 15

LogSoftmaxEvalFunction · 0.85
EvalFunction · 0.85
PickOutputMultiplierFunction · 0.85
DepthwiseConvGeneralFunction · 0.85
ConvFunction · 0.85
DilatedIm2colFunction · 0.85
Im2colFunction · 0.85
FullyConnectedFunction · 0.85
GEMVForLstmCellFunction · 0.85

Calls 2

DimsMethod · 0.80
minFunction · 0.50

Tested by 1

PickOutputMultiplierFunction · 0.68