Searched refs:detect_and_mark_batch (Results 1 – 1 of 1) sorted by relevance
4185 def detect_and_mark_batch(t): function4198 tree_map_only(torch.Tensor, detect_and_mark_batch, example_inputs)