Home
last modified time | relevance | path

Searched defs:__constants__ (Results 1 – 25 of 43) sorted by relevance

12

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dactivation.py76 __constants__ = ["threshold", "value", "inplace"] variable in Threshold
125 __constants__ = ["inplace"] variable in ReLU
178 __constants__ = ["lower", "upper", "inplace"] variable in RReLU
233 __constants__ = ["min_val", "max_val", "inplace"] variable in Hardtanh
358 __constants__ = ["inplace"] variable in Hardsigmoid
424 __constants__ = ["inplace"] variable in SiLU
463 __constants__ = ["inplace"] variable in Mish
508 __constants__ = ["inplace"] variable in Hardswish
551 __constants__ = ["alpha", "inplace"] variable in ELU
596 __constants__ = ["alpha", "inplace"] variable in CELU
[all …]
H A Dloss.py122 __constants__ = ["reduction"] variable in L1Loss
236 __constants__ = ["ignore_index", "reduction"] variable in NLLLoss
334 __constants__ = ["log_input", "full", "eps", "reduction"] variable in PoissonNLLLoss
430 __constants__ = ["full", "eps", "reduction"] variable in GaussianNLLLoss
528 __constants__ = ["reduction"] variable in KLDivLoss
602 __constants__ = ["reduction"] variable in MSELoss
685 __constants__ = ["reduction"] variable in BCELoss
877 __constants__ = ["margin", "reduction"] variable in HingeEmbeddingLoss
951 __constants__ = ["reduction"] variable in MultiLabelMarginLoss
1031 __constants__ = ["reduction"] variable in SmoothL1Loss
[all …]
H A Dpooling.py46 __constants__ = [ variable in _MaxPoolNd
568 __constants__ = [ variable in _AvgPoolNd
723 __constants__ = [ variable in AvgPool2d
836 __constants__ = [ variable in AvgPool3d
928 __constants__ = ["kernel_size", "return_indices", "output_size", "output_ratio"] variable in FractionalMaxPool2d
1014 __constants__ = ["kernel_size", "return_indices", "output_size", "output_ratio"] variable in FractionalMaxPool3d
1065 __constants__ = ["norm_type", "kernel_size", "stride", "ceil_mode"] variable in _LPPoolNd
1247 __constants__ = ["output_size", "return_indices"] variable in _AdaptiveMaxPoolNd
1381 __constants__ = ["output_size"] variable in _AdaptiveAvgPoolNd
H A Dpadding.py34 __constants__ = ["padding"] variable in _CircularPadNd
206 __constants__ = ["padding", "value"] variable in _ConstantPadNd
315 __constants__ = ["padding", "value"] variable in ConstantPad2d
364 __constants__ = ["padding"] variable in _ReflectionPadNd
523 __constants__ = ["padding"] variable in _ReplicationPadNd
H A Dnormalization.py47 __constants__ = ["size", "alpha", "beta", "k"] variable in LocalResponseNorm
172 __constants__ = ["normalized_shape", "eps", "elementwise_affine"] variable in LayerNorm
274 __constants__ = ["num_groups", "num_channels", "eps", "affine"] variable in GroupNorm
361 __constants__ = ["normalized_shape", "eps", "elementwise_affine"] variable in RMSNorm
H A Dtransformer.py329 __constants__ = ["norm"] variable in TransformerEncoder
543 __constants__ = ["norm"] variable in TransformerDecoder
694 __constants__ = ["norm_first"] variable in TransformerEncoderLayer
973 __constants__ = ["norm_first"] variable in TransformerDecoderLayer
H A Dpixelshuffle.py51 __constants__ = ["upscale_factor"] variable in PixelShuffle
104 __constants__ = ["downscale_factor"] variable in PixelUnshuffle
H A Ddistance.py44 __constants__ = ["norm", "eps", "keepdim"] variable in PairwiseDistance
83 __constants__ = ["dim", "eps"] variable in CosineSimilarity
H A Dfold.py123 __constants__ = ["output_size", "kernel_size", "dilation", "padding", "stride"] variable in Fold
287 __constants__ = ["kernel_size", "dilation", "padding", "stride"] variable in Unfold
H A Dflatten.py43 __constants__ = ["start_dim", "end_dim"] variable in Flatten
108 __constants__ = ["dim", "unflattened_size"] variable in Unflatten
H A Dlinear.py88 __constants__ = ["in_features", "out_features"] variable in Linear
188 __constants__ = ["in1_features", "in2_features", "out_features"] variable in Bilinear
H A Dsparse.py113 __constants__ = [ variable in Embedding
347 __constants__ = [ variable in EmbeddingBag
H A Drnn.py61 __constants__ = [ variable in RNNBase
1433 __constants__ = ["input_size", "hidden_size", "bias"] variable in RNNCellBase
1545 __constants__ = ["input_size", "hidden_size", "bias", "nonlinearity"] variable in RNNCell
H A Dchannelshuffle.py45 __constants__ = ["groups"] variable in ChannelShuffle
H A Dupsampling.py137 __constants__ = [ variable in Upsample
H A Ddropout.py18 __constants__ = ["p", "inplace"] variable in _DropoutNd
/aosp_15_r20/external/pytorch/benchmarks/fastrnns/
H A Dcustom_lstms.py256 __constants__ = ["directions"] variable in BidirLSTMLayer
293 __constants__ = ["layers"] # Necessary for iterating through self.layers variable in StackedLSTM
323 __constants__ = ["layers"] # Necessary for iterating through self.layers variable in StackedLSTM2
351 __constants__ = ["layers", "num_layers"] variable in StackedLSTMWithDropout
/aosp_15_r20/external/pytorch/torch/utils/
H A Dmkldnn.py40 __constants__ = ['stride', 'padding', 'dilation', 'groups'] variable in _MkldnnConvNd
135 __constants__ = ['exponential_average_factor', 'eps'] variable in MkldnnBatchNorm
/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dquantize.py394 __constants__ = ["in_features", "out_features"] variable in WeightOnlyInt8Linear
450 __constants__ = ["in_features", "out_features"] variable in Int8DynActInt8WeightLinear
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_type_sharing.py86 __constants__ = ["const"] variable in TestTypeSharing.test_constants.M
172 __constants__ = ["const"] variable in TestTypeSharing.test_same_but_different_classes.A
186 __constants__ = ["const"] variable in TestTypeSharing.test_same_but_different_classes.B
H A Dtest_models.py637 __constants__ = ["downsample"] variable in TestModels.test_script_module_script_resnet.BasicBlock
669 __constants__ = ["layer1", "layer2", "layer3", "layer4"] variable in TestModels.test_script_module_script_resnet.ResNet
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dquantize.py83 __constants__ = ["in_features", "out_features"] variable in WeightOnlyInt8Linear
H A Dmixtral_moe_quantize.py120 __constants__ = ["in_features", "out_features"] variable in WeightOnlyInt8Linear
/aosp_15_r20/external/executorch/backends/vulkan/_passes/
H A Dint4_weight_only_quantizer.py18 __constants__ = ["in_features", "out_features"] variable in VkWeightOnlyInt4Linear
/aosp_15_r20/external/pytorch/torch/ao/nn/quantizable/modules/
H A Dactivation.py60 __constants__ = ["batch_first"] variable in MultiheadAttention

12