Searched defs:collect_outputs (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/benchmarks/dynamo/ |
| D | timm_models.py | 396 def forward_pass(self, mod, inputs, collect_outputs=True): argument
400 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): argument
|
| D | huggingface.py | 516 def forward_pass(self, mod, inputs, collect_outputs=True): argument
520 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): argument
|
| D | torchbench.py | 436 def forward_pass(self, mod, inputs, collect_outputs=True): argument
443 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): argument
|
| D | dist_util.py | 83 def model_iter_fn(model, example_inputs, collect_outputs=False): argument
|
| D | common.py | 469 collect_outputs=False, argument
928 def onnxrt_model_iter_fn(model, inputs, collect_outputs=True): argument
941 def onnxrt_model_iter_fn(model, inputs, collect_outputs=True): argument
1222 def opt_aot_inductor(_, example_inputs, collect_outputs=False): argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 129 bool collect_outputs, in connected_subgraph()
|