Searched refs:tf_gather_output_rank (Results 1 – 1 of 1) sorted by relevance
3736 const int tf_gather_output_rank = params_tensor.GetTrtDims().nbDims + in ConvertGather() local3738 if (tf_gather_output_rank > nvinfer1::Dims::MAX_DIMS + 1) { in ConvertGather()3760 if (trt_gather_output_dims.nbDims != tf_gather_output_rank - 2) { in ConvertGather()3763 tf_gather_output_rank - 2, in ConvertGather()