Searched refs:trt_gather_output_dims (Results 1 – 1 of 1) sorted by relevance
5385 nvinfer1::Dims trt_gather_output_dims = output_tensor->getDimensions(); in ConvertGather() local5392 trt_gather_output_dims.nbDims != expected_trt_output_rank) { in ConvertGather()5396 ", actual nbDims: ", trt_gather_output_dims.nbDims); in ConvertGather()5401 for (int i = trt_gather_output_dims.nbDims; i > trt_axis; --i) { in ConvertGather()5402 trt_gather_output_dims.d[i] = trt_gather_output_dims.d[i - 1]; in ConvertGather()5404 trt_gather_output_dims.d[trt_axis] = 1; in ConvertGather()5405 ++trt_gather_output_dims.nbDims; in ConvertGather()5409 trt_gather_output_dims, in ConvertGather()