Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | Q | R | S | T | U | V | W | X | Z A Accumulator (class in archai.supergraph.utils.metrics) accuracy() (in module archai.common.ml_utils) Activation() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) activations (archai.supergraph.algos.divnas.divop.DivOp property) add() (archai.supergraph.utils.metrics.Accumulator method) add_constraint() (archai.discrete_search.api.search_objectives.SearchObjectives method) add_dict() (archai.supergraph.utils.metrics.Accumulator method) add_epoch() (archai.supergraph.utils.metrics.RunMetrics method) add_from_cols() (archai.common.delimited_text.DelimitedText method) add_from_cols_list() (archai.common.delimited_text.DelimitedText method) add_from_file() (archai.common.delimited_text.DelimitedText method) add_from_text() (archai.common.delimited_text.DelimitedText method) add_iteration_results() (archai.discrete_search.api.search_results.SearchResults method) add_named_augs() (in module archai.supergraph.datasets.augmentation) add_new_quant_operators() (in module archai.quantization.ptq) add_objective() (archai.discrete_search.api.search_objectives.SearchObjectives method) add_scalar() (archai.common.common.SummaryWriterDummy method) adjust_reshape_and_expand() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) AircraftDatasetProvider (class in archai.datasets.cv.aircraft_dataset_provider) AircraftProvider (class in archai.supergraph.datasets.providers.aircraft_provider) AlexNet (class in archai.supergraph.models.alexnet) alexnet() (in module archai.supergraph.models.alexnet) all_empty() (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) all_evaluated_objs (archai.discrete_search.api.search_results.SearchResults property) all_full() (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) all_owned() (archai.supergraph.nas.arch_module.ArchModule method) all_reduce() (in module archai.common.distributed_utils) alpha (archai.trainers.nlp.hf_training_args.DistillerTrainingArguments attribute) ApexUtils (class in archai.common.apex_utils) append() (archai.supergraph.utils.multi_optim.MultiOptim method) append_csv_file() (in module archai.common.utils) append_to_filename() (in module archai.common.utils) (in module archai.supergraph.algos.petridish.evaluater_petridish) apply_augment() (in module archai.supergraph.datasets.augmentation) apply_rotary_emb_qkv() (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.BaseRotaryEmbedding method) apply_rotary_pos_emb() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) arch_params() (archai.supergraph.nas.arch_module.ArchModule method) archai.api.dataset_provider module archai.api.trainer_base module archai.common.apex_utils module archai.common.atomic_file_handler module archai.common.azureml_helper module archai.common.common module archai.common.config module archai.common.delimited_text module archai.common.deprecation_utils module archai.common.distributed_utils module archai.common.file_utils module archai.common.ml_perf_utils module archai.common.ml_utils module archai.common.model_summary module archai.common.notebook_helper module archai.common.ordered_dict_logger module archai.common.ordered_dict_logger_utils module archai.common.stopwatch module archai.common.store module archai.common.timing module archai.common.utils module archai.common.yaml_utils module archai.datasets.cv.aircraft_dataset_provider module archai.datasets.cv.caltech_dataset_provider module archai.datasets.cv.cifar_dataset_provider module archai.datasets.cv.cityscapes_dataset_provider module archai.datasets.cv.coco_dataset_provider module archai.datasets.cv.flowers102_dataset_provider module archai.datasets.cv.food101_dataset_provider module archai.datasets.cv.image_folder_dataset_provider module archai.datasets.cv.imagenet_dataset_provider module archai.datasets.cv.mnist_dataset_provider module archai.datasets.cv.stanford_cars_dataset_provider module archai.datasets.cv.svhn_dataset_provider module archai.datasets.cv.tensorpack_lmdb_dataset_provider module archai.datasets.cv.tensorpack_lmdb_dataset_provider_utils module archai.datasets.cv.transforms.brightness module archai.datasets.cv.transforms.custom_cutout module archai.datasets.cv.transforms.lighting module archai.datasets.cv.usps_dataset_provider module archai.datasets.nlp.fast_hf_dataset_provider module archai.datasets.nlp.fast_hf_dataset_provider_utils module archai.datasets.nlp.hf_dataset_provider module archai.datasets.nlp.hf_dataset_provider_utils module archai.datasets.nlp.nvidia_data_loader_utils module archai.datasets.nlp.nvidia_dataset_provider module archai.datasets.nlp.nvidia_dataset_provider_utils module archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer module archai.datasets.nlp.tokenizer_utils.gpt2_tokenizer module archai.datasets.nlp.tokenizer_utils.token_config module archai.datasets.nlp.tokenizer_utils.tokenizer_base module archai.datasets.nlp.tokenizer_utils.word_tokenizer module archai.discrete_search.algos.bananas module archai.discrete_search.algos.evolution_pareto module archai.discrete_search.algos.local_search module archai.discrete_search.algos.random_search module archai.discrete_search.algos.regularized_evolution module archai.discrete_search.algos.successive_halving module archai.discrete_search.api.archai_model module archai.discrete_search.api.model_evaluator module archai.discrete_search.api.predictor module archai.discrete_search.api.search_objectives module archai.discrete_search.api.search_results module archai.discrete_search.api.search_space module archai.discrete_search.api.searcher module archai.discrete_search.evaluators.benchmark.natsbench_tss module archai.discrete_search.evaluators.functional module archai.discrete_search.evaluators.nlp.parameters module archai.discrete_search.evaluators.nlp.transformer_flex_latency module archai.discrete_search.evaluators.nlp.transformer_flex_memory module archai.discrete_search.evaluators.onnx_model module archai.discrete_search.evaluators.progressive_training module archai.discrete_search.evaluators.pt_profiler module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_eval module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model module archai.discrete_search.evaluators.ray module archai.discrete_search.evaluators.remote_azure_benchmark module archai.discrete_search.predictors.dnn_ensemble module archai.discrete_search.search_spaces.benchmark.natsbench_tss module archai.discrete_search.search_spaces.config.arch_config module archai.discrete_search.search_spaces.config.arch_param_tree module archai.discrete_search.search_spaces.config.discrete_choice module archai.discrete_search.search_spaces.config.helpers module archai.discrete_search.search_spaces.config.search_space module archai.discrete_search.search_spaces.config.utils module archai.discrete_search.search_spaces.cv.segmentation_dag.model module archai.discrete_search.search_spaces.cv.segmentation_dag.ops module archai.discrete_search.search_spaces.cv.segmentation_dag.search_space module archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block module archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model module archai.discrete_search.search_spaces.nlp.tfpp.mixed_attention module archai.discrete_search.search_spaces.nlp.tfpp.mixed_op module archai.discrete_search.search_spaces.nlp.tfpp.model module archai.discrete_search.search_spaces.nlp.tfpp.ops.causal_self_attn module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention module archai.discrete_search.search_spaces.nlp.tfpp.ops.lsh_attn module archai.discrete_search.search_spaces.nlp.tfpp.ops.mha module archai.discrete_search.search_spaces.nlp.tfpp.ops.sep_conv1d module archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv module archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3 module archai.discrete_search.search_spaces.nlp.tfpp.search_space module archai.discrete_search.search_spaces.nlp.tfpp.utils module archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_gpt2_flex module archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_mem_transformer module archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex module archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer module archai.discrete_search.search_spaces.nlp.transformer_flex.search_space module archai.discrete_search.utils.multi_objective module archai.onnx.config_utils.codegen_onnx_config module archai.onnx.config_utils.gpt2_onnx_config module archai.onnx.config_utils.onnx_config_base module archai.onnx.export module archai.onnx.export_utils module archai.onnx.onnx_forward module archai.onnx.onnx_loader module archai.onnx.optimization module archai.onnx.optimization_utils.fusion_options module archai.onnx.optimization_utils.transfo_xl_onnx_model module archai.quantization.mixed_qat module archai.quantization.modules module archai.quantization.nlp.modules module archai.quantization.observers module archai.quantization.ptq module archai.quantization.qat module archai.quantization.quantization_utils module archai.quantization.quantizers module archai.supergraph.algos.darts.bilevel_arch_trainer module archai.supergraph.algos.darts.bilevel_optimizer module archai.supergraph.algos.darts.bilevel_optimizer_slow module archai.supergraph.algos.darts.darts_exp_runner module archai.supergraph.algos.darts.darts_model_desc_builder module archai.supergraph.algos.darts.mixed_op module archai.supergraph.algos.didarts.didarts_arch_trainer module archai.supergraph.algos.didarts.didarts_exp_runner module archai.supergraph.algos.divnas.analyse_activations module archai.supergraph.algos.divnas.divnas_cell module archai.supergraph.algos.divnas.divnas_exp_runner module archai.supergraph.algos.divnas.divnas_finalizers module archai.supergraph.algos.divnas.divnas_model_desc_builder module archai.supergraph.algos.divnas.divnas_rank_finalizer module archai.supergraph.algos.divnas.divop module archai.supergraph.algos.divnas.seqopt module archai.supergraph.algos.divnas.wmr module archai.supergraph.algos.gumbelsoftmax.gs_arch_trainer module archai.supergraph.algos.gumbelsoftmax.gs_exp_runner module archai.supergraph.algos.gumbelsoftmax.gs_finalizers module archai.supergraph.algos.gumbelsoftmax.gs_model_desc_builder module archai.supergraph.algos.gumbelsoftmax.gs_op module archai.supergraph.algos.manual.manual_evaluater module archai.supergraph.algos.manual.manual_exp_runner module archai.supergraph.algos.manual.manual_searcher module archai.supergraph.algos.nasbench101.base_ops module archai.supergraph.algos.nasbench101.config module archai.supergraph.algos.nasbench101.graph_util module archai.supergraph.algos.nasbench101.model module archai.supergraph.algos.nasbench101.model_builder module archai.supergraph.algos.nasbench101.model_matrix module archai.supergraph.algos.nasbench101.model_metrics_pb2 module archai.supergraph.algos.nasbench101.model_spec module archai.supergraph.algos.nasbench101.nasbench101_exp_runner module archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder module archai.supergraph.algos.nasbench101.nasbench101_op module archai.supergraph.algos.petridish.evaluater_petridish module archai.supergraph.algos.petridish.petridish_exp_runner module archai.supergraph.algos.petridish.petridish_model_desc_builder module archai.supergraph.algos.petridish.petridish_op module archai.supergraph.algos.petridish.petridish_utils module archai.supergraph.algos.petridish.searcher_petridish module archai.supergraph.algos.random.random_exp_runner module archai.supergraph.algos.random.random_model_desc_builder module archai.supergraph.algos.xnas.xnas_arch_trainer module archai.supergraph.algos.xnas.xnas_exp_runner module archai.supergraph.algos.xnas.xnas_model_desc_builder module archai.supergraph.algos.xnas.xnas_op module archai.supergraph.datasets.aug_policies module archai.supergraph.datasets.augmentation module archai.supergraph.datasets.data module archai.supergraph.datasets.dataset_provider module archai.supergraph.datasets.distributed_stratified_sampler module archai.supergraph.datasets.limit_dataset module archai.supergraph.datasets.meta_dataset module archai.supergraph.datasets.providers.aircraft_provider module archai.supergraph.datasets.providers.cifar100_provider module archai.supergraph.datasets.providers.cifar10_provider module archai.supergraph.datasets.providers.fashion_mnist_provider module archai.supergraph.datasets.providers.flower102_provider module archai.supergraph.datasets.providers.food101_provider module archai.supergraph.datasets.providers.imagenet_folder module archai.supergraph.datasets.providers.imagenet_provider module archai.supergraph.datasets.providers.mit67_provider module archai.supergraph.datasets.providers.mnist_provider module archai.supergraph.datasets.providers.person_coco_provider module archai.supergraph.datasets.providers.sport8_provider module archai.supergraph.datasets.providers.stanfordcars_provider module archai.supergraph.datasets.providers.svhn_provider module archai.supergraph.models.alexnet module archai.supergraph.models.densenet module archai.supergraph.models.googlenet module archai.supergraph.models.inception module archai.supergraph.models.mobilenetv2 module archai.supergraph.models.pyramidnet module archai.supergraph.models.resnet module archai.supergraph.models.resnet_orig module archai.supergraph.models.resnet_paper module archai.supergraph.models.shakedrop module archai.supergraph.models.shakeshake.shake_resnet module archai.supergraph.models.shakeshake.shake_resnext module archai.supergraph.models.shakeshake.shakeshake module archai.supergraph.models.vgg module archai.supergraph.models.wideresnet module archai.supergraph.nas.arch_module module archai.supergraph.nas.arch_params module archai.supergraph.nas.arch_trainer module archai.supergraph.nas.cell module archai.supergraph.nas.dag_edge module archai.supergraph.nas.evaluater module archai.supergraph.nas.exp_runner module archai.supergraph.nas.finalizers module archai.supergraph.nas.model module archai.supergraph.nas.model_desc module archai.supergraph.nas.model_desc_builder module archai.supergraph.nas.nas_utils module archai.supergraph.nas.operations module archai.supergraph.nas.random_finalizers module archai.supergraph.nas.search_combinations module archai.supergraph.nas.searcher module archai.supergraph.nas.vis_model_desc module archai.supergraph.utils.augmented_searcher module archai.supergraph.utils.augmented_trainer module archai.supergraph.utils.checkpoint module archai.supergraph.utils.heatmap module archai.supergraph.utils.metrics module archai.supergraph.utils.multi_optim module archai.supergraph.utils.tester module archai.supergraph.utils.trainer module archai.trainers.coin_betting_optimizer module archai.trainers.cv.pl_trainer module archai.trainers.cyclic_cosine_scheduler module archai.trainers.gradual_warmup_scheduler module archai.trainers.lamb_optimizer module archai.trainers.losses module archai.trainers.nlp.ds_trainer module archai.trainers.nlp.ds_training_args module archai.trainers.nlp.hf_callbacks module archai.trainers.nlp.hf_trainer module archai.trainers.nlp.hf_training_args module archai.trainers.nlp.nvidia_trainer module archai.trainers.nlp.nvidia_training_args module ArchaiModel (class in archai.discrete_search.api.archai_model) ArchaiStore (class in archai.common.store) ArchConfig (class in archai.discrete_search.search_spaces.config.arch_config) ArchConfigList (class in archai.discrete_search.search_spaces.config.arch_config) ArchModule (class in archai.supergraph.nas.arch_module) ArchParams (class in archai.supergraph.nas.arch_params) ArchParamTree (class in archai.discrete_search.search_spaces.config.arch_param_tree) ArchTrainer (class in archai.supergraph.nas.arch_trainer) arsaug_policy() (in module archai.supergraph.datasets.augmentation) AsyncModelEvaluator (class in archai.discrete_search.api.model_evaluator) AtomicFileHandler (class in archai.common.atomic_file_handler) AttentionMask (archai.onnx.optimization_utils.fusion_options.AttentionMaskFormat attribute) AttentionMaskFormat (class in archai.onnx.optimization_utils.fusion_options) attentions (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerBaseOutput attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) attr_to_dict() (in module archai.common.utils) augment_list() (in module archai.supergraph.datasets.augmentation) Augmentation (class in archai.supergraph.datasets.augmentation) autoaug2arsaug() (in module archai.supergraph.datasets.augmentation) autoaug_paper_cifar10() (in module archai.supergraph.datasets.augmentation) autoaug_policy() (in module archai.supergraph.datasets.augmentation) autocast() (archai.common.apex_utils.ApexUtils method) AutoContrast() (in module archai.supergraph.datasets.augmentation) AuxTower (class in archai.supergraph.nas.model) AuxTowerDesc (class in archai.supergraph.nas.model_desc) AverageMeter (class in archai.common.utils) AvgOnnxLatency (class in archai.discrete_search.evaluators.onnx_model) AvgPool2d7x7 (class in archai.supergraph.nas.operations) B backend (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) backward() (archai.common.apex_utils.ApexUtils method) (archai.supergraph.models.shakedrop.ShakeDropFunction static method) (archai.supergraph.models.shakeshake.shakeshake.ShakeShake static method) barrier() (archai.common.apex_utils.ApexUtils method) (in module archai.common.distributed_utils) BaseRotaryEmbedding (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.mha) BasicBlock (class in archai.supergraph.models.pyramidnet) batch_upload() (archai.common.store.ArchaiStore method) (in module archai.common.store) BayesOptSearchSpace (class in archai.discrete_search.api.search_space) BbpeTokenizer (class in archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer) best_epoch() (archai.supergraph.utils.metrics.RunMetrics method) best_test_top1() (archai.supergraph.utils.metrics.Metrics method) best_train_top1() (archai.supergraph.utils.metrics.Metrics method) best_val_top1() (archai.supergraph.utils.metrics.Metrics method) bias (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) BilevelArchTrainer (class in archai.supergraph.algos.darts.bilevel_arch_trainer) BilevelOptimizer (class in archai.supergraph.algos.darts.bilevel_optimizer) (class in archai.supergraph.algos.darts.bilevel_optimizer_slow) Block (class in archai.discrete_search.search_spaces.cv.segmentation_dag.ops) BOS (archai.datasets.nlp.tokenizer_utils.token_config.SpecialTokenEnum attribute) Bottleneck (class in archai.supergraph.models.pyramidnet) BPCTrainerCallback (class in archai.trainers.nlp.hf_callbacks) Brightness (class in archai.datasets.cv.transforms.brightness) Brightness() (in module archai.supergraph.datasets.augmentation) build() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) (in module archai.supergraph.algos.nasbench101.model_builder) build_arch_config() (in module archai.discrete_search.search_spaces.config.arch_config) build_aux_tower() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_cell() (archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder.NasBench101CellBuilder method) (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_cell_post_op() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_cell_stems() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_cells() (archai.supergraph.algos.random.random_model_desc_builder.RandomModelDescBuilder method) (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_config() (in module archai.supergraph.algos.nasbench101.config) build_logits_op() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_model_desc() (archai.supergraph.algos.petridish.searcher_petridish.SearcherPetridish method) (archai.supergraph.nas.searcher.Searcher method) build_model_pool() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_model_stems() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_nodes() (archai.supergraph.algos.darts.darts_model_desc_builder.DartsModelDescBuilder method) (archai.supergraph.algos.divnas.divnas_model_desc_builder.DivnasModelDescBuilder method) (archai.supergraph.algos.gumbelsoftmax.gs_model_desc_builder.GsModelDescBuilder method) (archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder.NasBench101CellBuilder method) (archai.supergraph.algos.random.random_model_desc_builder.RandomModelDescBuilder method) (archai.supergraph.algos.xnas.xnas_model_desc_builder.XnasModelDescBuilder method) (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) build_nodes_from_template() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) C calculate_onnx_model_size() (in module archai.common.file_utils) calculate_qparams() (archai.quantization.observers.OnnxDynamicObserver method) calculate_torch_model_size() (in module archai.common.file_utils) CaltechDatasetProvider (class in archai.datasets.cv.caltech_dataset_provider) can_drop_path() (archai.supergraph.algos.darts.mixed_op.MixedOp method) (archai.supergraph.algos.divnas.divop.DivOp method) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp method) (archai.supergraph.algos.xnas.xnas_op.XnasOp method) (archai.supergraph.nas.operations.AvgPool2d7x7 method) (archai.supergraph.nas.operations.Identity method) (archai.supergraph.nas.operations.LinearOp method) (archai.supergraph.nas.operations.MergeOp method) (archai.supergraph.nas.operations.Op method) (archai.supergraph.nas.operations.PoolAdaptiveAvg2D method) (archai.supergraph.nas.operations.PoolMeanTensor method) (archai.supergraph.nas.operations.SkipConnect method) (archai.supergraph.nas.operations.StemConv3x3 method) (archai.supergraph.nas.operations.StemConv3x3Relu method) (archai.supergraph.nas.operations.StemConv3x3S4 method) (archai.supergraph.nas.operations.StemConv3x3S4S2 method) CausalSelfAttention (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.causal_self_attn) Cell (class in archai.supergraph.algos.nasbench101.model) (class in archai.supergraph.nas.cell) cell_descs() (archai.supergraph.nas.model_desc.ModelDesc method) cell_type_count() (archai.supergraph.nas.model_desc.ModelDesc method) CellDesc (class in archai.supergraph.nas.model_desc) CellType (class in archai.supergraph.nas.model_desc) change_filepath_ext() (in module archai.common.utils) change_filepath_name() (in module archai.common.utils) change_graph_input_type() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) change_graph_inputs_to_int32() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) channel_norm() (in module archai.common.ml_utils) cheap_objective_names (archai.discrete_search.api.search_objectives.SearchObjectives property) cheap_objectives (archai.discrete_search.api.search_objectives.SearchObjectives property) check_available_checkpoint() (in module archai.common.file_utils) CheckPoint (class in archai.supergraph.utils.checkpoint) checkpoint_empty() (in module archai.supergraph.nas.nas_utils) checkpoint_file_path (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) Cifar100Provider (class in archai.supergraph.datasets.providers.cifar100_provider) Cifar10Provider (class in archai.supergraph.datasets.providers.cifar10_provider) CifarDatasetProvider (class in archai.datasets.cv.cifar_dataset_provider) CityscapesDatasetProvider (class in archai.datasets.cv.cityscapes_dataset_provider) clean_ensure_expdir() (in module archai.common.common) clean_graph() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) clean_log_result() (archai.supergraph.nas.searcher.Searcher method) clear() (archai.common.stopwatch.StopWatch method) (archai.discrete_search.api.archai_model.ArchaiModel method) clear_collect_activations() (archai.supergraph.algos.divnas.divnas_cell.Divnas_Cell method) clear_gpu_memory() (in module archai.common.ml_utils) clear_trainables() (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.EdgeDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) (archai.supergraph.nas.model_desc.NodeDesc method) (archai.supergraph.nas.model_desc.OpDesc method) clip_grad() (archai.common.apex_utils.ApexUtils method) clone() (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.ConvMacroParams method) (archai.supergraph.nas.model_desc.EdgeDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) (archai.supergraph.nas.model_desc.NodeDesc method) (archai.supergraph.nas.model_desc.OpDesc method) close() (archai.common.atomic_file_handler.AtomicFileHandler method) (archai.common.ordered_dict_logger.OrderedDictLogger method) CocobBackprop (class in archai.trainers.coin_betting_optimizer) CocobOns (class in archai.trainers.coin_betting_optimizer) CocoDatasetProvider (class in archai.datasets.cv.coco_dataset_provider) CodeGenBlock (class in archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block) CodeGenForCausalLM (class in archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model) CodeGenModel (class in archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model) CodeGenOnnxConfig (class in archai.onnx.config_utils.codegen_onnx_config) collect_activations (archai.supergraph.algos.divnas.divop.DivOp property) collect_activations() (archai.supergraph.algos.divnas.divnas_cell.Divnas_Cell method) collect_features() (in module archai.supergraph.algos.divnas.analyse_activations) Color() (in module archai.supergraph.datasets.augmentation) commit() (archai.supergraph.utils.checkpoint.CheckPoint method) common_init() (in module archai.common.common) CommonState (class in archai.common.common) compute_brute_force_sol() (in module archai.supergraph.algos.divnas.analyse_activations) compute_correlation() (in module archai.supergraph.algos.divnas.analyse_activations) compute_covariance_offline() (in module archai.supergraph.algos.divnas.analyse_activations) compute_euclidean_dist_quantiles() (in module archai.supergraph.algos.divnas.analyse_activations) compute_loss() (archai.supergraph.nas.arch_trainer.ArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) (archai.trainers.nlp.hf_trainer.HfDistillerTrainer method) compute_marginal_gain() (in module archai.supergraph.algos.divnas.analyse_activations) compute_rbf_kernel_covariance() (in module archai.supergraph.algos.divnas.analyse_activations) ComputeVertexChannels() (in module archai.supergraph.algos.nasbench101.model) ConcateChannelsOp (class in archai.supergraph.nas.operations) Config (class in archai.common.config) config_class (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexLMHeadModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerLMHeadModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerModel attribute) ConfigSearchSpace (class in archai.discrete_search.search_spaces.config.search_space) constraints (archai.discrete_search.api.search_objectives.SearchObjectives property) Contrast() (in module archai.supergraph.datasets.augmentation) Conv1x1BnRelu (class in archai.supergraph.algos.nasbench101.base_ops) conv3x3() (in module archai.supergraph.models.pyramidnet) (in module archai.supergraph.models.wideresnet) Conv3x3BnRelu (class in archai.supergraph.algos.nasbench101.base_ops) conv_init() (in module archai.supergraph.models.wideresnet) ConvBnRelu (class in archai.supergraph.algos.nasbench101.base_ops) ConvBNReLU (class in archai.supergraph.nas.operations) ConvexHullPoint (class in archai.supergraph.algos.petridish.petridish_utils) ConvMacroParams (class in archai.supergraph.nas.model_desc) copy_dir() (in module archai.common.utils) (in module archai.supergraph.algos.petridish.petridish_exp_runner) copy_file() (in module archai.common.file_utils) (in module archai.common.utils) copy_file_basic() (in module archai.common.utils) copy_search_to_eval() (archai.supergraph.algos.manual.manual_exp_runner.ManualExperimentRunner method) (archai.supergraph.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) Corpus (class in archai.datasets.nlp.nvidia_dataset_provider_utils) create() (archai.supergraph.nas.operations.Op static method) create_arch_params() (archai.supergraph.nas.arch_module.ArchModule method) create_cell_templates() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) create_checkpoint() (in module archai.supergraph.nas.nas_utils) create_compute_cluster() (in module archai.common.azureml_helper) create_conf() (in module archai.common.common) create_dataset_provider() (in module archai.supergraph.datasets.data) create_dirs() (in module archai.common.common) create_empty_file() (in module archai.common.file_utils) create_environment_from_file() (in module archai.common.azureml_helper) create_file_name_identifier() (in module archai.common.file_utils) create_file_with_string() (in module archai.common.file_utils) create_lr_scheduler() (in module archai.common.ml_utils) create_model() (archai.supergraph.algos.manual.manual_evaluater.ManualEvaluater method) (archai.supergraph.nas.evaluater.Evaluater method) create_multi_optim() (archai.supergraph.algos.didarts.didarts_arch_trainer.DidartsArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) create_optimizer() (archai.supergraph.algos.gumbelsoftmax.gs_arch_trainer.GsArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) (in module archai.common.ml_utils) create_scheduler() (archai.supergraph.utils.trainer.Trainer method) create_submod_f() (in module archai.supergraph.algos.divnas.analyse_activations) create_tb_writer() (in module archai.common.common) crossover() (archai.discrete_search.api.search_space.EvolutionarySearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) crossover_parents() (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) cuda_device_names() (in module archai.common.utils) cur_epoch() (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.metrics.RunMetrics method) current_node (archai.common.ordered_dict_logger.OrderedDictLogger property) current_path (archai.common.ordered_dict_logger.OrderedDictLogger property) CustomCutout (class in archai.datasets.cv.transforms.custom_cutout) Cutout() (in module archai.supergraph.datasets.augmentation) CutoutAbs() (in module archai.supergraph.datasets.augmentation) CyclicCosineDecayLR (class in archai.trainers.cyclic_cosine_scheduler) D d_output (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv property) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 property) d_state (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv property) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 property) DagEdge (class in archai.supergraph.nas.dag_edge) DartsExperimentRunner (class in archai.supergraph.algos.darts.darts_exp_runner) DartsModelDescBuilder (class in archai.supergraph.algos.darts.darts_model_desc_builder) data_parallel_rank (archai.trainers.nlp.ds_trainer.DsTrainer property) data_parallel_world_size (archai.trainers.nlp.ds_trainer.DsTrainer property) dataloader_num_workers (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) dataloader_pin_memory (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) DataLoaders (class in archai.supergraph.datasets.data) dataset_cache_dir (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) dataset_dir (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) dataset_name (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) dataset_refresh_cache (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) DatasetProvider (class in archai.api.dataset_provider) (class in archai.supergraph.datasets.dataset_provider) debug() (archai.common.ordered_dict_logger.OrderedDictLogger method) decode_text() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) dedup_list() (in module archai.common.utils) deep_comp() (in module archai.common.utils) deep_update() (in module archai.common.config) (in module archai.common.utils) default_dataroot() (in module archai.common.common) DEFAULT_TASK_OUTPUTS (archai.onnx.config_utils.onnx_config_base.OnnxConfig attribute) delete() (in module archai.common.store) delete_blobs() (archai.common.store.ArchaiStore method) delete_file() (in module archai.common.utils) delete_status() (archai.common.store.ArchaiStore method) delete_status_entity() (archai.common.store.ArchaiStore method) DelimitedText (class in archai.common.delimited_text) DenseNet (class in archai.supergraph.models.densenet) densenet121() (in module archai.supergraph.models.densenet) densenet161() (in module archai.supergraph.models.densenet) densenet169() (in module archai.supergraph.models.densenet) densenet201() (in module archai.supergraph.models.densenet) deprecated() (in module archai.common.deprecation_utils) device (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments property) device_type() (archai.supergraph.nas.model.Model method) DidartsArchTrainer (class in archai.supergraph.algos.didarts.didarts_arch_trainer) DiDartsExperimentRunner (class in archai.supergraph.algos.didarts.didarts_exp_runner) dilation (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) DilConv (class in archai.supergraph.nas.operations) dir_downloads() (in module archai.common.utils) disable_attention_mask() (archai.onnx.optimization_utils.fusion_options.FusionOptions method) disable_functional_hooks() (in module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks) disable_tensor_hooks() (in module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks) DiscreteChoice (class in archai.discrete_search.search_spaces.config.discrete_choice) DiscreteSearchSpace (class in archai.discrete_search.api.search_space) DistillerTrainingArguments (class in archai.trainers.nlp.hf_training_args) DistributedStratifiedSampler (class in archai.supergraph.datasets.distributed_stratified_sampler) Divnas_Cell (class in archai.supergraph.algos.divnas.divnas_cell) DivnasExperimentRunner (class in archai.supergraph.algos.divnas.divnas_exp_runner) DivnasFinalizers (class in archai.supergraph.algos.divnas.divnas_finalizers) DivnasModelDescBuilder (class in archai.supergraph.algos.divnas.divnas_model_desc_builder) DivnasRankFinalizers (class in archai.supergraph.algos.divnas.divnas_rank_finalizer) DivOp (class in archai.supergraph.algos.divnas.divop) do_eval (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) download() (archai.common.store.ArchaiStore method) (archai.supergraph.datasets.providers.imagenet_folder.ImageNetFolder method) (in module archai.common.store) download_and_extract_tar() (in module archai.common.utils) download_and_extract_zip() (in module archai.common.utils) download_job_output() (in module archai.common.azureml_helper) draw_cell_desc() (in module archai.supergraph.nas.vis_model_desc) draw_model_desc() (in module archai.supergraph.nas.vis_model_desc) drop_path_() (in module archai.common.ml_utils) drop_path_prob() (archai.supergraph.nas.model.Model method) DropPath_ (class in archai.supergraph.nas.operations) ds_config (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) DsTrainer (class in archai.trainers.nlp.ds_trainer) DsTrainingArguments (class in archai.trainers.nlp.ds_training_args) duration() (archai.supergraph.utils.metrics.EpochMetrics method) (archai.supergraph.utils.metrics.RunMetrics method) dynamic_quantization_onnx() (in module archai.quantization.ptq) dynamic_quantization_torch() (in module archai.quantization.ptq) E EdgeDesc (class in archai.supergraph.nas.model_desc) elapsed (archai.common.timing.MeasureBlockTime property) embedding_dim (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) emit() (archai.common.atomic_file_handler.AtomicFileHandler method) empty() (archai.supergraph.nas.arch_params.ArchParams static method) enable_functional_hooks() (in module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks) enable_tensor_hooks() (in module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks) encode() (archai.discrete_search.api.search_space.BayesOptSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.discrete_choice.DiscreteChoice method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) encode_config() (archai.discrete_search.search_spaces.config.arch_param_tree.ArchParamTree method) encode_dataset() (in module archai.datasets.nlp.hf_dataset_provider_utils) encode_file() (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) encode_text() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) end() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) ensure_pytorch_ver() (in module archai.common.ml_utils) EOS (archai.datasets.nlp.tokenizer_utils.token_config.SpecialTokenEnum attribute) epoch() (archai.supergraph.utils.multi_optim.MultiOptim method) (archai.supergraph.utils.trainer.Trainer method) epoch_time_avg() (archai.supergraph.utils.metrics.RunMetrics method) EpochMetrics (class in archai.supergraph.utils.metrics) epochs() (archai.supergraph.utils.metrics.Metrics method) Equalize() (in module archai.supergraph.datasets.augmentation) error() (archai.common.ordered_dict_logger.OrderedDictLogger method) EVAL (archai.supergraph.algos.petridish.petridish_utils.ExperimentStage attribute) (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) eval_all_objs() (archai.discrete_search.api.search_objectives.SearchObjectives method) eval_batch_without_pipe_parallel() (archai.trainers.nlp.ds_trainer.DsTrainer method) eval_cheap_objs() (archai.discrete_search.api.search_objectives.SearchObjectives method) eval_expensive_objs() (archai.discrete_search.api.search_objectives.SearchObjectives method) eval_max_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) eval_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) EVAL_TRAINED (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) EvalResult (class in archai.supergraph.nas.evaluater) evaluate() (archai.api.trainer_base.TrainerBase method) (archai.discrete_search.api.model_evaluator.ModelEvaluator method) (archai.discrete_search.evaluators.benchmark.natsbench_tss.NatsbenchMetric method) (archai.discrete_search.evaluators.functional.EvaluationFunction method) (archai.discrete_search.evaluators.nlp.parameters.NonEmbeddingParamsProxy method) (archai.discrete_search.evaluators.nlp.transformer_flex_latency.TransformerFlexOnnxLatency method) (archai.discrete_search.evaluators.nlp.transformer_flex_memory.TransformerFlexOnnxMemory method) (archai.discrete_search.evaluators.onnx_model.AvgOnnxLatency method) (archai.discrete_search.evaluators.progressive_training.ProgressiveTraining method) (archai.discrete_search.evaluators.pt_profiler.TorchFlops method) (archai.discrete_search.evaluators.pt_profiler.TorchLatency method) (archai.discrete_search.evaluators.pt_profiler.TorchMacs method) (archai.discrete_search.evaluators.pt_profiler.TorchNumParameters method) (archai.discrete_search.evaluators.pt_profiler.TorchPeakCpuMemory method) (archai.discrete_search.evaluators.pt_profiler.TorchPeakCudaMemory method) (archai.supergraph.algos.petridish.evaluater_petridish.EvaluaterPetridish method) (archai.supergraph.nas.evaluater.Evaluater method) (archai.trainers.cv.pl_trainer.PlTrainer method) (archai.trainers.nlp.ds_trainer.DsTrainer method) (archai.trainers.nlp.nvidia_trainer.NvidiaTrainer method) Evaluater (class in archai.supergraph.nas.evaluater) evaluater() (archai.supergraph.algos.manual.manual_exp_runner.ManualExperimentRunner method) (archai.supergraph.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) EvaluaterPetridish (class in archai.supergraph.algos.petridish.evaluater_petridish) EvaluationFunction (class in archai.discrete_search.evaluators.functional) EvolutionarySearchSpace (class in archai.discrete_search.api.search_space) EvolutionParetoSearch (class in archai.discrete_search.algos.evolution_pareto) exec_shell_command() (in module archai.common.utils) expdir_abspath() (in module archai.common.common) expensive_objective_names (archai.discrete_search.api.search_objectives.SearchObjectives property) expensive_objectives (archai.discrete_search.api.search_objectives.SearchObjectives property) experiment_name (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) ExperimentRunner (class in archai.supergraph.nas.exp_runner) ExperimentStage (class in archai.supergraph.algos.petridish.petridish_utils) export_to_onnx() (in module archai.onnx.export) extra_repr() (archai.supergraph.datasets.providers.imagenet_folder.ImageNetFolder method) (archai.supergraph.nas.operations.DropPath_ method) extract_tar() (in module archai.common.utils) extract_zip() (in module archai.common.utils) F fa_reduced_cifar10() (in module archai.supergraph.datasets.aug_policies) fa_reduced_svhn() (in module archai.supergraph.datasets.aug_policies) fa_resnet50_rimagenet() (in module archai.supergraph.datasets.aug_policies) FacConv (class in archai.supergraph.nas.operations) FactorizedReduce (class in archai.supergraph.nas.operations) fake_quant_weight (archai.quantization.modules.FakeDynamicQuantConv1d property) (archai.quantization.modules.FakeDynamicQuantLinear property) (archai.quantization.modules.FakeQuantEmbedding property) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1D property) FakeDynamicQuant (class in archai.quantization.quantizers) FakeDynamicQuantConv1d (class in archai.quantization.modules) FakeDynamicQuantConv1dForOnnx (class in archai.quantization.modules) FakeDynamicQuantHFConv1D (class in archai.quantization.nlp.modules) FakeDynamicQuantHFConv1DForOnnx (class in archai.quantization.nlp.modules) FakeDynamicQuantLinear (class in archai.quantization.modules) FakeDynamicQuantLinearForOnnx (class in archai.quantization.modules) FakeQuantEmbedding (class in archai.quantization.modules) FakeQuantEmbeddingForOnnx (class in archai.quantization.modules) FashionMnistProvider (class in archai.supergraph.datasets.providers.fashion_mnist_provider) FastDataCollatorForLanguageModeling (class in archai.datasets.nlp.fast_hf_dataset_provider) FastHfDataset (class in archai.datasets.nlp.fast_hf_dataset_provider_utils) FastHfDatasetProvider (class in archai.datasets.nlp.fast_hf_dataset_provider) fetch_all() (archai.discrete_search.api.model_evaluator.AsyncModelEvaluator method) (archai.discrete_search.evaluators.progressive_training.RayProgressiveTraining method) (archai.discrete_search.evaluators.ray.RayParallelEvaluator method) (archai.discrete_search.evaluators.remote_azure_benchmark.RemoteAzureBenchmarkEvaluator method) FFEnsembleMember (class in archai.discrete_search.predictors.dnn_ensemble) fft_conv() (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv method) filepath_ext() (in module archai.common.utils) (in module archai.supergraph.algos.petridish.evaluater_petridish) filepath_name_ext() (in module archai.common.utils) filepath_name_only() (in module archai.common.utils) (in module archai.supergraph.algos.petridish.evaluater_petridish) filepath_without_ext() (in module archai.common.utils) filter() (archai.common.ordered_dict_logger_utils.RankFilter method) finalize() (archai.supergraph.algos.darts.mixed_op.MixedOp method) (archai.supergraph.algos.divnas.divop.DivOp method) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp method) (archai.supergraph.algos.petridish.petridish_op.PetridishOp method) (archai.supergraph.algos.xnas.xnas_op.XnasOp method) (archai.supergraph.nas.operations.Op method) finalize_cell() (archai.supergraph.algos.divnas.divnas_finalizers.DivnasFinalizers method) (archai.supergraph.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method) (archai.supergraph.nas.finalizers.Finalizers method) finalize_cells() (archai.supergraph.nas.finalizers.Finalizers method) finalize_edge() (archai.supergraph.nas.finalizers.Finalizers method) finalize_model() (archai.supergraph.algos.divnas.divnas_finalizers.DivnasFinalizers method) (archai.supergraph.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method) (archai.supergraph.nas.finalizers.Finalizers method) (archai.supergraph.nas.searcher.Searcher method) finalize_node() (archai.supergraph.algos.divnas.divnas_finalizers.DivnasFinalizers method) (archai.supergraph.algos.divnas.divnas_rank_finalizer.DivnasRankFinalizers method) (archai.supergraph.algos.gumbelsoftmax.gs_finalizers.GsFinalizers method) (archai.supergraph.nas.finalizers.Finalizers method) (archai.supergraph.nas.random_finalizers.RandomFinalizers method) Finalizers (class in archai.supergraph.nas.finalizers) finalizers() (archai.supergraph.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method) (archai.supergraph.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) find_unused_parameters (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) fine_tune_qat() (archai.trainers.nlp.nvidia_trainer.NvidiaTrainer method) first_or_default() (in module archai.common.utils) fit() (archai.discrete_search.api.predictor.Predictor method) (archai.discrete_search.predictors.dnn_ensemble.PredictiveDNNEnsemble method) (archai.supergraph.utils.trainer.Trainer method) flatten_dict() (in module archai.discrete_search.search_spaces.config.utils) Flip() (in module archai.supergraph.datasets.augmentation) float_parameter() (in module archai.supergraph.datasets.augmentation) float_to_qat_modules() (in module archai.quantization.qat) Flower102Provider (class in archai.supergraph.datasets.providers.flower102_provider) Flowers102DatasetProvider (class in archai.datasets.cv.flowers102_dataset_provider) flush() (archai.common.atomic_file_handler.AtomicFileHandler method) (archai.common.common.SummaryWriterDummy method) fmt() (in module archai.common.utils) Food101DatasetProvider (class in archai.datasets.cv.food101_dataset_provider) Food101Provider (class in archai.supergraph.datasets.providers.food101_provider) forward() (archai.discrete_search.predictors.dnn_ensemble.FFEnsembleMember method) (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.Block method) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.NormalConvBlock method) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.SeparableConvBlock method) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block.CodeGenBlock method) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block.Mlp method) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenForCausalLM method) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenModel method) (archai.discrete_search.search_spaces.nlp.tfpp.mixed_attention.MixedAttentionBlock method) (archai.discrete_search.search_spaces.nlp.tfpp.mixed_op.MixedAttentionBlock method) (archai.discrete_search.search_spaces.nlp.tfpp.model.LanguageModel method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.causal_self_attn.CausalSelfAttention method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.LocalAttention method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.LocalMHA method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.SinusoidalEmbeddings method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.lsh_attn.LSHAttention method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.BaseRotaryEmbedding method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.MHA method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.SelfAttention method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sep_conv1d.SeparableConv1d method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.modrelu method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.Normalization method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.SGConv method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.TransposedLinear method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.TransposedLN method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.SGConv3 method) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexMLP method) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerLMHeadModel method) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerModel method) (archai.quantization.mixed_qat.MixedQAT method) (archai.quantization.modules.FakeDynamicQuantConv1d method) (archai.quantization.modules.FakeDynamicQuantLinear method) (archai.quantization.modules.FakeQuantEmbedding method) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1D method) (archai.quantization.quantizers.FakeDynamicQuant method) (archai.supergraph.algos.darts.mixed_op.MixedOp method) (archai.supergraph.algos.divnas.divop.DivOp method) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp method) (archai.supergraph.algos.nasbench101.base_ops.Conv1x1BnRelu method) (archai.supergraph.algos.nasbench101.base_ops.Conv3x3BnRelu method) (archai.supergraph.algos.nasbench101.base_ops.ConvBnRelu method) (archai.supergraph.algos.nasbench101.base_ops.MaxPool3x3 method) (archai.supergraph.algos.nasbench101.model.Cell method) (archai.supergraph.algos.nasbench101.model.Network method) (archai.supergraph.algos.nasbench101.nasbench101_op.NasBench101Op method) (archai.supergraph.algos.petridish.petridish_op.PetridishOp method) (archai.supergraph.algos.petridish.petridish_op.StopForward method) (archai.supergraph.algos.petridish.petridish_op.StopForwardReductionOp method) (archai.supergraph.algos.petridish.petridish_op.StopGradient method) (archai.supergraph.algos.petridish.petridish_op.StopGradientReduction method) (archai.supergraph.algos.petridish.petridish_op.TempIdentityOp method) (archai.supergraph.algos.xnas.xnas_op.XnasOp method) (archai.supergraph.models.alexnet.AlexNet method) (archai.supergraph.models.densenet.DenseNet method) (archai.supergraph.models.googlenet.GoogLeNet method) (archai.supergraph.models.inception.Inception3 method) (archai.supergraph.models.mobilenetv2.MobileNetV2 method) (archai.supergraph.models.pyramidnet.BasicBlock method) (archai.supergraph.models.pyramidnet.Bottleneck method) (archai.supergraph.models.pyramidnet.PyramidNet method) (archai.supergraph.models.resnet.ResNet method) (archai.supergraph.models.resnet_paper.ResNet method) (archai.supergraph.models.shakedrop.ShakeDrop method) (archai.supergraph.models.shakedrop.ShakeDropFunction static method) (archai.supergraph.models.shakeshake.shake_resnet.ShakeBlock method) (archai.supergraph.models.shakeshake.shake_resnet.ShakeResNet method) (archai.supergraph.models.shakeshake.shake_resnext.ShakeBottleNeck method) (archai.supergraph.models.shakeshake.shake_resnext.ShakeResNeXt method) (archai.supergraph.models.shakeshake.shakeshake.ShakeShake static method) (archai.supergraph.models.shakeshake.shakeshake.Shortcut method) (archai.supergraph.models.vgg.VGG method) (archai.supergraph.models.wideresnet.WideBasic method) (archai.supergraph.models.wideresnet.WideResNet method) (archai.supergraph.nas.cell.Cell method) (archai.supergraph.nas.dag_edge.DagEdge method) (archai.supergraph.nas.model.AuxTower method) (archai.supergraph.nas.model.Model method) (archai.supergraph.nas.operations.AvgPool2d7x7 method) (archai.supergraph.nas.operations.ConcateChannelsOp method) (archai.supergraph.nas.operations.ConvBNReLU method) (archai.supergraph.nas.operations.DilConv method) (archai.supergraph.nas.operations.DropPath_ method) (archai.supergraph.nas.operations.FacConv method) (archai.supergraph.nas.operations.FactorizedReduce method) (archai.supergraph.nas.operations.Identity method) (archai.supergraph.nas.operations.LinearOp method) (archai.supergraph.nas.operations.MergeOp method) (archai.supergraph.nas.operations.MultiOp method) (archai.supergraph.nas.operations.PoolAdaptiveAvg2D method) (archai.supergraph.nas.operations.PoolBN method) (archai.supergraph.nas.operations.PoolMeanTensor method) (archai.supergraph.nas.operations.ProjectChannelsOp method) (archai.supergraph.nas.operations.ReLUConvBN method) (archai.supergraph.nas.operations.SepConv method) (archai.supergraph.nas.operations.SkipConnect method) (archai.supergraph.nas.operations.StemConv3x3 method) (archai.supergraph.nas.operations.StemConv3x3Relu method) (archai.supergraph.nas.operations.StemConv3x3S4 method) (archai.supergraph.nas.operations.StemConv3x3S4S2 method) (archai.supergraph.nas.operations.Zero method) (archai.trainers.losses.SmoothCrossEntropyLoss method) fp16 (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) freeze (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) from_cache() (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider class method) from_disk() (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider class method) from_file() (archai.discrete_search.search_spaces.config.arch_config.ArchConfig class method) (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel class method) from_float() (archai.quantization.modules.FakeDynamicQuantConv1d class method) (archai.quantization.modules.FakeDynamicQuantLinear class method) (archai.quantization.modules.FakeQuantEmbedding class method) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1D class method) from_hub() (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider class method) from_json_file() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) from_module() (archai.supergraph.nas.arch_params.ArchParams static method) from_yaml_file() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) full_path() (in module archai.common.utils) fuse_add_bias_skip_layer_norm() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) fuse_attention() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) fuse_layer_norm() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) fuse_reshape() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) fuse_shape() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) fuse_skip_layer_norm() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) FusionOptions (class in archai.onnx.optimization_utils.fusion_options) G GConv (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) GConv3 (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3) GemmQuant (class in archai.quantization.ptq) gen_is_edge_fn() (in module archai.supergraph.algos.nasbench101.graph_util) generate_dummy_inputs() (archai.onnx.config_utils.gpt2_onnx_config.GPT2FlexOnnxConfig method) (archai.onnx.config_utils.onnx_config_base.OnnxConfig method) (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast method) get() (archai.common.stopwatch.StopWatch static method) get_all_status_entities() (archai.common.store.ArchaiStore method) get_all_tensor_stats() (in module archai.common.model_summary) get_aml_client_from_file() (in module archai.common.azureml_helper) get_arch_abs_path() (in module archai.common.notebook_helper) get_archid() (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) get_attn_head_simplex() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) get_augment() (in module archai.supergraph.datasets.augmentation) get_batch() (archai.datasets.nlp.nvidia_data_loader_utils.LMOrderedIterator method) (in module archai.supergraph.algos.divnas.analyse_activations) get_cell_template() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_cell_type() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_ch() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_col() (archai.common.delimited_text.DelimitedText method) get_combinations() (archai.supergraph.nas.search_combinations.SearchCombinations method) get_conf() (archai.supergraph.nas.exp_runner.ExperimentRunner method) (in module archai.common.common) get_conf_cell() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_conf_common() (in module archai.common.common) get_conf_dataset() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) (in module archai.common.common) get_conf_model_stems() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_console_handler() (in module archai.common.ordered_dict_logger_utils) get_data() (archai.supergraph.nas.evaluater.Evaluater method) (archai.supergraph.nas.searcher.Searcher method) (in module archai.supergraph.datasets.data) get_datadir() (in module archai.common.common) get_dataloaders() (in module archai.supergraph.datasets.data) get_dataset() (archai.datasets.nlp.hf_dataset_provider.HfHubDatasetProvider method) get_datasets() (archai.supergraph.datasets.dataset_provider.DatasetProvider method) (archai.supergraph.datasets.providers.aircraft_provider.AircraftProvider method) (archai.supergraph.datasets.providers.cifar100_provider.Cifar100Provider method) (archai.supergraph.datasets.providers.cifar10_provider.Cifar10Provider method) (archai.supergraph.datasets.providers.fashion_mnist_provider.FashionMnistProvider method) (archai.supergraph.datasets.providers.flower102_provider.Flower102Provider method) (archai.supergraph.datasets.providers.food101_provider.Food101Provider method) (archai.supergraph.datasets.providers.imagenet_provider.ImagenetProvider method) (archai.supergraph.datasets.providers.mit67_provider.Mit67Provider method) (archai.supergraph.datasets.providers.mnist_provider.MnistProvider method) (archai.supergraph.datasets.providers.person_coco_provider.PersonCocoProvider method) (archai.supergraph.datasets.providers.sport8_provider.Sport8Provider method) (archai.supergraph.datasets.providers.stanfordcars_provider.StanfordCarsProvider method) (archai.supergraph.datasets.providers.svhn_provider.SvhnProvider method) get_device() (archai.supergraph.utils.tester.Tester method) (archai.supergraph.utils.trainer.Trainer method) get_dict() (archai.supergraph.utils.metrics.Accumulator method) get_edge_ranks() (archai.supergraph.nas.finalizers.Finalizers method) get_elapsed() (archai.common.stopwatch.StopWatch method) get_existing_status() (archai.common.store.ArchaiStore method) get_expdir() (in module archai.common.common) get_experiment_name() (in module archai.common.common) get_expname() (archai.supergraph.nas.exp_runner.ExperimentRunner method) get_fixlen_iter() (archai.datasets.nlp.nvidia_data_loader_utils.LMOrderedIterator method) get_flops() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) get_full_path() (in module archai.common.file_utils) get_global_instance() (archai.common.ordered_dict_logger.OrderedDictLogger static method) get_global_logger() (in module archai.common.ordered_dict_logger) get_hf_config_cls() (archai.discrete_search.search_spaces.nlp.tfpp.model.LanguageModel static method) get_initializer() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) get_input_embeddings() (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenModel method) get_inputs() (archai.onnx.config_utils.onnx_config_base.OnnxConfig method) (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast method) get_inst() (archai.common.config.Config static method) get_kernels_forward() (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 method) get_latency() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) get_lock() (archai.common.store.ArchaiStore method) get_logger() (in module archai.common.ordered_dict_logger_utils) get_lossfn() (in module archai.common.ml_utils) get_lr() (archai.supergraph.utils.multi_optim.MultiOptim method) (archai.trainers.cyclic_cosine_scheduler.CyclicCosineDecayLR method) (archai.trainers.gradual_warmup_scheduler.GradualWarmupScheduler method) get_macs() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) get_metrics() (archai.supergraph.utils.tester.Tester method) (archai.supergraph.utils.trainer.Trainer method) get_model_stats() (in module archai.supergraph.nas.nas_utils) get_node_channels() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_node_count() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_non_dominated_sorting() (in module archai.discrete_search.utils.multi_objective) get_optim_flag() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) get_optim_lr() (in module archai.common.ml_utils) get_optimizer() (archai.supergraph.utils.trainer.Trainer method) get_output_embeddings() (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenForCausalLM method) get_outputs() (archai.onnx.config_utils.onnx_config_base.OnnxConfig method) (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast method) get_param_name_list() (archai.discrete_search.search_spaces.config.arch_param_tree.ArchParamTree method) get_params() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) get_pareto_frontier() (archai.discrete_search.api.search_results.SearchResults method) (in module archai.discrete_search.utils.multi_objective) get_peak_memory() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) get_provider_type() (in module archai.supergraph.datasets.dataset_provider) get_rank() (in module archai.common.distributed_utils) get_ranks() (in module archai.common.utils) get_reduction_indices() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_scheduler() (archai.supergraph.utils.trainer.Trainer method) get_search_csv() (in module archai.common.notebook_helper) get_search_state_df() (archai.discrete_search.api.search_results.SearchResults method) get_sequences() (archai.datasets.nlp.nvidia_data_loader_utils.LMMultiFileIterator method) get_shape() (in module archai.common.model_summary) get_special_tokens() (archai.datasets.nlp.tokenizer_utils.token_config.TokenConfig method) get_state() (in module archai.common.common) get_status() (archai.common.store.ArchaiStore method) get_surrogate_iter_dataset() (archai.discrete_search.algos.bananas.MoBananasSearch method) get_tb_writer() (in module archai.common.common) get_temp_file() (archai.common.file_utils.TemporaryFiles method) get_tensor_stat() (in module archai.common.model_summary) get_test_dataset() (archai.api.dataset_provider.DatasetProvider method) (archai.datasets.cv.aircraft_dataset_provider.AircraftDatasetProvider method) (archai.datasets.cv.caltech_dataset_provider.CaltechDatasetProvider method) (archai.datasets.cv.cifar_dataset_provider.CifarDatasetProvider method) (archai.datasets.cv.cityscapes_dataset_provider.CityscapesDatasetProvider method) (archai.datasets.cv.coco_dataset_provider.CocoDatasetProvider method) (archai.datasets.cv.flowers102_dataset_provider.Flowers102DatasetProvider method) (archai.datasets.cv.food101_dataset_provider.Food101DatasetProvider method) (archai.datasets.cv.image_folder_dataset_provider.ImageFolderDatasetProvider method) (archai.datasets.cv.imagenet_dataset_provider.ImageNetDatasetProvider method) (archai.datasets.cv.mnist_dataset_provider.MnistDatasetProvider method) (archai.datasets.cv.stanford_cars_dataset_provider.StanfordCarsDatasetProvider method) (archai.datasets.cv.svhn_dataset_provider.SVHNDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.MultiFileTensorpackLmdbDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.TensorpackLmdbDatasetProvider method) (archai.datasets.cv.usps_dataset_provider.USPSDatasetProvider method) (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfDiskDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfHubDatasetProvider method) (archai.datasets.nlp.nvidia_dataset_provider.NvidiaDatasetProvider method) get_timed_file_handler() (in module archai.common.ordered_dict_logger_utils) get_top1_for_stage() (in module archai.supergraph.algos.petridish.petridish_utils) get_train_dataset() (archai.api.dataset_provider.DatasetProvider method) (archai.datasets.cv.aircraft_dataset_provider.AircraftDatasetProvider method) (archai.datasets.cv.caltech_dataset_provider.CaltechDatasetProvider method) (archai.datasets.cv.cifar_dataset_provider.CifarDatasetProvider method) (archai.datasets.cv.cityscapes_dataset_provider.CityscapesDatasetProvider method) (archai.datasets.cv.coco_dataset_provider.CocoDatasetProvider method) (archai.datasets.cv.flowers102_dataset_provider.Flowers102DatasetProvider method) (archai.datasets.cv.food101_dataset_provider.Food101DatasetProvider method) (archai.datasets.cv.image_folder_dataset_provider.ImageFolderDatasetProvider method) (archai.datasets.cv.imagenet_dataset_provider.ImageNetDatasetProvider method) (archai.datasets.cv.mnist_dataset_provider.MnistDatasetProvider method) (archai.datasets.cv.stanford_cars_dataset_provider.StanfordCarsDatasetProvider method) (archai.datasets.cv.svhn_dataset_provider.SVHNDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.MultiFileTensorpackLmdbDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.TensorpackLmdbDatasetProvider method) (archai.datasets.cv.usps_dataset_provider.USPSDatasetProvider method) (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfDiskDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfHubDatasetProvider method) (archai.datasets.nlp.nvidia_dataset_provider.NvidiaDatasetProvider method) get_trainables() (archai.supergraph.nas.operations.Op method) get_trainables_from() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) get_transforms() (archai.supergraph.datasets.dataset_provider.DatasetProvider method) (archai.supergraph.datasets.providers.aircraft_provider.AircraftProvider method) (archai.supergraph.datasets.providers.cifar100_provider.Cifar100Provider method) (archai.supergraph.datasets.providers.cifar10_provider.Cifar10Provider method) (archai.supergraph.datasets.providers.fashion_mnist_provider.FashionMnistProvider method) (archai.supergraph.datasets.providers.flower102_provider.Flower102Provider method) (archai.supergraph.datasets.providers.food101_provider.Food101Provider method) (archai.supergraph.datasets.providers.imagenet_provider.ImagenetProvider method) (archai.supergraph.datasets.providers.mit67_provider.Mit67Provider method) (archai.supergraph.datasets.providers.mnist_provider.MnistProvider method) (archai.supergraph.datasets.providers.person_coco_provider.PersonCocoProvider method) (archai.supergraph.datasets.providers.sport8_provider.Sport8Provider method) (archai.supergraph.datasets.providers.stanfordcars_provider.StanfordCarsProvider method) (archai.supergraph.datasets.providers.svhn_provider.SvhnProvider method) get_updated_status() (archai.common.store.ArchaiStore method) get_used_params() (archai.discrete_search.search_spaces.config.arch_config.ArchConfig method) get_utc_date() (archai.common.store.ArchaiStore method) get_val() (archai.common.config.Config method) get_val_dataset() (archai.api.dataset_provider.DatasetProvider method) (archai.datasets.cv.aircraft_dataset_provider.AircraftDatasetProvider method) (archai.datasets.cv.caltech_dataset_provider.CaltechDatasetProvider method) (archai.datasets.cv.cifar_dataset_provider.CifarDatasetProvider method) (archai.datasets.cv.cityscapes_dataset_provider.CityscapesDatasetProvider method) (archai.datasets.cv.coco_dataset_provider.CocoDatasetProvider method) (archai.datasets.cv.flowers102_dataset_provider.Flowers102DatasetProvider method) (archai.datasets.cv.food101_dataset_provider.Food101DatasetProvider method) (archai.datasets.cv.image_folder_dataset_provider.ImageFolderDatasetProvider method) (archai.datasets.cv.imagenet_dataset_provider.ImageNetDatasetProvider method) (archai.datasets.cv.mnist_dataset_provider.MnistDatasetProvider method) (archai.datasets.cv.stanford_cars_dataset_provider.StanfordCarsDatasetProvider method) (archai.datasets.cv.svhn_dataset_provider.SVHNDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.MultiFileTensorpackLmdbDatasetProvider method) (archai.datasets.cv.tensorpack_lmdb_dataset_provider.TensorpackLmdbDatasetProvider method) (archai.datasets.cv.usps_dataset_provider.USPSDatasetProvider method) (archai.datasets.nlp.fast_hf_dataset_provider.FastHfDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfDiskDatasetProvider method) (archai.datasets.nlp.hf_dataset_provider.HfHubDatasetProvider method) (archai.datasets.nlp.nvidia_dataset_provider.NvidiaDatasetProvider method) get_varlen_iter() (archai.datasets.nlp.nvidia_data_loader_utils.LMOrderedIterator method) get_world_size() (in module archai.common.distributed_utils) global_batch_size (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) GoogLeNet (class in archai.supergraph.models.googlenet) googlenet() (in module archai.supergraph.models.googlenet) gpt2_onnx_forward() (in module archai.onnx.onnx_forward) GPT2FlexAttention (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex) GPT2FlexBlock (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex) GPT2FlexConfig (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_gpt2_flex) GPT2FlexLMHeadModel (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex) GPT2FlexMLP (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex) GPT2FlexModel (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex) GPT2FlexOnnxConfig (class in archai.onnx.config_utils.gpt2_onnx_config) GPT2OnnxConfig (class in archai.onnx.config_utils.gpt2_onnx_config) Gpt2Tokenizer (class in archai.datasets.nlp.tokenizer_utils.gpt2_tokenizer) gradient_accumulation_steps (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) GradualWarmupScheduler (class in archai.trainers.gradual_warmup_scheduler) greedy_op_selection() (in module archai.supergraph.algos.divnas.analyse_activations) group_texts() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) groups (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) GsArchTrainer (class in archai.supergraph.algos.gumbelsoftmax.gs_arch_trainer) GsExperimentRunner (class in archai.supergraph.algos.gumbelsoftmax.gs_exp_runner) GsFinalizers (class in archai.supergraph.algos.gumbelsoftmax.gs_finalizers) GsModelDescBuilder (class in archai.supergraph.algos.gumbelsoftmax.gs_model_desc_builder) GsOp (class in archai.supergraph.algos.gumbelsoftmax.gs_op) H has_aux_tower() (archai.supergraph.nas.model_desc.ModelDesc method) has_kind() (archai.supergraph.nas.arch_params.ArchParams method) has_method() (in module archai.common.utils) hash_module() (in module archai.supergraph.algos.nasbench101.graph_util) hash_spec() (archai.supergraph.algos.nasbench101.model_spec.ModelSpec method) heatmap() (in module archai.supergraph.utils.heatmap) HfDiskDatasetProvider (class in archai.datasets.nlp.hf_dataset_provider) HfDistillerTrainer (class in archai.trainers.nlp.hf_trainer) HfHubDatasetProvider (class in archai.datasets.nlp.hf_dataset_provider) HfTrainer (class in archai.trainers.nlp.hf_trainer) hidden_size (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast property) hidden_states (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerBaseOutput attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) hull_points2tsv() (in module archai.supergraph.algos.petridish.petridish_utils) I id_to_token() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) Identity (class in archai.supergraph.nas.operations) ids_to_tokens() (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) ImageFolderDatasetProvider (class in archai.datasets.cv.image_folder_dataset_provider) ImageNetDatasetProvider (class in archai.datasets.cv.imagenet_dataset_provider) ImageNetFolder (class in archai.supergraph.datasets.providers.imagenet_folder) ImagenetProvider (class in archai.supergraph.datasets.providers.imagenet_provider) in_channels (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) in_features (archai.quantization.modules.FakeDynamicQuantLinear attribute) (archai.quantization.modules.FakeDynamicQuantLinearForOnnx attribute) Inception3 (class in archai.supergraph.models.inception) inception_v3() (in module archai.supergraph.models.inception) inference_stats() (in module archai.common.ml_perf_utils) info() (archai.common.ordered_dict_logger.OrderedDictLogger method) init_distributed() (in module archai.common.distributed_utils) init_from() (in module archai.common.common) init_kernels() (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 method) init_weights() (archai.supergraph.models.alexnet.AlexNet method) int_parameter() (in module archai.supergraph.datasets.augmentation) Invert() (in module archai.supergraph.datasets.augmentation) is_better_metrics() (archai.supergraph.nas.search_combinations.SearchCombinations method) is_debugging() (in module archai.common.utils) is_dist() (archai.common.apex_utils.ApexUtils method) (archai.supergraph.utils.metrics.Metrics method) is_empty() (archai.supergraph.utils.checkpoint.CheckPoint method) is_full_dag() (in module archai.supergraph.algos.nasbench101.graph_util) is_isomorphic() (in module archai.supergraph.algos.nasbench101.graph_util) is_locked() (archai.common.store.ArchaiStore method) is_locked_by_other() (archai.common.store.ArchaiStore method) is_locked_by_self() (archai.common.store.ArchaiStore method) is_main_process() (in module archai.common.utils) is_master() (archai.common.apex_utils.ApexUtils method) is_mixed() (archai.common.apex_utils.ApexUtils method) is_model_valid() (archai.discrete_search.api.search_objectives.SearchObjectives method) is_ort_graph_optimizable (archai.onnx.config_utils.codegen_onnx_config.CodeGenOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2FlexOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2OnnxConfig property) (archai.onnx.config_utils.onnx_config_base.OnnxConfig property) is_proper_path() (in module archai.common.yaml_utils) is_pt() (in module archai.common.common) is_ray() (archai.common.apex_utils.ApexUtils method) is_scaler() (in module archai.common.model_summary) is_trained() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) is_trained_stage() (archai.supergraph.algos.petridish.petridish_utils.ConvexHullPoint method) is_upper_triangular() (in module archai.supergraph.algos.nasbench101.model_spec) is_valid_model() (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) is_windows() (in module archai.common.utils) items() (archai.supergraph.utils.metrics.Accumulator method) iterator_roll (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) J JITLamb (class in archai.trainers.lamb_optimizer) JobStage (class in archai.supergraph.algos.petridish.petridish_utils) join_chunks() (in module archai.common.ml_utils) K kernel_size (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) keys() (archai.common.stopwatch.StopWatch method) L Lamb (class in archai.trainers.lamb_optimizer) LanguageModel (class in archai.discrete_search.search_spaces.nlp.tfpp.model) last_hidden_state (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerBaseOutput attribute) learning_rate (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) Lighting (class in archai.datasets.cv.transforms.lighting) LimitDataset (class in archai.supergraph.datasets.limit_dataset) LinearActivation() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) LinearOp (class in archai.supergraph.nas.operations) list_blobs() (archai.common.store.ArchaiStore method) LMMultiFileIterator (class in archai.datasets.nlp.nvidia_data_loader_utils) LMOrderedIterator (class in archai.datasets.nlp.nvidia_data_loader_utils) load() (archai.common.ordered_dict_logger.OrderedDictLogger method) (archai.datasets.nlp.nvidia_dataset_provider_utils.Corpus method) (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) (archai.supergraph.nas.model_desc.ModelDesc static method) load_arch() (archai.discrete_search.api.search_space.DiscreteSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) load_cache() (archai.discrete_search.api.search_objectives.SearchObjectives method) load_checkpoint() (archai.trainers.nlp.nvidia_trainer.NvidiaTrainer method) load_existing() (archai.supergraph.utils.checkpoint.CheckPoint method) load_from_graph() (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) load_from_onnx() (in module archai.onnx.onnx_loader) load_model() (in module archai.common.ml_utils) load_model_weights() (archai.discrete_search.api.search_space.DiscreteSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) load_state_dict() (archai.common.apex_utils.ApexUtils method) (archai.supergraph.algos.darts.bilevel_optimizer.BilevelOptimizer method) (archai.supergraph.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method) (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.EdgeDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) (archai.supergraph.nas.model_desc.NodeDesc method) (archai.supergraph.nas.model_desc.OpDesc method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.multi_optim.MultiOptim method) (archai.supergraph.utils.tester.Tester method) (in module archai.common.utils) local_rank (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) LocalAttention (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) LocalMHA (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) LocalSearch (class in archai.discrete_search.algos.local_search) lock() (archai.common.store.ArchaiStore method) (in module archai.common.store) lock_entity() (archai.common.store.ArchaiStore method) log() (archai.common.ordered_dict_logger.OrderedDictLogger method) logging_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) logits (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput property) look_around() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) lookup_cache() (archai.discrete_search.api.search_objectives.SearchObjectives method) loss (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) lr_qat_scheduler_type (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_decay_rate (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_max_steps (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_min_lr (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_patience (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_type (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) lr_scheduler_warmup_steps (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) LSHAttention (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.lsh_attn) M main() (in module archai.supergraph.algos.divnas.analyse_activations) main_process_pid() (in module archai.common.utils) make_asso_map() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) make_broadcast_map() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) ManualEvaluater (class in archai.supergraph.algos.manual.manual_evaluater) ManualExperimentRunner (class in archai.supergraph.algos.manual.manual_exp_runner) ManualSearcher (class in archai.supergraph.algos.manual.manual_searcher) map_to_list() (in module archai.common.utils) MASK (archai.datasets.nlp.tokenizer_utils.token_config.SpecialTokenEnum attribute) MaskIndexEnd (archai.onnx.optimization_utils.fusion_options.AttentionMaskFormat attribute) MaskIndexEndAndStart (archai.onnx.optimization_utils.fusion_options.AttentionMaskFormat attribute) max_grad_norm (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) max_neg_value() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) max_norm (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) max_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) MaxPool3x3 (class in archai.supergraph.algos.nasbench101.base_ops) mean (archai.discrete_search.api.predictor.MeanVar attribute) MeanVar (class in archai.discrete_search.api.predictor) MeasureBlockTime (class in archai.common.timing) memory_allocated() (in module archai.common.ml_utils) mems (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerBaseOutput attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) MemTransformerBaseOutput (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer) MemTransformerConfig (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_mem_transformer) MemTransformerLMHeadModel (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer) MemTransformerModel (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer) MemTransformerOutput (class in archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer) merge_heads() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) merge_status_entity() (archai.common.store.ArchaiStore method) MergeOp (class in archai.supergraph.nas.operations) meta_file (archai.supergraph.datasets.providers.imagenet_folder.ImageNetFolder property) MetaDataset (class in archai.supergraph.datasets.meta_dataset) Metrics (class in archai.supergraph.utils.metrics) MHA (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.mha) Mit67Provider (class in archai.supergraph.datasets.providers.mit67_provider) mixed_qat (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) MixedAttentionBlock (class in archai.discrete_search.search_spaces.nlp.tfpp.mixed_attention) (class in archai.discrete_search.search_spaces.nlp.tfpp.mixed_op) MixedOp (class in archai.supergraph.algos.darts.mixed_op) MixedQAT (class in archai.quantization.mixed_qat) Mlp (class in archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block) MnistDatasetProvider (class in archai.datasets.cv.mnist_dataset_provider) MnistProvider (class in archai.supergraph.datasets.providers.mnist_provider) MoBananasSearch (class in archai.discrete_search.algos.bananas) mobilenet_v2() (in module archai.supergraph.models.mobilenetv2) MobileNetV2 (class in archai.supergraph.models.mobilenetv2) Model (class in archai.supergraph.nas.model) model_desc_builder() (archai.supergraph.algos.darts.darts_exp_runner.DartsExperimentRunner method) (archai.supergraph.algos.didarts.didarts_exp_runner.DiDartsExperimentRunner method) (archai.supergraph.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method) (archai.supergraph.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method) (archai.supergraph.algos.manual.manual_exp_runner.ManualExperimentRunner method) (archai.supergraph.algos.nasbench101.nasbench101_exp_runner.PetridishExperimentRunner method) (archai.supergraph.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method) (archai.supergraph.algos.random.random_exp_runner.RandomExperimentRunner method) (archai.supergraph.algos.xnas.xnas_exp_runner.XnasExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) model_descs_on_front() (in module archai.supergraph.algos.petridish.petridish_utils) model_from_desc() (archai.supergraph.nas.evaluater.Evaluater method) model_from_natsbench_tss() (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) model_memory() (in module archai.common.ml_perf_utils) model_type (archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_gpt2_flex.GPT2FlexConfig attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_mem_transformer.MemTransformerConfig attribute) ModelDesc (class in archai.supergraph.nas.model_desc) ModelDescBuilder (class in archai.supergraph.nas.model_desc_builder) ModelEvaluator (class in archai.discrete_search.api.model_evaluator) ModelMetrics (class in archai.supergraph.nas.searcher) ModelSpec (class in archai.supergraph.algos.nasbench101.model_spec) Modrelu (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) modrelu (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) module archai.api.dataset_provider archai.api.trainer_base archai.common.apex_utils archai.common.atomic_file_handler archai.common.azureml_helper archai.common.common archai.common.config archai.common.delimited_text archai.common.deprecation_utils archai.common.distributed_utils archai.common.file_utils archai.common.ml_perf_utils archai.common.ml_utils archai.common.model_summary archai.common.notebook_helper archai.common.ordered_dict_logger archai.common.ordered_dict_logger_utils archai.common.stopwatch archai.common.store archai.common.timing archai.common.utils archai.common.yaml_utils archai.datasets.cv.aircraft_dataset_provider archai.datasets.cv.caltech_dataset_provider archai.datasets.cv.cifar_dataset_provider archai.datasets.cv.cityscapes_dataset_provider archai.datasets.cv.coco_dataset_provider archai.datasets.cv.flowers102_dataset_provider archai.datasets.cv.food101_dataset_provider archai.datasets.cv.image_folder_dataset_provider archai.datasets.cv.imagenet_dataset_provider archai.datasets.cv.mnist_dataset_provider archai.datasets.cv.stanford_cars_dataset_provider archai.datasets.cv.svhn_dataset_provider archai.datasets.cv.tensorpack_lmdb_dataset_provider archai.datasets.cv.tensorpack_lmdb_dataset_provider_utils archai.datasets.cv.transforms.brightness archai.datasets.cv.transforms.custom_cutout archai.datasets.cv.transforms.lighting archai.datasets.cv.usps_dataset_provider archai.datasets.nlp.fast_hf_dataset_provider archai.datasets.nlp.fast_hf_dataset_provider_utils archai.datasets.nlp.hf_dataset_provider archai.datasets.nlp.hf_dataset_provider_utils archai.datasets.nlp.nvidia_data_loader_utils archai.datasets.nlp.nvidia_dataset_provider archai.datasets.nlp.nvidia_dataset_provider_utils archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer archai.datasets.nlp.tokenizer_utils.gpt2_tokenizer archai.datasets.nlp.tokenizer_utils.token_config archai.datasets.nlp.tokenizer_utils.tokenizer_base archai.datasets.nlp.tokenizer_utils.word_tokenizer archai.discrete_search.algos.bananas archai.discrete_search.algos.evolution_pareto archai.discrete_search.algos.local_search archai.discrete_search.algos.random_search archai.discrete_search.algos.regularized_evolution archai.discrete_search.algos.successive_halving archai.discrete_search.api.archai_model archai.discrete_search.api.model_evaluator archai.discrete_search.api.predictor archai.discrete_search.api.search_objectives archai.discrete_search.api.search_results archai.discrete_search.api.search_space archai.discrete_search.api.searcher archai.discrete_search.evaluators.benchmark.natsbench_tss archai.discrete_search.evaluators.functional archai.discrete_search.evaluators.nlp.parameters archai.discrete_search.evaluators.nlp.transformer_flex_latency archai.discrete_search.evaluators.nlp.transformer_flex_memory archai.discrete_search.evaluators.onnx_model archai.discrete_search.evaluators.progressive_training archai.discrete_search.evaluators.pt_profiler archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_eval archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_hooks archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model archai.discrete_search.evaluators.ray archai.discrete_search.evaluators.remote_azure_benchmark archai.discrete_search.predictors.dnn_ensemble archai.discrete_search.search_spaces.benchmark.natsbench_tss archai.discrete_search.search_spaces.config.arch_config archai.discrete_search.search_spaces.config.arch_param_tree archai.discrete_search.search_spaces.config.discrete_choice archai.discrete_search.search_spaces.config.helpers archai.discrete_search.search_spaces.config.search_space archai.discrete_search.search_spaces.config.utils archai.discrete_search.search_spaces.cv.segmentation_dag.model archai.discrete_search.search_spaces.cv.segmentation_dag.ops archai.discrete_search.search_spaces.cv.segmentation_dag.search_space archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model archai.discrete_search.search_spaces.nlp.tfpp.mixed_attention archai.discrete_search.search_spaces.nlp.tfpp.mixed_op archai.discrete_search.search_spaces.nlp.tfpp.model archai.discrete_search.search_spaces.nlp.tfpp.ops.causal_self_attn archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention archai.discrete_search.search_spaces.nlp.tfpp.ops.lsh_attn archai.discrete_search.search_spaces.nlp.tfpp.ops.mha archai.discrete_search.search_spaces.nlp.tfpp.ops.sep_conv1d archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3 archai.discrete_search.search_spaces.nlp.tfpp.search_space archai.discrete_search.search_spaces.nlp.tfpp.utils archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_gpt2_flex archai.discrete_search.search_spaces.nlp.transformer_flex.models.configuration_mem_transformer archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer archai.discrete_search.search_spaces.nlp.transformer_flex.search_space archai.discrete_search.utils.multi_objective archai.onnx.config_utils.codegen_onnx_config archai.onnx.config_utils.gpt2_onnx_config archai.onnx.config_utils.onnx_config_base archai.onnx.export archai.onnx.export_utils archai.onnx.onnx_forward archai.onnx.onnx_loader archai.onnx.optimization archai.onnx.optimization_utils.fusion_options archai.onnx.optimization_utils.transfo_xl_onnx_model archai.quantization.mixed_qat archai.quantization.modules archai.quantization.nlp.modules archai.quantization.observers archai.quantization.ptq archai.quantization.qat archai.quantization.quantization_utils archai.quantization.quantizers archai.supergraph.algos.darts.bilevel_arch_trainer archai.supergraph.algos.darts.bilevel_optimizer archai.supergraph.algos.darts.bilevel_optimizer_slow archai.supergraph.algos.darts.darts_exp_runner archai.supergraph.algos.darts.darts_model_desc_builder archai.supergraph.algos.darts.mixed_op archai.supergraph.algos.didarts.didarts_arch_trainer archai.supergraph.algos.didarts.didarts_exp_runner archai.supergraph.algos.divnas.analyse_activations archai.supergraph.algos.divnas.divnas_cell archai.supergraph.algos.divnas.divnas_exp_runner archai.supergraph.algos.divnas.divnas_finalizers archai.supergraph.algos.divnas.divnas_model_desc_builder archai.supergraph.algos.divnas.divnas_rank_finalizer archai.supergraph.algos.divnas.divop archai.supergraph.algos.divnas.seqopt archai.supergraph.algos.divnas.wmr archai.supergraph.algos.gumbelsoftmax.gs_arch_trainer archai.supergraph.algos.gumbelsoftmax.gs_exp_runner archai.supergraph.algos.gumbelsoftmax.gs_finalizers archai.supergraph.algos.gumbelsoftmax.gs_model_desc_builder archai.supergraph.algos.gumbelsoftmax.gs_op archai.supergraph.algos.manual.manual_evaluater archai.supergraph.algos.manual.manual_exp_runner archai.supergraph.algos.manual.manual_searcher archai.supergraph.algos.nasbench101.base_ops archai.supergraph.algos.nasbench101.config archai.supergraph.algos.nasbench101.graph_util archai.supergraph.algos.nasbench101.model archai.supergraph.algos.nasbench101.model_builder archai.supergraph.algos.nasbench101.model_matrix archai.supergraph.algos.nasbench101.model_metrics_pb2 archai.supergraph.algos.nasbench101.model_spec archai.supergraph.algos.nasbench101.nasbench101_exp_runner archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder archai.supergraph.algos.nasbench101.nasbench101_op archai.supergraph.algos.petridish.evaluater_petridish archai.supergraph.algos.petridish.petridish_exp_runner archai.supergraph.algos.petridish.petridish_model_desc_builder archai.supergraph.algos.petridish.petridish_op archai.supergraph.algos.petridish.petridish_utils archai.supergraph.algos.petridish.searcher_petridish archai.supergraph.algos.random.random_exp_runner archai.supergraph.algos.random.random_model_desc_builder archai.supergraph.algos.xnas.xnas_arch_trainer archai.supergraph.algos.xnas.xnas_exp_runner archai.supergraph.algos.xnas.xnas_model_desc_builder archai.supergraph.algos.xnas.xnas_op archai.supergraph.datasets.aug_policies archai.supergraph.datasets.augmentation archai.supergraph.datasets.data archai.supergraph.datasets.dataset_provider archai.supergraph.datasets.distributed_stratified_sampler archai.supergraph.datasets.limit_dataset archai.supergraph.datasets.meta_dataset archai.supergraph.datasets.providers.aircraft_provider archai.supergraph.datasets.providers.cifar100_provider archai.supergraph.datasets.providers.cifar10_provider archai.supergraph.datasets.providers.fashion_mnist_provider archai.supergraph.datasets.providers.flower102_provider archai.supergraph.datasets.providers.food101_provider archai.supergraph.datasets.providers.imagenet_folder archai.supergraph.datasets.providers.imagenet_provider archai.supergraph.datasets.providers.mit67_provider archai.supergraph.datasets.providers.mnist_provider archai.supergraph.datasets.providers.person_coco_provider archai.supergraph.datasets.providers.sport8_provider archai.supergraph.datasets.providers.stanfordcars_provider archai.supergraph.datasets.providers.svhn_provider archai.supergraph.models.alexnet archai.supergraph.models.densenet archai.supergraph.models.googlenet archai.supergraph.models.inception archai.supergraph.models.mobilenetv2 archai.supergraph.models.pyramidnet archai.supergraph.models.resnet archai.supergraph.models.resnet_orig archai.supergraph.models.resnet_paper archai.supergraph.models.shakedrop archai.supergraph.models.shakeshake.shake_resnet archai.supergraph.models.shakeshake.shake_resnext archai.supergraph.models.shakeshake.shakeshake archai.supergraph.models.vgg archai.supergraph.models.wideresnet archai.supergraph.nas.arch_module archai.supergraph.nas.arch_params archai.supergraph.nas.arch_trainer archai.supergraph.nas.cell archai.supergraph.nas.dag_edge archai.supergraph.nas.evaluater archai.supergraph.nas.exp_runner archai.supergraph.nas.finalizers archai.supergraph.nas.model archai.supergraph.nas.model_desc archai.supergraph.nas.model_desc_builder archai.supergraph.nas.nas_utils archai.supergraph.nas.operations archai.supergraph.nas.random_finalizers archai.supergraph.nas.search_combinations archai.supergraph.nas.searcher archai.supergraph.nas.vis_model_desc archai.supergraph.utils.augmented_searcher archai.supergraph.utils.augmented_trainer archai.supergraph.utils.checkpoint archai.supergraph.utils.heatmap archai.supergraph.utils.metrics archai.supergraph.utils.multi_optim archai.supergraph.utils.tester archai.supergraph.utils.trainer archai.trainers.coin_betting_optimizer archai.trainers.cv.pl_trainer archai.trainers.cyclic_cosine_scheduler archai.trainers.gradual_warmup_scheduler archai.trainers.lamb_optimizer archai.trainers.losses archai.trainers.nlp.ds_trainer archai.trainers.nlp.ds_training_args archai.trainers.nlp.hf_callbacks archai.trainers.nlp.hf_trainer archai.trainers.nlp.hf_training_args archai.trainers.nlp.nvidia_trainer archai.trainers.nlp.nvidia_training_args momentum (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) MultiFileTensorpackLmdbDatasetProvider (class in archai.datasets.cv.tensorpack_lmdb_dataset_provider) MultiOp (class in archai.supergraph.nas.operations) MultiOptim (class in archai.supergraph.utils.multi_optim) mutate() (archai.discrete_search.api.search_space.EvolutionarySearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) mutate_parents() (archai.discrete_search.algos.bananas.MoBananasSearch method) (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) (archai.discrete_search.algos.local_search.LocalSearch method) (archai.discrete_search.algos.regularized_evolution.RegularizedEvolutionSearch method) N NasBench101CellBuilder (class in archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder) NasBench101Op (class in archai.supergraph.algos.nasbench101.nasbench101_op) NatsbenchMetric (class in archai.discrete_search.evaluators.benchmark.natsbench_tss) NatsbenchTssSearchSpace (class in archai.discrete_search.search_spaces.benchmark.natsbench_tss) Network (class in archai.supergraph.algos.nasbench101.model) new() (archai.supergraph.utils.checkpoint.CheckPoint method) next_stage() (archai.supergraph.algos.petridish.petridish_utils.ConvexHullPoint method) no_cuda (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) no_duplicates() (in module archai.supergraph.datasets.augmentation) NodeDesc (class in archai.supergraph.nas.model_desc) nodes() (archai.supergraph.nas.model_desc.CellDesc method) NoMask (archai.onnx.optimization_utils.fusion_options.AttentionMaskFormat attribute) nonarch_from_module() (archai.supergraph.nas.arch_params.ArchParams static method) nonarch_params() (archai.supergraph.nas.arch_module.ArchModule method) NonEmbeddingParamsProxy (class in archai.discrete_search.evaluators.nlp.parameters) norm_type (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) NormalConvBlock (class in archai.discrete_search.search_spaces.cv.segmentation_dag.ops) Normalization (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) num_archs (archai.discrete_search.search_spaces.config.arch_param_tree.ArchParamTree property) num_attention_heads (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast property) num_edges() (in module archai.supergraph.algos.nasbench101.graph_util) num_embeddings (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) num_layers (archai.onnx.config_utils.codegen_onnx_config.CodeGenOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2FlexOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2OnnxConfig property) (archai.onnx.config_utils.onnx_config_base.OnnxConfigWithPast property) num_primitive_ops (archai.supergraph.algos.divnas.divop.DivOp property) NvidiaDatasetProvider (class in archai.datasets.nlp.nvidia_dataset_provider) NvidiaTrainer (class in archai.trainers.nlp.nvidia_trainer) NvidiaTrainingArguments (class in archai.trainers.nlp.nvidia_training_args) O objective_names (archai.discrete_search.api.search_objectives.SearchObjectives property) objectives (archai.discrete_search.api.search_objectives.SearchObjectives property) on_app_exit() (in module archai.common.common) on_calc_task_accuracy_end() (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) on_evaluate() (archai.trainers.nlp.hf_callbacks.BPCTrainerCallback method) (archai.trainers.nlp.hf_callbacks.PerplexityTrainerCallback method) on_log() (archai.trainers.nlp.hf_callbacks.BPCTrainerCallback method) (archai.trainers.nlp.hf_callbacks.PerplexityTrainerCallback method) on_search_iteration_start() (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) on_start_iteration() (archai.discrete_search.api.searcher.Searcher method) OnnxConfig (class in archai.onnx.config_utils.onnx_config_base) OnnxConfigWithPast (class in archai.onnx.config_utils.onnx_config_base) OnnxDynamicObserver (class in archai.quantization.observers) Op (class in archai.supergraph.nas.operations) op() (archai.supergraph.nas.dag_edge.DagEdge method) OpDesc (class in archai.supergraph.nas.model_desc) OPS (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace attribute) ops() (archai.supergraph.algos.darts.mixed_op.MixedOp method) (archai.supergraph.algos.divnas.divop.DivOp method) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp method) (archai.supergraph.algos.petridish.petridish_op.PetridishOp method) (archai.supergraph.nas.cell.Cell method) (archai.supergraph.nas.model.Model method) (archai.supergraph.nas.operations.Op method) optim (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) optimize() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) optimize_onnx() (in module archai.onnx.optimization) OptimSched (class in archai.supergraph.utils.multi_optim) order_dict_keys() (in module archai.discrete_search.search_spaces.config.utils) OrderedDictLogger (class in archai.common.ordered_dict_logger) ort_graph_optimizer_args (archai.onnx.config_utils.codegen_onnx_config.CodeGenOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2FlexOnnxConfig property) (archai.onnx.config_utils.gpt2_onnx_config.GPT2OnnxConfig property) (archai.onnx.config_utils.onnx_config_base.OnnxConfig property) out_channels (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) out_features (archai.quantization.modules.FakeDynamicQuantLinear attribute) (archai.quantization.modules.FakeDynamicQuantLinearForOnnx attribute) outchannel_ratio (archai.supergraph.models.pyramidnet.BasicBlock attribute) (archai.supergraph.models.pyramidnet.Bottleneck attribute) output_dir (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) output_padding (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) P PAD (archai.datasets.nlp.tokenizer_utils.token_config.SpecialTokenEnum attribute) pad_to_multiple() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) padding (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) padding_idx (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) padding_mode (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) param_by_kind() (archai.supergraph.nas.arch_params.ArchParams method) param_size() (in module archai.common.ml_utils) paramdict_by_kind() (archai.supergraph.nas.arch_params.ArchParams method) paramlist_by_kind() (archai.supergraph.nas.arch_params.ArchParams method) parse_connection_string() (archai.common.store.ArchaiStore static method) past_key_values (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerBaseOutput attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) path2uri() (in module archai.common.utils) pause() (archai.common.stopwatch.StopWatch method) per_device_global_batch_size (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) permute_graph() (in module archai.supergraph.algos.nasbench101.graph_util) PerplexityTrainerCallback (class in archai.trainers.nlp.hf_callbacks) PersonCocoProvider (class in archai.supergraph.datasets.providers.person_coco_provider) PetridishExperimentRunner (class in archai.supergraph.algos.nasbench101.nasbench101_exp_runner) (class in archai.supergraph.algos.petridish.petridish_exp_runner) PetridishModelBuilder (class in archai.supergraph.algos.petridish.petridish_model_desc_builder) PetridishOp (class in archai.supergraph.algos.petridish.petridish_op) pick() (archai.discrete_search.search_spaces.config.arch_config.ArchConfig method) (archai.discrete_search.search_spaces.config.arch_config.ArchConfigList method) pipe_parallel_activation_checkpoint_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) pipe_parallel_loss_fn (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) pipe_parallel_partition_method (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) pipe_parallel_size (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) plot_2d_pareto_evolution() (archai.discrete_search.api.search_results.SearchResults method) plot_all_covs() (in module archai.supergraph.algos.divnas.analyse_activations) plot_frontier() (in module archai.supergraph.algos.petridish.petridish_utils) plot_pool() (in module archai.supergraph.algos.petridish.petridish_utils) plot_seed_model_stats() (in module archai.supergraph.algos.petridish.petridish_utils) PlTrainer (class in archai.trainers.cv.pl_trainer) policy_decoder() (in module archai.supergraph.datasets.augmentation) PoolAdaptiveAvg2D (class in archai.supergraph.nas.operations) PoolBN (class in archai.supergraph.nas.operations) PoolMeanTensor (class in archai.supergraph.nas.operations) popd() (archai.common.ordered_dict_logger.OrderedDictLogger method) post_epoch() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.nas.arch_trainer.ArchTrainer method) (archai.supergraph.utils.metrics.EpochMetrics method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.trainer.Trainer method) post_fit() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) post_run() (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.metrics.RunMetrics method) post_step() (archai.supergraph.utils.metrics.EpochMetrics method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.trainer.Trainer method) Posterize() (in module archai.supergraph.datasets.augmentation) Posterize2() (in module archai.supergraph.datasets.augmentation) pre_build() (archai.supergraph.algos.darts.darts_model_desc_builder.DartsModelDescBuilder method) (archai.supergraph.algos.divnas.divnas_model_desc_builder.DivnasModelDescBuilder method) (archai.supergraph.algos.gumbelsoftmax.gs_model_desc_builder.GsModelDescBuilder method) (archai.supergraph.algos.nasbench101.nasbench101_model_desc_builder.NasBench101CellBuilder method) (archai.supergraph.algos.petridish.petridish_model_desc_builder.PetridishModelBuilder method) (archai.supergraph.algos.xnas.xnas_model_desc_builder.XnasModelDescBuilder method) (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) pre_epoch() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.metrics.EpochMetrics method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.trainer.Trainer method) pre_fit() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) pre_run() (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.metrics.RunMetrics method) pre_step() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.gumbelsoftmax.gs_arch_trainer.GsArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.metrics.EpochMetrics method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.trainer.Trainer method) predict() (archai.api.trainer_base.TrainerBase method) (archai.discrete_search.api.predictor.Predictor method) (archai.discrete_search.predictors.dnn_ensemble.PredictiveDNNEnsemble method) (archai.trainers.cv.pl_trainer.PlTrainer method) (archai.trainers.nlp.ds_trainer.DsTrainer method) (archai.trainers.nlp.nvidia_trainer.NvidiaTrainer method) predict_expensive_objectives() (archai.discrete_search.algos.bananas.MoBananasSearch method) prediction_scores (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerOutput attribute) PredictiveDNNEnsemble (class in archai.discrete_search.predictors.dnn_ensemble) Predictor (class in archai.discrete_search.api.predictor) prepare_inputs_for_generation() (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenForCausalLM method) prepare_model_for_onnx() (in module archai.onnx.export_utils) prepare_with_qat() (in module archai.quantization.qat) PRIMITIVES (archai.supergraph.algos.darts.mixed_op.MixedOp attribute) (archai.supergraph.algos.divnas.divop.DivOp attribute) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp attribute) (archai.supergraph.algos.petridish.petridish_op.PetridishOp attribute) (archai.supergraph.algos.random.random_model_desc_builder.RandOps attribute) (archai.supergraph.algos.xnas.xnas_op.XnasOp attribute) print_entities() (archai.common.store.ArchaiStore method) print_memory_objects() (in module archai.common.ml_utils) process_name() (in module archai.common.utils) process_with_memory_map_files() (in module archai.datasets.nlp.fast_hf_dataset_provider_utils) process_with_shared_memory() (in module archai.datasets.nlp.fast_hf_dataset_provider_utils) profile() (in module archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_eval) ProfilerModel (class in archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model) ProgressiveTraining (class in archai.discrete_search.evaluators.progressive_training) ProjectChannelsOp (class in archai.supergraph.nas.operations) Projection() (in module archai.supergraph.algos.nasbench101.model) prune() (in module archai.supergraph.algos.nasbench101.model_matrix) pt_dirs() (in module archai.common.common) pushd() (archai.common.ordered_dict_logger.OrderedDictLogger method) pyramidal_make_layer() (archai.supergraph.models.pyramidnet.PyramidNet method) PyramidNet (class in archai.supergraph.models.pyramidnet) Q qat (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) qat_to_float_modules() (in module archai.quantization.qat) quantize() (archai.quantization.ptq.GemmQuant method) R random_neighbor() (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) random_sample() (archai.discrete_search.api.search_space.DiscreteSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.discrete_choice.DiscreteChoice method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) RandomBrightness (class in archai.datasets.cv.transforms.brightness) RandomExperimentRunner (class in archai.supergraph.algos.random.random_exp_runner) RandomFinalizers (class in archai.supergraph.nas.random_finalizers) RandomModelDescBuilder (class in archai.supergraph.algos.random.random_model_desc_builder) RandomSearch (class in archai.discrete_search.algos.random_search) RandOps (class in archai.supergraph.algos.random.random_model_desc_builder) RankFilter (class in archai.common.ordered_dict_logger_utils) RayParallelEvaluator (class in archai.discrete_search.evaluators.ray) RayProgressiveTraining (class in archai.discrete_search.evaluators.progressive_training) rbf() (in module archai.supergraph.algos.divnas.analyse_activations) read_string() (in module archai.common.utils) record_checkpoint() (archai.supergraph.nas.search_combinations.SearchCombinations method) reduce() (archai.common.apex_utils.ApexUtils method) reduce_max() (archai.supergraph.utils.metrics.Metrics method) reduce_mean() (archai.supergraph.utils.metrics.Metrics method) reduce_min() (archai.supergraph.utils.metrics.Metrics method) reduce_sum() (archai.supergraph.utils.metrics.Metrics method) Reduction (archai.supergraph.nas.model_desc.CellType attribute) reduction (archai.trainers.losses.SmoothCrossEntropyLoss attribute) register_dataset_provider() (in module archai.supergraph.datasets.dataset_provider) register_op() (archai.supergraph.nas.operations.Op static method) Regular (archai.supergraph.nas.model_desc.CellType attribute) RegularizedEvolutionSearch (class in archai.discrete_search.algos.regularized_evolution) ReLUConvBN (class in archai.supergraph.nas.operations) RemoteAzureBenchmarkEvaluator (class in archai.discrete_search.evaluators.remote_azure_benchmark) remove_deplicates() (in module archai.supergraph.datasets.augmentation) rename_dag_node_list() (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) repeat_config() (in module archai.discrete_search.search_spaces.config.helpers) replace_ptree_choices() (in module archai.discrete_search.search_spaces.config.utils) replace_ptree_pair_choices() (in module archai.discrete_search.search_spaces.config.utils) requires_length (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 attribute) reset() (archai.common.store.ArchaiStore method) (archai.common.utils.AverageMeter method) (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) (in module archai.common.store) reset_all() (archai.common.store.ArchaiStore method) reset_cells() (archai.supergraph.nas.model_desc.ModelDesc method) reset_nodes() (archai.supergraph.nas.model_desc.CellDesc method) reset_parameters() (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.Modrelu method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.modrelu method) ResNet (class in archai.supergraph.models.resnet) (class in archai.supergraph.models.resnet_paper) resnet101() (in module archai.supergraph.models.resnet) resnet110() (in module archai.supergraph.models.resnet_paper) resnet1202() (in module archai.supergraph.models.resnet_paper) resnet152() (in module archai.supergraph.models.resnet) resnet18() (in module archai.supergraph.models.resnet) resnet20() (in module archai.supergraph.models.resnet_paper) resnet32() (in module archai.supergraph.models.resnet_paper) resnet34() (in module archai.supergraph.models.resnet) resnet44() (in module archai.supergraph.models.resnet_paper) resnet50() (in module archai.supergraph.models.resnet) resnet56() (in module archai.supergraph.models.resnet_paper) resnet_orig() (in module archai.supergraph.models.resnet_orig) resnext101_32x8d() (in module archai.supergraph.models.resnet) resnext50_32x4d() (in module archai.supergraph.models.resnet) resolve_all() (in module archai.common.yaml_utils) restore_checkpoint() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.nas.search_combinations.SearchCombinations method) (archai.supergraph.utils.trainer.Trainer method) rgetattr() (in module archai.quantization.quantization_utils) roll() (archai.datasets.nlp.nvidia_data_loader_utils.LMMultiFileIterator method) (archai.datasets.nlp.nvidia_data_loader_utils.LMOrderedIterator method) root_node (archai.common.ordered_dict_logger.OrderedDictLogger property) Rotate() (in module archai.supergraph.datasets.augmentation) rotate_half() (in module archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) rsetattr() (in module archai.quantization.quantization_utils) run() (archai.supergraph.nas.exp_runner.ExperimentRunner method) run_epoch() (in module archai.supergraph.utils.augmented_trainer) run_eval() (archai.supergraph.nas.exp_runner.ExperimentRunner method) run_search() (archai.supergraph.nas.exp_runner.ExperimentRunner method) RunMetrics (class in archai.supergraph.utils.metrics) S sample() (archai.supergraph.algos.divnas.wmr.Wmr method) sample_config() (archai.discrete_search.search_spaces.config.arch_param_tree.ArchParamTree method) sample_from_hull() (in module archai.supergraph.algos.petridish.petridish_utils) sample_models() (archai.discrete_search.algos.bananas.MoBananasSearch method) (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) (archai.discrete_search.algos.local_search.LocalSearch method) (archai.discrete_search.algos.random_search.RandomSearch method) (archai.discrete_search.algos.regularized_evolution.RegularizedEvolutionSearch method) sample_sequence() (archai.supergraph.algos.divnas.seqopt.SeqOpt method) SamplePairing() (in module archai.supergraph.datasets.augmentation) save() (archai.common.config.Config method) (archai.common.delimited_text.DelimitedText method) (archai.common.ordered_dict_logger.OrderedDictLogger method) (archai.supergraph.nas.model_desc.ModelDesc method) (archai.supergraph.utils.metrics.Metrics method) save_2d_pareto_evolution_plot() (archai.discrete_search.api.search_results.SearchResults method) save_all_2d_pareto_evolution_plots() (archai.discrete_search.api.search_results.SearchResults method) save_all_checkpoints (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) save_arch() (archai.discrete_search.api.search_space.DiscreteSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) save_as_yaml() (in module archai.common.utils) save_cache() (archai.datasets.nlp.nvidia_dataset_provider_utils.Corpus method) (archai.discrete_search.api.search_objectives.SearchObjectives method) save_checkpoint() (in module archai.trainers.nlp.nvidia_trainer) save_hull() (in module archai.supergraph.algos.petridish.petridish_utils) save_hull_frontier() (in module archai.supergraph.algos.petridish.petridish_utils) save_model() (in module archai.common.ml_utils) save_model_weights() (archai.discrete_search.api.search_space.DiscreteSearchSpace method) (archai.discrete_search.search_spaces.benchmark.natsbench_tss.NatsbenchTssSearchSpace method) (archai.discrete_search.search_spaces.config.search_space.ConfigSearchSpace method) (archai.discrete_search.search_spaces.cv.segmentation_dag.search_space.SegmentationDagSearchSpace method) (archai.discrete_search.search_spaces.nlp.transformer_flex.search_space.TransformerFlexSearchSpace method) save_pareto_frontier_models() (archai.discrete_search.api.search_results.SearchResults method) save_search_state() (archai.discrete_search.api.search_results.SearchResults method) save_steps (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) save_trained() (archai.supergraph.nas.search_combinations.SearchCombinations method) scale_grad_by_freq (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) SEARCH (archai.supergraph.algos.petridish.petridish_utils.ExperimentStage attribute) (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) search() (archai.discrete_search.algos.bananas.MoBananasSearch method) (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) (archai.discrete_search.algos.local_search.LocalSearch method) (archai.discrete_search.algos.random_search.RandomSearch method) (archai.discrete_search.algos.regularized_evolution.RegularizedEvolutionSearch method) (archai.discrete_search.algos.successive_halving.SuccessiveHalvingSearch method) (archai.discrete_search.api.searcher.Searcher method) (archai.supergraph.algos.manual.manual_searcher.ManualSearcher method) (archai.supergraph.algos.petridish.searcher_petridish.SearcherPetridish method) (archai.supergraph.nas.search_combinations.SearchCombinations method) (archai.supergraph.nas.searcher.Searcher method) (in module archai.supergraph.utils.augmented_searcher) search_model_desc() (archai.supergraph.nas.searcher.Searcher method) search_model_desc_dist (archai.supergraph.algos.petridish.searcher_petridish.SearcherPetridish attribute) SEARCH_TRAINED (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) SearchCombinations (class in archai.supergraph.nas.search_combinations) SearchConstraint (class in archai.discrete_search.api.search_objectives) Searcher (class in archai.discrete_search.api.searcher) (class in archai.supergraph.nas.searcher) searcher() (archai.supergraph.algos.manual.manual_exp_runner.ManualExperimentRunner method) (archai.supergraph.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) SearcherPetridish (class in archai.supergraph.algos.petridish.searcher_petridish) SearchObjective (class in archai.discrete_search.api.search_objectives) SearchObjectives (class in archai.discrete_search.api.search_objectives) SearchResult (class in archai.supergraph.nas.searcher) SearchResults (class in archai.discrete_search.api.search_results) SEED (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) seed (archai.trainers.nlp.ds_training_args.DsTrainingArguments attribute) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) seed_cell() (archai.supergraph.nas.model_desc_builder.ModelDescBuilder method) SEED_TRAINED (archai.supergraph.algos.petridish.petridish_utils.JobStage attribute) SegmentationDagModel (class in archai.discrete_search.search_spaces.cv.segmentation_dag.model) SegmentationDagSearchSpace (class in archai.discrete_search.search_spaces.cv.segmentation_dag.search_space) select_edges() (archai.supergraph.nas.finalizers.Finalizers method) select_next_population() (archai.discrete_search.algos.evolution_pareto.EvolutionParetoSearch method) SelfAttention (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.mha) send() (archai.discrete_search.api.model_evaluator.AsyncModelEvaluator method) (archai.discrete_search.evaluators.progressive_training.RayProgressiveTraining method) (archai.discrete_search.evaluators.ray.RayParallelEvaluator method) (archai.discrete_search.evaluators.remote_azure_benchmark.RemoteAzureBenchmarkEvaluator method) SeparableConv1d (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sep_conv1d) SeparableConvBlock (class in archai.discrete_search.search_spaces.cv.segmentation_dag.ops) SepConv (class in archai.supergraph.nas.operations) seq_len (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) SeqOpt (class in archai.supergraph.algos.divnas.seqopt) set() (archai.common.stopwatch.StopWatch static method) set_arch_params() (archai.supergraph.nas.arch_module.ArchModule method) set_col() (archai.common.delimited_text.DelimitedText method) set_data() (archai.common.delimited_text.DelimitedText method) set_epoch() (archai.supergraph.datasets.distributed_stratified_sampler.DistributedStratifiedSampler method) set_global_instance() (archai.common.ordered_dict_logger.OrderedDictLogger static method) set_input_embeddings() (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenModel method) set_inst() (archai.common.config.Config static method) set_op_sampled_weights() (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp method) set_optim_lr() (in module archai.common.ml_utils) set_output_embeddings() (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenForCausalLM method) set_trainables() (archai.supergraph.nas.operations.Op method) setup_cuda() (in module archai.common.utils) SGConv (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) SGConv3 (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3) ShakeBlock (class in archai.supergraph.models.shakeshake.shake_resnet) ShakeBottleNeck (class in archai.supergraph.models.shakeshake.shake_resnext) ShakeDrop (class in archai.supergraph.models.shakedrop) ShakeDropFunction (class in archai.supergraph.models.shakedrop) ShakeResNet (class in archai.supergraph.models.shakeshake.shake_resnet) ShakeResNeXt (class in archai.supergraph.models.shakeshake.shake_resnext) ShakeShake (class in archai.supergraph.models.shakeshake.shakeshake) Sharpness() (in module archai.supergraph.datasets.augmentation) ShearX() (in module archai.supergraph.datasets.augmentation) ShearY() (in module archai.supergraph.datasets.augmentation) SHMArray (class in archai.datasets.nlp.fast_hf_dataset_provider_utils) Shortcut (class in archai.supergraph.models.shakeshake.shakeshake) should_refresh_cache() (in module archai.datasets.nlp.hf_dataset_provider_utils) SinusoidalEmbeddings (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention) SkipConnect (class in archai.supergraph.nas.operations) SmoothCrossEntropyLoss (class in archai.trainers.losses) Solarize() (in module archai.supergraph.datasets.augmentation) sparse (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) special_token_id() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) special_token_name() (archai.datasets.nlp.tokenizer_utils.token_config.TokenConfig method) SpecialTokenEnum (class in archai.datasets.nlp.tokenizer_utils.token_config) split_folder (archai.supergraph.datasets.providers.imagenet_folder.ImageNetFolder property) split_heads() (in module archai.discrete_search.search_spaces.nlp.tfpp.utils) Sport8Provider (class in archai.supergraph.datasets.providers.sport8_provider) StanfordCarsDatasetProvider (class in archai.datasets.cv.stanford_cars_dataset_provider) StanfordCarsProvider (class in archai.supergraph.datasets.providers.stanfordcars_provider) start() (archai.common.stopwatch.StopWatch method) (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) state_dict() (archai.common.apex_utils.ApexUtils method) (archai.supergraph.algos.darts.bilevel_optimizer.BilevelOptimizer method) (archai.supergraph.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method) (archai.supergraph.nas.model_desc.CellDesc method) (archai.supergraph.nas.model_desc.EdgeDesc method) (archai.supergraph.nas.model_desc.ModelDesc method) (archai.supergraph.nas.model_desc.NodeDesc method) (archai.supergraph.nas.model_desc.OpDesc method) (archai.supergraph.utils.metrics.Metrics method) (archai.supergraph.utils.multi_optim.MultiOptim method) (archai.supergraph.utils.tester.Tester method) (in module archai.common.utils) state_to_tensor (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv property) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 property) StatefulDistributedSampler (class in archai.trainers.nlp.ds_trainer) status() (in module archai.common.store) StemBase (class in archai.supergraph.nas.operations) StemConv3x3 (class in archai.supergraph.nas.operations) StemConv3x3Relu (class in archai.supergraph.nas.operations) StemConv3x3S4 (class in archai.supergraph.nas.operations) StemConv3x3S4S2 (class in archai.supergraph.nas.operations) step() (archai.common.apex_utils.ApexUtils method) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.Normalization method) (archai.supergraph.algos.darts.bilevel_optimizer.BilevelOptimizer method) (archai.supergraph.algos.darts.bilevel_optimizer_slow.BilevelOptimizer method) (archai.supergraph.utils.multi_optim.MultiOptim method) (archai.trainers.coin_betting_optimizer.CocobBackprop method) (archai.trainers.coin_betting_optimizer.CocobOns method) (archai.trainers.gradual_warmup_scheduler.GradualWarmupScheduler method) (archai.trainers.lamb_optimizer.JITLamb method) (archai.trainers.lamb_optimizer.Lamb method) step_time_avg() (archai.supergraph.utils.metrics.RunMetrics method) stop() (archai.discrete_search.evaluators.pt_profiler_utils.pt_profiler_model.ProfilerModel method) StopForward (class in archai.supergraph.algos.petridish.petridish_op) StopForwardReductionOp (class in archai.supergraph.algos.petridish.petridish_op) StopGradient (class in archai.supergraph.algos.petridish.petridish_op) StopGradientReduction (class in archai.supergraph.algos.petridish.petridish_op) stopwatch (archai.common.stopwatch.StopWatch attribute) StopWatch (class in archai.common.stopwatch) strategy (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) stream_iterator() (archai.datasets.nlp.nvidia_data_loader_utils.LMMultiFileIterator method) stride (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) subscribe() (archai.supergraph.utils.checkpoint.CheckPoint method) subscribe_start_iteration() (archai.discrete_search.api.searcher.Searcher method) SubsetSampler (class in archai.supergraph.datasets.data) SuccessiveHalvingSearch (class in archai.discrete_search.algos.successive_halving) summary() (archai.supergraph.nas.model.Model method) (in module archai.common.model_summary) summary_string() (in module archai.common.model_summary) SummaryWriterDummy (class in archai.common.common) SUPPORTED_DATASETS (archai.datasets.cv.caltech_dataset_provider.CaltechDatasetProvider attribute) (archai.datasets.cv.cifar_dataset_provider.CifarDatasetProvider attribute) (archai.datasets.cv.coco_dataset_provider.CocoDatasetProvider attribute) (archai.datasets.cv.mnist_dataset_provider.MnistDatasetProvider attribute) SVHNDatasetProvider (class in archai.datasets.cv.svhn_dataset_provider) SvhnProvider (class in archai.supergraph.datasets.providers.svhn_provider) sync_devices() (archai.common.apex_utils.ApexUtils method) sync_workers() (in module archai.common.distributed_utils) T temperature (archai.trainers.nlp.hf_training_args.DistillerTrainingArguments attribute) TempIdentityOp (class in archai.supergraph.algos.petridish.petridish_op) TemporaryFiles (class in archai.common.file_utils) TensorpackLmdbDataset (class in archai.datasets.cv.tensorpack_lmdb_dataset_provider_utils) TensorpackLmdbDatasetProvider (class in archai.datasets.cv.tensorpack_lmdb_dataset_provider) terminator (archai.common.atomic_file_handler.AtomicFileHandler attribute) test() (archai.supergraph.utils.tester.Tester method) Tester (class in archai.supergraph.utils.tester) TfppSearchSpace (class in archai.discrete_search.search_spaces.nlp.tfpp.search_space) thompson_sampling() (archai.discrete_search.algos.bananas.MoBananasSearch method) tie_weights() (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerLMHeadModel method) to_amp() (archai.common.apex_utils.ApexUtils method) to_config() (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) to_cpu() (archai.discrete_search.predictors.dnn_ensemble.PredictiveDNNEnsemble method) to_cuda() (archai.discrete_search.predictors.dnn_ensemble.PredictiveDNNEnsemble method) to_dict() (archai.common.config.Config method) (archai.discrete_search.search_spaces.config.arch_config.ArchConfig method) (archai.discrete_search.search_spaces.config.arch_config.ArchConfigList method) (archai.discrete_search.search_spaces.config.arch_param_tree.ArchParamTree method) (archai.trainers.nlp.ds_training_args.DsTrainingArguments method) (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments method) to_file() (archai.discrete_search.search_spaces.config.arch_config.ArchConfig method) (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) to_float() (archai.quantization.modules.FakeDynamicQuantConv1d method) (archai.quantization.modules.FakeDynamicQuantLinear method) (archai.quantization.modules.FakeQuantEmbedding method) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1D method) to_hash() (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) to_tuple() (in module archai.discrete_search.search_spaces.nlp.tfpp.search_space) token_to_id() (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) TokenConfig (class in archai.datasets.nlp.tokenizer_utils.token_config) tokenize_concatenated_dataset() (in module archai.datasets.nlp.hf_dataset_provider_utils) tokenize_contiguous_dataset() (in module archai.datasets.nlp.hf_dataset_provider_utils) tokenize_dataset() (in module archai.datasets.nlp.hf_dataset_provider_utils) tokenize_nsp_dataset() (in module archai.datasets.nlp.hf_dataset_provider_utils) TokenizerBase (class in archai.datasets.nlp.tokenizer_utils.tokenizer_base) tokens_to_ids() (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) torch_call() (archai.datasets.nlp.fast_hf_dataset_provider.FastDataCollatorForLanguageModeling method) TorchFlops (class in archai.discrete_search.evaluators.pt_profiler) TorchLatency (class in archai.discrete_search.evaluators.pt_profiler) TorchMacs (class in archai.discrete_search.evaluators.pt_profiler) TorchNumParameters (class in archai.discrete_search.evaluators.pt_profiler) TorchPeakCpuMemory (class in archai.discrete_search.evaluators.pt_profiler) TorchPeakCudaMemory (class in archai.discrete_search.evaluators.pt_profiler) train() (archai.api.trainer_base.TrainerBase method) (archai.datasets.nlp.tokenizer_utils.bbpe_tokenizer.BbpeTokenizer method) (archai.datasets.nlp.tokenizer_utils.tokenizer_base.TokenizerBase method) (archai.datasets.nlp.tokenizer_utils.word_tokenizer.WordTokenizer method) (archai.trainers.cv.pl_trainer.PlTrainer method) (archai.trainers.nlp.ds_trainer.DsTrainer method) (archai.trainers.nlp.nvidia_trainer.NvidiaTrainer method) train_and_encode() (archai.datasets.nlp.nvidia_dataset_provider_utils.Corpus method) train_and_eval() (in module archai.supergraph.utils.augmented_trainer) train_batch_without_pipe_parallel() (archai.trainers.nlp.ds_trainer.DsTrainer method) train_model() (archai.supergraph.nas.evaluater.Evaluater method) train_model_desc() (archai.supergraph.nas.searcher.Searcher method) train_model_desc_dist (archai.supergraph.algos.petridish.searcher_petridish.SearcherPetridish attribute) Trainer (class in archai.supergraph.utils.trainer) trainer_class() (archai.supergraph.algos.darts.darts_exp_runner.DartsExperimentRunner method) (archai.supergraph.algos.didarts.didarts_exp_runner.DiDartsExperimentRunner method) (archai.supergraph.algos.divnas.divnas_exp_runner.DivnasExperimentRunner method) (archai.supergraph.algos.gumbelsoftmax.gs_exp_runner.GsExperimentRunner method) (archai.supergraph.algos.manual.manual_exp_runner.ManualExperimentRunner method) (archai.supergraph.algos.nasbench101.nasbench101_exp_runner.PetridishExperimentRunner method) (archai.supergraph.algos.petridish.petridish_exp_runner.PetridishExperimentRunner method) (archai.supergraph.algos.random.random_exp_runner.RandomExperimentRunner method) (archai.supergraph.algos.xnas.xnas_exp_runner.XnasExperimentRunner method) (archai.supergraph.nas.exp_runner.ExperimentRunner method) TrainerBase (class in archai.api.trainer_base) training (archai.discrete_search.predictors.dnn_ensemble.FFEnsembleMember attribute) (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel attribute) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.Block attribute) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.NormalConvBlock attribute) (archai.discrete_search.search_spaces.cv.segmentation_dag.ops.SeparableConvBlock attribute) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block.CodeGenBlock attribute) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.block.Mlp attribute) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenForCausalLM attribute) (archai.discrete_search.search_spaces.nlp.tfpp.backbones.codegen.model.CodeGenModel attribute) (archai.discrete_search.search_spaces.nlp.tfpp.mixed_attention.MixedAttentionBlock attribute) (archai.discrete_search.search_spaces.nlp.tfpp.mixed_op.MixedAttentionBlock attribute) (archai.discrete_search.search_spaces.nlp.tfpp.model.LanguageModel attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.causal_self_attn.CausalSelfAttention attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.LocalAttention attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.LocalMHA attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.local_attention.SinusoidalEmbeddings attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.lsh_attn.LSHAttention attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.BaseRotaryEmbedding attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.MHA attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.mha.SelfAttention attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sep_conv1d.SeparableConv1d attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.GConv attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.Modrelu attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.modrelu attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.Normalization attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.SGConv attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.TransposedLinear attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv.TransposedLN attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.GConv3 attribute) (archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv3.SGConv3 attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexAttention attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexBlock attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexLMHeadModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexMLP attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_gpt2_flex.GPT2FlexModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerLMHeadModel attribute) (archai.discrete_search.search_spaces.nlp.transformer_flex.models.modeling_mem_transformer.MemTransformerModel attribute) (archai.quantization.mixed_qat.MixedQAT attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) (archai.quantization.modules.FakeDynamicQuantLinearForOnnx attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1D attribute) (archai.quantization.nlp.modules.FakeDynamicQuantHFConv1DForOnnx attribute) (archai.quantization.quantizers.FakeDynamicQuant attribute) (archai.supergraph.algos.darts.mixed_op.MixedOp attribute) (archai.supergraph.algos.divnas.divop.DivOp attribute) (archai.supergraph.algos.gumbelsoftmax.gs_op.GsOp attribute) (archai.supergraph.algos.nasbench101.base_ops.Conv1x1BnRelu attribute) (archai.supergraph.algos.nasbench101.base_ops.Conv3x3BnRelu attribute) (archai.supergraph.algos.nasbench101.base_ops.ConvBnRelu attribute) (archai.supergraph.algos.nasbench101.base_ops.MaxPool3x3 attribute) (archai.supergraph.algos.nasbench101.model.Cell attribute) (archai.supergraph.algos.nasbench101.model.Network attribute) (archai.supergraph.algos.nasbench101.nasbench101_op.NasBench101Op attribute) (archai.supergraph.algos.petridish.petridish_op.PetridishOp attribute) (archai.supergraph.algos.petridish.petridish_op.StopForward attribute) (archai.supergraph.algos.petridish.petridish_op.StopForwardReductionOp attribute) (archai.supergraph.algos.petridish.petridish_op.StopGradient attribute) (archai.supergraph.algos.petridish.petridish_op.StopGradientReduction attribute) (archai.supergraph.algos.petridish.petridish_op.TempIdentityOp attribute) (archai.supergraph.algos.xnas.xnas_op.XnasOp attribute) (archai.supergraph.models.alexnet.AlexNet attribute) (archai.supergraph.models.densenet.DenseNet attribute) (archai.supergraph.models.googlenet.GoogLeNet attribute) (archai.supergraph.models.inception.Inception3 attribute) (archai.supergraph.models.mobilenetv2.MobileNetV2 attribute) (archai.supergraph.models.pyramidnet.BasicBlock attribute) (archai.supergraph.models.pyramidnet.Bottleneck attribute) (archai.supergraph.models.pyramidnet.PyramidNet attribute) (archai.supergraph.models.resnet.ResNet attribute) (archai.supergraph.models.resnet_paper.ResNet attribute) (archai.supergraph.models.shakedrop.ShakeDrop attribute) (archai.supergraph.models.shakeshake.shake_resnet.ShakeBlock attribute) (archai.supergraph.models.shakeshake.shake_resnet.ShakeResNet attribute) (archai.supergraph.models.shakeshake.shake_resnext.ShakeBottleNeck attribute) (archai.supergraph.models.shakeshake.shake_resnext.ShakeResNeXt attribute) (archai.supergraph.models.shakeshake.shakeshake.Shortcut attribute) (archai.supergraph.models.vgg.VGG attribute) (archai.supergraph.models.wideresnet.WideBasic attribute) (archai.supergraph.models.wideresnet.WideResNet attribute) (archai.supergraph.nas.arch_module.ArchModule attribute) (archai.supergraph.nas.cell.Cell attribute) (archai.supergraph.nas.dag_edge.DagEdge attribute) (archai.supergraph.nas.model.AuxTower attribute) (archai.supergraph.nas.model.Model attribute) (archai.supergraph.nas.operations.AvgPool2d7x7 attribute) (archai.supergraph.nas.operations.ConcateChannelsOp attribute) (archai.supergraph.nas.operations.ConvBNReLU attribute) (archai.supergraph.nas.operations.DilConv attribute) (archai.supergraph.nas.operations.DropPath_ attribute) (archai.supergraph.nas.operations.FacConv attribute) (archai.supergraph.nas.operations.FactorizedReduce attribute) (archai.supergraph.nas.operations.Identity attribute) (archai.supergraph.nas.operations.LinearOp attribute) (archai.supergraph.nas.operations.MergeOp attribute) (archai.supergraph.nas.operations.MultiOp attribute) (archai.supergraph.nas.operations.Op attribute) (archai.supergraph.nas.operations.PoolAdaptiveAvg2D attribute) (archai.supergraph.nas.operations.PoolBN attribute) (archai.supergraph.nas.operations.PoolMeanTensor attribute) (archai.supergraph.nas.operations.ProjectChannelsOp attribute) (archai.supergraph.nas.operations.ReLUConvBN attribute) (archai.supergraph.nas.operations.SepConv attribute) (archai.supergraph.nas.operations.SkipConnect attribute) (archai.supergraph.nas.operations.StemBase attribute) (archai.supergraph.nas.operations.StemConv3x3 attribute) (archai.supergraph.nas.operations.StemConv3x3Relu attribute) (archai.supergraph.nas.operations.StemConv3x3S4 attribute) (archai.supergraph.nas.operations.StemConv3x3S4S2 attribute) (archai.supergraph.nas.operations.Zero attribute) TransformerFlexOnnxLatency (class in archai.discrete_search.evaluators.nlp.transformer_flex_latency) TransformerFlexOnnxMemory (class in archai.discrete_search.evaluators.nlp.transformer_flex_memory) TransformerFlexSearchSpace (class in archai.discrete_search.search_spaces.nlp.transformer_flex.search_space) TransfoXLOnnxModel (class in archai.onnx.optimization_utils.transfo_xl_onnx_model) TranslateX() (in module archai.supergraph.datasets.augmentation) TranslateXAbs() (in module archai.supergraph.datasets.augmentation) TranslateY() (in module archai.supergraph.datasets.augmentation) TranslateYAbs() (in module archai.supergraph.datasets.augmentation) transposed (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) TransposedLinear (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) TransposedLN (class in archai.discrete_search.search_spaces.nlp.tfpp.ops.sgconv) Truncate() (in module archai.supergraph.algos.nasbench101.model) U UNK (archai.datasets.nlp.tokenizer_utils.token_config.SpecialTokenEnum attribute) unlock() (archai.common.store.ArchaiStore method) (in module archai.common.store) unlock_all() (archai.common.store.ArchaiStore method) unlock_entity() (archai.common.store.ArchaiStore method) update() (archai.common.utils.AverageMeter method) (archai.supergraph.algos.divnas.seqopt.SeqOpt method) (archai.supergraph.algos.divnas.wmr.Wmr method) update_alphas() (archai.supergraph.algos.xnas.xnas_op.XnasOp method) update_checkpoint() (archai.supergraph.algos.darts.bilevel_arch_trainer.BilevelArchTrainer method) (archai.supergraph.algos.xnas.xnas_arch_trainer.XnasArchTrainer method) (archai.supergraph.utils.trainer.Trainer method) update_covs() (archai.supergraph.algos.divnas.divnas_cell.Divnas_Cell method) update_envvars() (in module archai.common.common) update_status() (archai.common.store.ArchaiStore method) update_status_entity() (archai.common.store.ArchaiStore method) upload() (archai.common.store.ArchaiStore method) (in module archai.common.store) upload_blob() (archai.common.store.ArchaiStore method) uri2path() (in module archai.common.utils) use_dynamic_axes() (archai.onnx.optimization_utils.transfo_xl_onnx_model.TransfoXLOnnxModel method) use_raw_attention_mask() (archai.onnx.optimization_utils.fusion_options.FusionOptions method) use_shifted_labels (archai.datasets.nlp.fast_hf_dataset_provider.FastDataCollatorForLanguageModeling attribute) USPSDatasetProvider (class in archai.datasets.cv.usps_dataset_provider) V valid_splits (archai.supergraph.datasets.providers.imagenet_folder.ImageNetFolder property) validate_constraints() (archai.discrete_search.api.search_objectives.SearchObjectives method) validate_forward() (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) validate_onnx_outputs() (in module archai.onnx.export) var (archai.discrete_search.api.predictor.MeanVar attribute) VGG (class in archai.supergraph.models.vgg) vgg11() (in module archai.supergraph.models.vgg) vgg11_bn() (in module archai.supergraph.models.vgg) vgg13() (in module archai.supergraph.models.vgg) vgg13_bn() (in module archai.supergraph.models.vgg) vgg16() (in module archai.supergraph.models.vgg) vgg16_bn() (in module archai.supergraph.models.vgg) vgg19() (in module archai.supergraph.models.vgg) vgg19_bn() (in module archai.supergraph.models.vgg) view() (archai.discrete_search.search_spaces.cv.segmentation_dag.model.SegmentationDagModel method) visualize() (archai.supergraph.algos.nasbench101.model_spec.ModelSpec method) vocab_size (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) vocab_type (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) W warn() (archai.common.ordered_dict_logger.OrderedDictLogger method) weight (archai.quantization.modules.FakeDynamicQuantConv1d attribute) (archai.quantization.modules.FakeDynamicQuantConv1dForOnnx attribute) (archai.quantization.modules.FakeDynamicQuantLinear attribute) (archai.quantization.modules.FakeDynamicQuantLinearForOnnx attribute) (archai.quantization.modules.FakeQuantEmbedding attribute) (archai.quantization.modules.FakeQuantEmbeddingForOnnx attribute) weight_decay (archai.trainers.nlp.nvidia_training_args.NvidiaTrainingArguments attribute) weight_sharing() (in module archai.onnx.export_utils) weights (archai.supergraph.algos.divnas.wmr.Wmr property) WideBasic (class in archai.supergraph.models.wideresnet) WideResNet (class in archai.supergraph.models.wideresnet) Wmr (class in archai.supergraph.algos.divnas.wmr) WordTokenizer (class in archai.datasets.nlp.tokenizer_utils.word_tokenizer) write_string() (in module archai.common.utils) X XnasArchTrainer (class in archai.supergraph.algos.xnas.xnas_arch_trainer) XnasExperimentRunner (class in archai.supergraph.algos.xnas.xnas_exp_runner) XnasModelDescBuilder (class in archai.supergraph.algos.xnas.xnas_model_desc_builder) XnasOp (class in archai.supergraph.algos.xnas.xnas_op) xor() (in module archai.datasets.nlp.fast_hf_dataset_provider_utils) Z Zero (class in archai.supergraph.nas.operations) zero_file() (in module archai.common.utils) zero_grad() (archai.supergraph.utils.multi_optim.MultiOptim method) zip_eq() (in module archai.common.utils)
