Searched refs:skip_dim (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 530 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() argument
531 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < N); in FlatSizeSkipDim()
534 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
541 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
544 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
548 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
552 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
556 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
560 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
564 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 546 inline int FlatSizeSkipDim(const Dims<N>& dims, int skip_dim) { in FlatSizeSkipDim() argument
547 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < N); in FlatSizeSkipDim()
550 flat_size *= (i == skip_dim) ? 1 : dims.sizes[i]; in FlatSizeSkipDim()
557 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
560 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
564 return FlatSizeSkipDim(dims, skip_dim); in MatchingFlatSizeSkipDim()
568 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
572 if (i != skip_dim) { in MatchingFlatSizeSkipDim()
576 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
580 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() argument
[all …]
|