Searched refs:remove_tensors_idxs (Results 1 – 1 of 1) sorted by relevance
599 def _remove_tensors_from_model(model, remove_tensors_idxs): argument601 if not remove_tensors_idxs:610 logging.debug("Removing tensors at indices : %s", remove_tensors_idxs)613 if min(remove_tensors_idxs) == len(tensors) - len(remove_tensors_idxs):615 del tensors[min(remove_tensors_idxs):]622 if idx in remove_tensors_idxs:637 for idx in sorted(remove_tensors_idxs, reverse=True):722 remove_tensors_idxs = set()725 remove_tensors_idxs.add(op.inputs[0])728 _remove_tensors_from_model(model, remove_tensors_idxs)[all …]