Home
last modified time | relevance | path

Searched full:weight3 (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/grpc-grpc-java/xds/src/test/java/io/grpc/xds/
H A DWeightedRoundRobinLoadBalancerTest.java378 double weight3 = 100 / (0.86 + 3 / 100F * weightedConfig.errorUtilizationPenalty); in pickByWeight_largeWeight_withEps_defaultErrorUtilizationPenalty() local
379 double totalWeight = weight1 + weight2 + weight3; in pickByWeight_largeWeight_withEps_defaultErrorUtilizationPenalty()
382 weight3 / totalWeight); in pickByWeight_largeWeight_withEps_defaultErrorUtilizationPenalty()
423 double weight3 = 100 / (0.86 + 3.14159 / 100F * weightedConfig.errorUtilizationPenalty); in pickByWeight_normalWeight_withEps_defaultErrorUtilizationPenalty() local
424 double totalWeight = weight1 + weight2 + weight3; in pickByWeight_normalWeight_withEps_defaultErrorUtilizationPenalty()
427 weight3 / totalWeight); in pickByWeight_normalWeight_withEps_defaultErrorUtilizationPenalty()
443 double weight3 = 100 / (0.86 + 3.14159 / 100F * weightedConfig.errorUtilizationPenalty); in pickByWeight_normalWeight_withEps_customErrorUtilizationPenalty() local
444 double totalWeight = weight1 + weight2 + weight3; in pickByWeight_normalWeight_withEps_customErrorUtilizationPenalty()
447 weight3 / totalWeight); in pickByWeight_normalWeight_withEps_customErrorUtilizationPenalty()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dinit_baseline.py60 weight3 = layer3.weight.data.numpy()
62 return [weight1, weight2, weight3]
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DRNN.cpp52 const Tensor& weight3, in mkldnn_rnn_layer_backward() argument
299 const Tensor& weight3, in mkldnn_rnn_layer_backward() argument
341 ? _shuffle_bias(weight2, weight3, rnn.mode) in mkldnn_rnn_layer_backward()
H A DMKLDNNConversions.cpp391 const at::Tensor& weight3, in get_lstm_packed_weights() argument
/aosp_15_r20/external/mesa3d/src/panfrost/lib/genxml/
H A Ddecode.c62 nonzero_weights |= w.weight3 != 0x0; in pandecode_midgard_tiler_descriptor()
H A Dv4.xml746 <field name="Weight3" size="32" start="3:0" type="uint"/>
H A Dv5.xml772 <field name="Weight3" size="32" start="3:0" type="uint"/>
H A Dv6.xml1100 <field name="Weight3" size="16" start="3:16" type="uint"/>
H A Dv9.xml1196 <field name="Weight3" size="16" start="3:16" type="uint"/>
H A Dv7.xml1188 <field name="Weight3" size="16" start="3:16" type="uint"/>
H A Dv10.xml1667 <field name="Weight3" size="16" start="3:16" type="uint"/>
/aosp_15_r20/external/pytorch/tools/autograd/
H A Dderivatives.yaml2751 …yer(Tensor input, Tensor weight0, Tensor weight1, Tensor weight2, Tensor weight3, Tensor hx_, Tens…
2753weight3, hx_, cx_: "GradMode::is_enabled() ? mkldnn_rnn_layer_differentiable_backward(input, weigh…
2755 - name: mkldnn_rnn_layer_backward(Tensor input, Tensor weight1, Tensor weight2, Tensor weight3, Ten…
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py406 self.weight3 = nn.Parameter(torch.randn(512, 512))
417 z = (x + y2 @ self.weight3) * a.sum()
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dmkldnn_fusion.py1014 Arg(), # weight3
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A DFunctionsManual.h1093 const Tensor& weight3,
H A DFunctionsManual.cpp7029 const Tensor& weight3, in mkldnn_rnn_layer_differentiable_backward() argument
7073 bias_hh = weight3; in mkldnn_rnn_layer_differentiable_backward()
/aosp_15_r20/external/pytorch/torch/
H A D_meta_registrations.py6163 weight3, argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml4037 …yer(Tensor input, Tensor weight0, Tensor weight1, Tensor weight2, Tensor weight3, Tensor hx_, Tens…
4043 - func: mkldnn_rnn_layer_backward(Tensor input, Tensor weight1, Tensor weight2, Tensor weight3, Ten…