Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dsoftmax.h34 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
87 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
171 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt16() local
H A Dl2normalization.h34 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
58 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
H A 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
H A Dreference_ops.h508 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dl2norm.cc133 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
H A Dactivations.cc1313 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1406 const int trailing_dim = input_shape.DimensionsCount() - 1; variable
1433 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
3450 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
3594 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt8LUT() local
3758 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
3804 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
H A Dlegacy_optimized_ops.h4014 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local