Home
last modified time | relevance | path

Searched full:add_ (Results 1 – 25 of 493) sorted by relevance

12345678910>>...20

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_remove_mutation.py29 y.add_(2)
35 FileCheck().check("aten::add_").run(graph)
53 y.add_(2)
59 FileCheck().check("aten::add_").run(graph)
64 x.add_(1)
65 x.add_(3)
72 FileCheck().check_not("aten::add_").run(graph)
77 x.add_(1)
79 x.add_(3)
84 FileCheck().check_count("aten::add_", 2).run(graph)
[all …]
/aosp_15_r20/external/pytorch/test/
H A Dtest_fx_reinplace_pass.py20 # into add_()
37 add = torch.ops.aten.add_.Tensor(clone, 1); add = None
65 add_1 = torch.ops.aten.add_.Tensor(view, 1); add_1 = None
128 d.add_(1)
154 add = torch.ops.aten.add_.Tensor(view_2, 1); add = None
163 add_1 = torch.ops.aten.add_.Tensor(view_5, view_8); view_8 = add_1 = None
173 c.add_(1)
193 add = torch.ops.aten.add_.Tensor(select_1, 1); select_1 = add = None
233 add = torch.ops.aten.add_.Tensor(select_1, 1); select_1 = add = None
332 add = torch.ops.aten.add_.Tensor(diagonal, 1); diagonal = add = None
[all …]
H A Dtest_functionalization.py163 x.add_(1)
177 x.add_(1)
178 self.assertRaises(RuntimeError, lambda: y.add_(1))
179 self.assertRaises(RuntimeError, lambda: z.add_(1))
279 y.add_(tmp)
424 z.add_(1)
459 add = torch.ops.aten.add_.Tensor(view, 1); add = None
498 x.add_(tmp)
569 y.add_(1)
672 y.add_(tmp)
[all …]
H A Dtest_schema_check.py147 actual.add_(y)
150 ('aten::add_', 'input'),
151 ('aten::add_', 'other')
157 ('aten::add_', 'input', 'output_0'),
158 ('aten::add_', 'other', 'output_0')
285 expected.add_(x)
287 actual.add_(y)
/aosp_15_r20/external/pytorch/torch/optim/
H A D_functional.py67 exp_avg.add_(make_sparse(exp_avg_update_values))
72 exp_avg_sq.add_(make_sparse(exp_avg_sq_update_values))
75 numer = exp_avg_update_values.add_(old_exp_avg_values)
76 exp_avg_sq_update_values.add_(old_exp_avg_sq_values)
77 denom = exp_avg_sq_update_values.sqrt_().add_(eps)
84 param.add_(make_sparse(-step_size * numer.div_(denom)))
/aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/upb/protos_generator/
Dgen_repeated_fields.cc61 using $0Access::add_$1; in WriteRepeatedFieldUsingAccessors()
76 using $0Access::add_$1; in WriteRepeatedFieldUsingAccessors()
107 absl::StatusOr<$0> add_$2(); in WriteRepeatedFieldsInMessageHeader()
117 bool add_$1($0 val); in WriteRepeatedFieldsInMessageHeader()
146 absl::StatusOr<$1> $0::add_$2() { in WriteRepeatedMessageAccessor()
200 bool $0::add_$2($1 val) { in WriteRepeatedStringAccessor()
245 bool $0::add_$2($1 val) { return $3_add_$4(msg_, val, arena_); } in WriteRepeatedScalarAccessor()
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py957 return x.add_(1.0) + torch.nn.functional.relu_(x)
973 add_: "f32[3, 4]" = l_x_.add_(1.0)
975 add: "f32[3, 4]" = add_ + relu_; add_ = relu_ = None
995 add_: "f32[3, 4]" = l_x_.add_(1.0)
997 add: "f32[3, 4]" = add_ + relu_; add_ = relu_ = None
1037 add_: "f32[3, 4]" = l_x_.add_(1.0)
1039 add: "f32[3, 4]" = add_ + relu_; add_ = relu_ = None
1063 return wrap(lambda x: x.add_(1.0), x)
1097 add_: "f32[3, 4]" = l_x_.add_(1.0); l_x_ = None
1098 return (add_,)
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A D_functions.cpp46 scale_first.add_(input_square.select(1, c)); in forward()
60 scale_current.add_(square_next, 1); in forward()
65 scale_current.add_(square_previous, -1); in forward()
73 .add_(ctx->saved_data["k"].toInt()); in forward()
129 accum_ratio.add_(padded_ratio[c + ctx->saved_data["size"].toInt() - 1]); in backward()
131 accum_ratio.add_(padded_ratio[c], -1); in backward()
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DPyTorchOperatorHacks.cpp69 return output.add_(*bias); in linear_hack()
98 auto log_weight = (pos_weight - 1).mul(target).add_(1); in binary_cross_entropy_with_logits_hack()
99 …nput).add(log_weight.mul(((-max_val).exp_().add((-input - max_val).exp_())).log_().add_(max_val))); in binary_cross_entropy_with_logits_hack()
101 …loss = (1 - target).mul(input).add_(max_val).add_((-max_val).exp_().add((-input -max_val).exp_()).… in binary_cross_entropy_with_logits_hack()
191 b = noise.add(-1).mul_(alpha * a).add_(alpha * a * p); in _dropout_impl()
200 return multiply<inplace>(input, noise).add_(b); in _dropout_impl()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dinference_mode.cpp465 b.add_(1), in TEST()
472 c.add_(1), in TEST()
490 b_el.add_(1), in TEST()
496 c_el.add_(1), in TEST()
561 // Testing both copy_ from VariableTypeManual and add_ from generated code. in TEST()
563 s.add_(t); in TEST()
564 t.add_(s); in TEST()
568 s.add_(t); in TEST()
574 t.add_(s), in TEST()
/aosp_15_r20/external/protobuf/src/google/protobuf/compiler/cpp/
H A Dstring_field.cc750 "$deprecated_attr$std::string* ${1$add_$name$$}$();\n" in GenerateAccessorDeclarations()
751 "$deprecated_attr$void ${1$add_$name$$}$(const std::string& value);\n" in GenerateAccessorDeclarations()
752 "$deprecated_attr$void ${1$add_$name$$}$(std::string&& value);\n" in GenerateAccessorDeclarations()
753 "$deprecated_attr$void ${1$add_$name$$}$(const char* value);\n", in GenerateAccessorDeclarations()
757 "$deprecated_attr$void ${1$add_$name$$}$(StringPiece value);\n", in GenerateAccessorDeclarations()
761 "$deprecated_attr$void ${1$add_$name$$}$(const $pointer_type$* " in GenerateAccessorDeclarations()
787 "inline std::string* $classname$::add_$name$() {\n" in GenerateInlineAccessorDefinitions()
857 "inline void $classname$::add_$name$(const std::string& value) {\n" in GenerateInlineAccessorDefinitions()
862 "inline void $classname$::add_$name$(std::string&& value) {\n" in GenerateInlineAccessorDefinitions()
867 "inline void $classname$::add_$name$(const char* value) {\n" in GenerateInlineAccessorDefinitions()
[all …]
/aosp_15_r20/external/cronet/third_party/protobuf/src/google/protobuf/compiler/cpp/
H A Dstring_field.cc746 "$deprecated_attr$std::string* ${1$add_$name$$}$();\n" in GenerateAccessorDeclarations()
747 "$deprecated_attr$void ${1$add_$name$$}$(const std::string& value);\n" in GenerateAccessorDeclarations()
748 "$deprecated_attr$void ${1$add_$name$$}$(std::string&& value);\n" in GenerateAccessorDeclarations()
749 "$deprecated_attr$void ${1$add_$name$$}$(const char* value);\n", in GenerateAccessorDeclarations()
753 "$deprecated_attr$void ${1$add_$name$$}$(StringPiece value);\n", in GenerateAccessorDeclarations()
757 "$deprecated_attr$void ${1$add_$name$$}$(const $pointer_type$* " in GenerateAccessorDeclarations()
783 "inline std::string* $classname$::add_$name$() {\n" in GenerateInlineAccessorDefinitions()
853 "inline void $classname$::add_$name$(const std::string& value) {\n" in GenerateInlineAccessorDefinitions()
858 "inline void $classname$::add_$name$(std::string&& value) {\n" in GenerateInlineAccessorDefinitions()
863 "inline void $classname$::add_$name$(const char* value) {\n" in GenerateInlineAccessorDefinitions()
[all …]
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A D_functions.py247 scale_first.add_(input_square.select(1, c))
257 scale_current.add_(square_next, alpha=1)
261 scale_current.add_(square_previous, alpha=-1)
263 ctx.scale.mul_(ctx.alpha / ctx.size).add_(ctx.k)
302 accum_ratio.add_(paddded_ratio[c + ctx.size - 1])
306 accum_ratio.add_(paddded_ratio[c], alpha=-1)
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler_tree.py855 aten::add_
1030 aten::add_
1041 aten::add_
1087 <built-in method add_ of Tensor object at 0xXXXXXXXXXXXX>
1088 aten::add_
1091 <built-in method add_ of Tensor object at 0xXXXXXXXXXXXX>
1092 aten::add_
1101 <built-in method add_ of Tensor object at 0xXXXXXXXXXXXX>
1102 aten::add_
1105 <built-in method add_ of Tensor object at 0xXXXXXXXXXXXX>
[all …]
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_auto_functionalize.py157 x.add_(y[0] + w)
158 z.add_(y[1] + n)
209 x.add_(y[0] + w)
210 z.add_(y[1] + n)
303 x.add_(y[0] + w)
305 z.add_(y[1] + n)
374 x.add_(y[0] + w)
375 z.add_(y[1] + n)
475 x.add_(y[0] + w)
476 z.add_(y[1] + n)
[all …]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dhop_db.py23 return [x[0].cos().add_(1.) * y0, (x[1] + y1.sin()).cos_().view(x[1].size())]
76 x.add_(5)
77 z.add_(5)
83 x.add_(5)
84 z.add_(5)
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_dce.cpp37 # CHECK: add_ in TEST()
38 %tot.3 : Tensor = aten::add_(%tot.6, %35, %12) in TEST()
41 # CHECK: add_ in TEST()
42 %46 : Tensor = aten::add_(%44, %12, %12) in TEST()
/aosp_15_r20/external/pytorch/test/fx/
H A Dtest_common_passes.py41 y.add_(1)
46 x.add_(1)
53 y.add_(1)
59 y.add_(1)
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py125 y = x.clone().add_(1).add_(-1)
129 y = x.clone().add_(1).add_(-1)
1708 add_ = torch.ops.aten.add_.Tensor(clone, 1); clone = None
1709 add__1 = torch.ops.aten.add_.Tensor(add_, -1); add_ = None
1720 add_ = torch.ops.aten.add_.Tensor(clone, 1); clone = None
1721 add__1 = torch.ops.aten.add_.Tensor(add_, -1); add_ = None
2033 y.add_(4)
2152 y.add_(4)
2211 view_x.add_(1)
2255 view_x.add_(1)
[all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp47 // add_(Tensor(a!) self, Tensor other, *, Scalar alpha=1) -> Tensor(a!)
131 // vmap(Tensor.add_, in_dims=(None, 0))(torch.ones(3), torch.ones(B0, 3)) in batchedTensorInplaceForLoopFallback()
134 // There's no way to do self.add_(other) because `other` has more elements in batchedTensorInplaceForLoopFallback()
143 // The following prints out "vmap: aten::add_(tensor, ...) is not possible", in batchedTensorInplaceForLoopFallback()
144 // but it would be better to print out "tensor.add_(...) is not possible". in batchedTensorInplaceForLoopFallback()
145 // Afaict there's no official way to get the add_ and there is no way to in batchedTensorInplaceForLoopFallback()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/
H A Dadagrad.cpp111 state.sum(state.sum().add_(make_sparse(grad_values.pow(2)))); in step()
113 const auto std_values = std._values().sqrt_().add_(options.eps()); in step()
115 p.add_(make_sparse(grad_values / std_values), -clr); in step()
118 const auto std = state.sum().sqrt().add_(options.eps()); in step()
H A Drmsprop.cpp121 grad_avg.mul_(alpha).add_(grad, 1 - alpha); in step()
124 .add_(options.eps()); in step()
126 avg = square_avg.sqrt().add_(options.eps()); in step()
133 p.add_(buf, -options.lr()); in step()
/aosp_15_r20/external/executorch/backends/mediatek/quantizer/
H A Dannotator.py125 torch.ops.aten.add_.Scalar,
126 torch.ops.aten.add_.Tensor,
219 torch.ops.aten.add_.Scalar,
220 torch.ops.aten.add_.Tensor,
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dremove_mutation.cpp125 // x = add_(1) in tryMakeUnaliasedIfOutputAndMutationAtomic()
128 // then removing mutation add_ does not change observable semantics in tryMakeUnaliasedIfOutputAndMutationAtomic()
282 // x.add_(1) in RemoveTensorMutation()
283 // x.add_(2) in RemoveTensorMutation()
287 // x0.add_(2) in RemoveTensorMutation()
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_verifier.py53 node.target = torch.ops.aten.add_.Tensor
95 node.target = torch.ops.aten.add_.Tensor
178 self.my_buffer2.add_(1.0)
201 self.my_buffer2.add_(1.0)

12345678910>>...20