Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A DTracerRunner.cpp250 module_runner.for_each_tensor_in_bundled_inputs(consume_tensor); in run_model()
261 module_runner.for_each_tensor_in_bundled_inputs(consume_tensor); in run_model()
H A DMobileModelRunner.h111 void for_each_tensor_in_bundled_inputs(
H A DMobileModelRunner.cpp211 void MobileModelRunner::for_each_tensor_in_bundled_inputs( in for_each_tensor_in_bundled_inputs() function in torch::jit::mobile::MobileModelRunner