Searched refs:trt_shape (Results 1 – 1 of 1) sorted by relevance
453 std::vector<int> trt_shape(dims.nbDims + 1); in ExecuteTrtEngine() local454 trt_shape[0] = num_batch; in ExecuteTrtEngine()455 for (int j = 0; j < dims.nbDims; j++) trt_shape[j + 1] = dims.d[j]; in ExecuteTrtEngine()457 trt_shape.data(), trt_shape.size(), &output_shape); in ExecuteTrtEngine()