Searched refs:num_trt_dims (Results 1 – 1 of 1) sorted by relevance
5647 const int num_trt_dims = tensor->getDimensions().nbDims; in ConvertSoftmax() local5648 if (num_trt_dims == 0) { in ConvertSoftmax()5660 layer->setAxes(1 << (num_trt_dims - 1)); in ConvertSoftmax()