Searched refs:n_network_inputs (Results 1 – 2 of 2) sorted by relevance
567 const nvinfer1::ICudaEngine* engine, int n_network_inputs) { in SetPrunedMask() argument568 is_pruned_input_.resize(n_network_inputs); in SetPrunedMask()570 for (int j = 0; j < n_network_inputs; j++) { in SetPrunedMask()586 const nvinfer1::ICudaEngine* engine, int n_network_inputs) { in RestoreProfiles() argument599 if (n_inputs > n_network_inputs) { in RestoreProfiles()604 SetShapeTensorMask(engine, n_network_inputs); in RestoreProfiles()606 TF_RETURN_IF_ERROR(SetPrunedMask(engine, n_network_inputs)); in RestoreProfiles()611 cfg.min.resize(n_network_inputs * 2); in RestoreProfiles()612 cfg.max.resize(n_network_inputs * 2); in RestoreProfiles()613 cfg.opt.resize(n_network_inputs * 2); in RestoreProfiles()[all …]
264 int n_network_inputs);333 int n_network_inputs);