| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 620 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
631 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
639 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
643 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
652 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2); in MatchingFlatSizeSkipDim()
656 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
666 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2, in MatchingFlatSizeSkipDim()
685 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
696 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
705 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1); in MatchingFlatSizeSkipDim()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 769 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
780 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
788 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1);
792 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
801 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2);
805 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim,
815 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2,
834 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim,
845 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim,
854 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1);
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | l2normalization.h | 36 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape);
62 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
|
| D | softmax.h | 35 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
88 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
172 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxInt16()
|
| D | reference_ops.h | 692 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
885 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LocalResponseNormalization()
910 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
960 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
1512 MatchingFlatSizeSkipDim(input_x_shape, 0, input_y_shape, output_shape); in RankOneSelect()
1804 MatchingFlatSizeSkipDim(input_shape, 0, output_shape); in SegmentSum()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 51 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 70 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxHifimini()
220 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in EvalHifi4()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | l2norm.cc | 126 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | l2norm.cc | 137 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
|
| D | activations.cc | 1228 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1739 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape);
1768 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2945 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
3768 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
3910 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxInt8LUT()
4074 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
4120 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
|
| D | legacy_optimized_ops.h | 3170 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
4011 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
|