- __abs__() (Array method)
- __add__() (Array method)
- __adjust_cols_sp1_m_ls_n() (SquareDiagTiles method), [1], [2]
- __adjust_last_row_sp0_m_ge_n() (SquareDiagTiles method), [1], [2]
- __adjust_lshape_sp0_1tile() (SquareDiagTiles method), [1], [2]
- __allgather_like() (MPICommunication method), [1], [2]
- __alltoall_like() (MPICommunication method), [1], [2]
- __and__() (Array method)
- __array (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __array__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __array_function__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __array_namespace__() (Array method)
- __array_namespace_info__() (in module heat.array_api._inspection)
- __array_ufunc__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __balanced (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __bool__() (Array method)
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __broadcast_like() (MPICommunication method), [1], [2]
- __cast() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __cat_halo() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __check_partial_fit_first_call() (GaussianNB method), [1]
- __col_inds (SquareDiagTiles attribute), [1], [2]
- __col_per_proc_list (SquareDiagTiles attribute), [1], [2]
- __comm (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __complex__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __create_cols() (SquareDiagTiles method), [1], [2]
- __create_random_np_array() (TestCase method)
- __def_end_row_inds_sp0_m_ge_n() (SquareDiagTiles method), [1], [2]
- __del__() (MPICommunication method), [1], [2]
- __derived_op() (MPICommunication method), [1], [2]
- __device (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __device_id (Device attribute), [1], [2]
- __device_type (Device attribute), [1], [2]
- __dlpack__() (Array method)
- __dlpack_device__() (Array method)
- __DNDarray (SplitTiles attribute), [1], [2]
- __dtype (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __eq__() (Array method)
- __float__() (Array method)
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __floordiv__() (Array method)
- __gather_like() (MPICommunication method), [1], [2]
- __ge__() (Array method)
- __get_tile_slices() (SplitTiles method), [1], [2]
- __getattr__() (in module heat.nn)
- __getitem__() (Array method)
- (Dataset method), [1]
- (DistributedDataset method), [1]
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- (PartialH5Dataset method), [1]
- (SplitTiles method), [1], [2]
- (SquareDiagTiles method), [1], [2]
- __getitems__() (DistributedDataset method), [1]
- __gshape (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __gt__() (Array method)
- __halo_next (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __halo_prev (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __index__() (Array method)
- __init_checktypes() (DASO method)
- __int__() (Array method)
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __invert__() (Array method)
- __ishalo (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __iter__() (DataLoader method), [1]
- __joint_log_likelihood() (GaussianNB method), [1]
- __key_adds_dimension() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __key_is_singular() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __lam (Lasso attribute), [1]
- __last_diag_pr (SquareDiagTiles attribute), [1], [2]
- __last_tile_row_adjust_sp1() (SquareDiagTiles method), [1], [2]
- __le__() (Array method)
- __len__() (Array method)
- (DataLoader method), [1]
- (Dataset method), [1]
- (DistributedDataset method), [1]
- (DistributedSampler method), [1]
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- (PartialH5DataLoaderIter method), [1]
- (PartialH5Dataset method), [1]
- __lshape_map (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __lshift__() (Array method)
- __lt__() (Array method)
- __matmul__() (Array method)
- __mod__() (Array method)
- __mpi_dtype2ctype (MPICommunication attribute), [1], [2]
- __mpi_type_mappings (MPICommunication attribute), [1], [2]
- __mul__() (Array method)
- __ne__() (Array method)
- __neg__() (Array method)
- __next__() (PartialH5DataLoaderIter method), [1]
- __or__() (Array method)
- __pack_data() (DASO method)
- __partial_fit() (GaussianNB method), [1]
- __partitions_dict__ (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __pos__() (Array method)
- __pow__() (Array method)
- __prephalo() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __radd__() (Array method)
- __redistribute_shuffle() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __reduce_like() (MPICommunication method), [1], [2]
- __repr__() (Array method)
- (BaseEstimator method), [1], [2]
- (Device method), [1], [2]
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __rfloordiv__() (Array method)
- __rmod__() (Array method)
- __rmul__() (Array method)
- __row_inds (SquareDiagTiles attribute), [1], [2]
- __row_per_proc_list (SquareDiagTiles attribute), [1], [2]
- __rpow__() (Array method)
- __rshift__() (Array method)
- __rsub__() (Array method)
- __rtruediv__() (Array method)
- __scatter_like() (MPICommunication method), [1], [2]
- __send_like() (MPICommunication method), [1], [2]
- __setattr__() (DataParallel method)
- __setitem__() (Array method)
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- (SplitTiles method), [1], [2]
- (SquareDiagTiles method), [1], [2]
- __setter() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __split (DNDarray attribute), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __str__() (Array method)
- (Device method), [1], [2]
- (DMD method), [1]
- (DMDc method), [1]
- (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __sub__() (Array method)
- __theta (Lasso attribute), [1]
- __thread_convert_all() (PartialH5DataLoaderIter method), [1]
|
- __tile_dims (SplitTiles attribute), [1], [2]
- __tile_ends_g (SplitTiles attribute), [1], [2]
- __tile_locations (SplitTiles attribute), [1], [2]
- __tile_map (SquareDiagTiles attribute), [1], [2]
- __torch_device (Device attribute), [1], [2]
- __torch_function__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __torch_proxy__() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __truediv__() (Array method)
- __update_mean_variance() (GaussianNB method), [1]
- __version__ (in module heat), [1]
- __xitem_get_key_start_stop() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- __xor__() (Array method)
- _active_layers (DataParallel attribute)
- _alltoall_shuffle() (DistributedSampler method), [1]
- _array (Array attribute)
- _assign_to_cluster() (_KCluster method), [1], [2], [3], [4]
- _auto_collation (PartialH5DataLoaderIter attribute), [1]
- _base_seed (PartialH5DataLoaderIter attribute), [1]
- _BatchParallelKCluster (class in heat.cluster)
- _blocking_hook() (DataParallel method)
- _can_be_cast_intuitively_to (datatype attribute), [1], [2]
- _can_be_cast_safely_to (bool attribute), [1], [2]
- (complex128 attribute), [1], [2]
- (complex64 attribute), [1], [2]
- (datatype attribute), [1], [2]
- (float16 attribute), [1], [2]
- (float32 attribute), [1], [2]
- (float64 attribute), [1], [2]
- (int16 attribute), [1], [2]
- (int32 attribute), [1], [2]
- (int64 attribute), [1], [2]
- (int8 attribute), [1], [2]
- (uint8 attribute), [1], [2]
- _centroid_sampling_helper() (_KCluster method), [1], [2], [3], [4]
- _check_allowed_dtypes() (Array method)
- _check_gpu_aware_mpi() (in module heat)
- _cluster_centers (_BatchParallelKCluster attribute), [1]
- _collate_fn (PartialH5DataLoaderIter attribute), [1]
- _create_recursive_vectortype() (MPICommunication method), [1], [2]
- _cut_slice (Dataset attribute), [1]
- _dataset_fetcher (PartialH5DataLoaderIter attribute), [1]
- _dataset_kind (PartialH5DataLoaderIter attribute), [1]
- _dp_optimizers (DataParallel attribute)
- _drop_last (PartialH5DataLoaderIter attribute), [1]
- _first_iter (DataLoader attribute), [1]
- _forward_hook() (DataParallel method)
- _full_dataset_shuffle_iter() (DataLoader method), [1]
- _functional_value (_BatchParallelKCluster attribute), [1]
- _fwd_hook_handles (DataParallel attribute)
- _get_mpi_library() (in module heat)
- _global_sync() (DASO method)
- _gs8_waited (DASO attribute)
- _gs8_waits (DASO attribute)
- _gs_create_param_dict() (DASO method)
- _gs_rcv_update_params() (DASO method)
- _gs_rcv_update_params_last_batch() (DASO method)
- _gs_send_params() (DASO method)
- _handle_large_count() (MPICommunication method), [1], [2]
- _has_n_features() (in module heat.preprocessing)
- _hostnames (TestCase attribute)
- _in_slice() (DistributedSampler method), [1]
- _index_sampler (PartialH5DataLoaderIter attribute), [1]
- _inertia (_KCluster attribute), [1], [2], [3], [4]
- _init (_BatchParallelKCluster attribute), [1]
- _init() (finfo method), [1]
- _init_is_better() (DetectMetricPlateau method)
- _initialize_cluster_centers() (_KCluster method), [1], [2], [3], [4]
- _initialize_plus_plus() (in module heat.cluster)
- _iparam_update() (DataParallel method)
- _is_2D_float_DNDarray() (in module heat.preprocessing)
- _isvd() (in module heat.decomposition)
- _IterableDataset_len_called (PartialH5DataLoaderIter attribute), [1]
- _KCluster (class in heat.cluster), [1], [2], [3]
- _kmex() (in module heat.cluster), [1]
- _labels (_KCluster attribute), [1], [2], [3], [4]
- _layer_wait_handles (DataParallel attribute)
- _local_update() (DASO method)
- _metric (_KCluster attribute), [1], [2], [3], [4]
- _minmax_op() (MPICommunication method), [1], [2]
- _moveToCompDevice() (MPICommunication method), [1], [2]
- _n_iter (_BatchParallelKCluster attribute), [1]
- _new() (Array method)
- _next_data() (PartialH5DataLoaderIter method), [1]
- _nonblocking_hook() (DataParallel method)
- _normalize_two_args() (Array method)
- _normalized_symmetric_L() (Laplacian method), [1]
- _num_workers (PartialH5DataLoaderIter attribute), [1]
- _num_yielded (PartialH5DataLoaderIter attribute), [1]
- _p (_BatchParallelKCluster attribute), [1]
- _parallel_batched_kmex_predict() (in module heat.cluster)
- _param_indices (DataParallel attribute)
- _param_send_buffer_size (DASO attribute)
- _param_send_shp (DASO attribute)
- _param_slices (DataParallel attribute)
- _parameter_names() (BaseEstimator method), [1], [2]
- _pin_memory (PartialH5DataLoaderIter attribute), [1]
- _prev_params (DASO attribute)
- _promote_scalar() (Array method)
- _repr_pretty_() (DNDarray method), [1], [2], [3], [4], [5], [6], [7], [8], [9], [10], [11], [12], [13], [14]
- _reset_parameters() (DataParallel method)
- _result_type() (in module heat.array_api._dtypes)
- _sampler_iter (PartialH5DataLoaderIter attribute), [1]
- _shuffle() (DistributedSampler method), [1]
- _simple_L() (Laplacian method), [1]
- _spectral_embedding() (Spectral method), [1]
- _start_local_sync() (DASO method)
- _stop_local_sync() (DASO method)
- _timeout (PartialH5DataLoaderIter attribute), [1]
- _tol_wrt_dtype() (in module heat.preprocessing)
- _torch_matrix_diag() (in module heat.decomposition)
- _update_centroids() (_KCluster method), [1], [2], [3], [4]
- _validate_index() (Array method)
- _validate_input() (in module heat.cluster)
|