Searched defs:trailing_dim (Results 1 – 7 of 7) sorted by relevance
31 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local78 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
474 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local497 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local2545 const int trailing_dim = input_shape.DimensionsCount() - 1; in LocalResponseNormalization() local2570 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local2620 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
42 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local
110 const int trailing_dim = input_shape.DimensionsCount() - 1; in Eval() local
866 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmaxEval() local
2566 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local2592 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local4470 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local4668 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local4721 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local
2734 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local2809 const int trailing_dim = input_shape.DimensionsCount() - 1; in L2Normalization() local4595 const int trailing_dim = input_shape.DimensionsCount() - 1; in Softmax() local4806 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local4982 const int trailing_dim = input_shape.DimensionsCount() - 1; in LogSoftmax() local