Searched refs:trt_gather_output_dims (Results 1 – 1 of 1) sorted by relevance
3756 nvinfer1::Dims trt_gather_output_dims = gather_output->getDimensions(); in ConvertGather() local3760 if (trt_gather_output_dims.nbDims != tf_gather_output_rank - 2) { in ConvertGather()3764 ", actual nbDims: ", trt_gather_output_dims.nbDims); in ConvertGather()3768 for (int i = trt_gather_output_dims.nbDims; i > trt_axis; --i) { in ConvertGather()3769 trt_gather_output_dims.d[i] = trt_gather_output_dims.d[i - 1]; in ConvertGather()3771 trt_gather_output_dims.d[trt_axis] = 1; in ConvertGather()3772 ++trt_gather_output_dims.nbDims; in ConvertGather()3776 TRT_TensorOrWeights(gather_output), trt_gather_output_dims, in ConvertGather()