Searched defs:skip_dim (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/ |
| H A D | types.h | 454 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim()
465 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim()
476 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim()
488 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim()
501 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim()
518 inline int FlatSizeSkipDim(const RuntimeShape& shape, int skip_dim) { in FlatSizeSkipDim()
530 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim()
541 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim()
553 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim()
566 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim()
|