Home
last modified time | relevance | path

Searched refs:trailing_dim (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h31 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
33 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
35 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
78 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
80 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
82 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
Dreference_ops.h474 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
476 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
478 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
497 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
499 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
501 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2545 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local
2547 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LocalResponseNormalization()
2549 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LocalResponseNormalization()
2570 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dsoftmax.h42 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
44 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
46 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm.cc110 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
112 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Eval()
114 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
Dactivations.cc866 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
868 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmaxEval()
870 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in LogSoftmaxEval()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2734 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
2736 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2738 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
2809 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
2811 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
2813 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
4595 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
4597 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
4599 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
4806 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2566 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
2568 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2570 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
2592 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
2594 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in L2Normalization()
2596 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
4470 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
4472 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
4474 MatchingDim(input_shape, trailing_dim, output_shape, trailing_dim); in Softmax()
4668 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
[all …]