Searched refs:check_shape_1 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 508 const RuntimeShape& check_shape_1) { in MatchingElementsSize() argument
511 const int size_3 = check_shape_1.FlatSize(); in MatchingElementsSize()
531 const RuntimeShape& check_shape_1) { in MatchingFlatSize() argument
537 return MatchingFlatSize(shape, check_shape_1); in MatchingFlatSize()
542 const RuntimeShape& check_shape_1, in MatchingFlatSize() argument
549 return MatchingFlatSize(shape, check_shape_1, check_shape_2); in MatchingFlatSize()
554 const RuntimeShape& check_shape_1, in MatchingFlatSize() argument
562 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3); in MatchingFlatSize()
698 const RuntimeShape& check_shape_1) { in MatchingFlatSizeSkipDim() argument
705 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1); in MatchingFlatSizeSkipDim()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 657 const RuntimeShape& check_shape_1) {
660 const int size_3 = check_shape_1.FlatSize();
680 const RuntimeShape& check_shape_1) {
686 return MatchingFlatSize(shape, check_shape_1);
691 const RuntimeShape& check_shape_1,
698 return MatchingFlatSize(shape, check_shape_1, check_shape_2);
703 const RuntimeShape& check_shape_1,
711 return MatchingFlatSize(shape, check_shape_1, check_shape_2, check_shape_3);
847 const RuntimeShape& check_shape_1) {
854 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1);
[all …]
|