Home
last modified time | relevance | path

Searched refs:detect_and_mark_batch (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dcommon.py4185 def detect_and_mark_batch(t): function
4198 tree_map_only(torch.Tensor, detect_and_mark_batch, example_inputs)