Searched refs:clip_value (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/utils/ |
| H A D | clip_grad.h | 122 double clip_value) { in clip_grad_value_() argument
125 param.grad().data().clamp_(-clip_value, clip_value); in clip_grad_value_()
134 double clip_value) { in clip_grad_value_() argument
135 clip_grad_value_(std::vector<Tensor>(parameters), clip_value); in clip_grad_value_()
140 inline void clip_grad_value_(Tensor parameter, double clip_value) { in clip_grad_value_() argument
142 clip_grad_value_(std::move(params), clip_value); in clip_grad_value_()
|
| /aosp_15_r20/external/pytorch/torch/nn/utils/ |
| H A D | clip_grad.py | 151 clip_value: float,
171 clip_value = float(clip_value)
181 torch._foreach_clamp_min_(cast(List[Tensor], grads), -clip_value)
182 torch._foreach_clamp_max_(cast(List[Tensor], grads), clip_value)
189 cast(Tensor, grad).clamp_(min=-clip_value, max=clip_value)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/ |
| H A D | clip_ops_test.py | 54 clip_value = 4.4
55 ans = clip_ops.clip_by_value(x, -clip_value, clip_value)
167 clip_value = 4.0
168 ans = clip_ops.clip_by_value(x, -clip_value, clip_value)
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | nn_utils.cpp | 315 float clip_value = 2.5; in TEST_F() local
328 utils::clip_grad_value_(l->parameters(), clip_value); in TEST_F()
331 ASSERT_LE(p.grad().data().max().item().toFloat(), clip_value); in TEST_F()
332 ASSERT_GE(p.grad().data().min().item().toFloat(), -clip_value); in TEST_F()
343 utils::clip_grad_value_(p1, clip_value); in TEST_F()
344 utils::clip_grad_value_({p2}, clip_value); in TEST_F()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_nn.py | 12703 clip_value = 2.5
12710 clip_grad_value_(l.parameters(), clip_value, foreach=foreach)
12712 self.assertLessEqual(p.grad.data.max(), clip_value)
12713 self.assertGreaterEqual(p.grad.data.min(), -clip_value)
12720 clip_grad_value_(p1, clip_value, foreach=foreach)
12721 clip_grad_value_([p2], clip_value, foreach=foreach)
|