| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | l2normalization.h | 34 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
36 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape);
38 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim);
58 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
60 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
62 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
|
| D | softmax.h | 33 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
35 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
37 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
86 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
88 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
90 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
170 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt16() local
172 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxInt16()
174 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in SoftmaxInt16()
|
| D | reference_ops.h | 883 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local
885 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LocalResponseNormalization()
887 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LocalResponseNormalization()
908 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
910 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
912 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LogSoftmax()
958 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
960 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
962 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/xtensa/ |
| D | softmax.cc | 68 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxHifimini() local
70 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxHifimini()
72 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in SoftmaxHifimini()
194 const int trailing_dim = input_shape.DimensionsCount() - 1; in PrepareHifi4() local
196 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in PrepareHifi4()
218 const int trailing_dim = input_shape.DimensionsCount() - 1; in EvalHifi4() local
220 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in EvalHifi4()
222 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in EvalHifi4()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/ |
| D | softmax.cc | 49 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxQuantized() local
51 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in SoftmaxQuantized()
53 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in SoftmaxQuantized()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | l2norm.cc | 122 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
124 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Eval()
126 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | l2norm.cc | 133 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
135 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Eval()
137 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
|
| D | activations.cc | 1226 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
1228 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmaxEval()
1230 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LogSoftmaxEval()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1737 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
1739 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape);
1741 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim);
1764 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
1766 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
1768 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
3766 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
3768 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
3770 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
3908 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt8LUT() local
[all …]
|
| D | legacy_optimized_ops.h | 4009 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
4011 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
4013 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
|