Searched defs:trailing_dim (Results 1 – 10 of 10) sorted by relevance
34 const int trailing_dim = input_shape.DimensionsCount() - 1; variable58 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local
33 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local86 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local170 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt16() local
883 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local908 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local958 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
68 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxHifimini() local194 const int trailing_dim = input_shape.DimensionsCount() - 1; in PrepareHifi4() local218 const int trailing_dim = input_shape.DimensionsCount() - 1; in EvalHifi4() local
49 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxQuantized() local
122 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
133 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
1226 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
1737 const int trailing_dim = input_shape.DimensionsCount() - 1; variable1764 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local3766 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local3908 const int trailing_dim = input_shape.DimensionsCount() - 1; in SoftmaxInt8LUT() local4072 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local4118 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
4009 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local