Home
last modified time | relevance | path

Searched defs:trailing_dim (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dl2normalization.h34 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
58 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
Dsoftmax.h33 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
86 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
170 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt16() local
Dlog_softmax.h31 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
83 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
Dreference_ops.h885 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm.cc133 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
Dactivations.cc1281 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1735 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
1762 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
3775 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
3917 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt8LUT() local
4081 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
4127 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
Dlegacy_optimized_ops.h4012 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local