Home
last modified time | relevance | path

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

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