Searched refs:check_dims_1 (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | types.h | 577 const Dims<N>& check_dims_1) { in MatchingFlatSize() argument 581 return MatchingFlatSize(dims, check_dims_1); in MatchingFlatSize() 586 const Dims<N>& check_dims_1, in MatchingFlatSize() argument 591 return MatchingFlatSize(dims, check_dims_1, check_dims_2); in MatchingFlatSize() 596 const Dims<N>& check_dims_1, in MatchingFlatSize() argument 602 return MatchingFlatSize(dims, check_dims_1, check_dims_2, check_dims_3); in MatchingFlatSize() 633 const Dims<N>& check_dims_1) { in MatchingFlatSizeSkipDim() argument 639 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim() 645 const Dims<N>& check_dims_1, in MatchingFlatSizeSkipDim() argument 652 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2); in MatchingFlatSizeSkipDim() [all …]
|
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
D | types.h | 726 const Dims<N>& check_dims_1) { 730 return MatchingFlatSize(dims, check_dims_1); 735 const Dims<N>& check_dims_1, 740 return MatchingFlatSize(dims, check_dims_1, check_dims_2); 745 const Dims<N>& check_dims_1, 751 return MatchingFlatSize(dims, check_dims_1, check_dims_2, check_dims_3); 782 const Dims<N>& check_dims_1) { 788 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); 794 const Dims<N>& check_dims_1, 801 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2); [all …]
|