Home
last modified time | relevance | path

Searched defs:num_models (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/aoti_runtime/
H A Dinterface.cpp53 size_t num_models, in AOTInductorModelContainerCreate()
65 size_t num_models, in AOTInductorModelContainerCreateWithDevice()
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runner/
H A Dmodel_container_runner_cpu.cpp10 size_t num_models) in AOTIModelContainerRunnerCpu()
H A Dmodel_container_runner_cuda.cpp8 size_t num_models, in AOTIModelContainerRunnerCuda()
H A Dmodel_container_runner.cpp36 size_t num_models, in AOTIModelContainerRunner()
/aosp_15_r20/external/pytorch/functorch/notebooks/_src/
H A Dplot_ensembling.py56 num_models = 10 variable
H A Dplot_per_sample_gradients.py51 num_models = 10 variable
/aosp_15_r20/external/pytorch/functorch/examples/ensembling/
H A Dparallel_train.py116 def init_fn(num_models): argument
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dmodel_container.h373 size_t num_models() const { in num_models() function
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_eager_transforms.py4421 def init_fn(num_models): argument
4519 def init_fn(num_models): argument