Searched refs:skip_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 609 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() argument
610 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < N); in FlatSizeSkipDim()
613 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
620 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
623 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
627 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
631 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
635 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
639 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
643 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 758 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) {
759 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < N);
762 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i];
769 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
772 if (i != skip_dim) {
776 return FlatSizeSkipDim(dims, skip_dim);
780 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
784 if (i != skip_dim) {
788 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1);
792 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
[all …]
|