| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | PointwiseOpsKernel.cu | 15 CONSTEXPR_EXCEPT_WIN_CUDA char addcmul_name[] = "addcmul";
20 // When using Jiterator, addcmul and addcdiv kernels get stuck during a in addcmul_cuda_kernel()
27 template <typename T> T addcmul(T a, T b, T c, T alpha) { return a + alpha * (b * c); }); in addcmul_cuda_kernel()
67 // When using Jiterator, addcmul and addcdiv kernels get stuck during a in addcdiv_cuda_kernel()
|
| H A D | ForeachPointwiseOp.cu | 265 FOREACH_POINTWISE_OP_SCALAR(addcmul, std::multiplies);
267 FOREACH_POINTWISE_OP_SCALARLIST(addcmul, std::multiplies);
270 FOREACH_POINTWISE_OP_TENSOR(addcmul, std::multiplies);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | broadcast_test.cpp | 65 ASSERT_TRUE(aTensorScalar.addcmul(b, c).equal( in TestOut3WithScalar()
67 .addcmul(b.expand(expanded_sizes), c.expand(expanded_sizes)))); in TestOut3WithScalar()
76 ASSERT_ANY_THROW(a.addcmul(b, c)); in TestOut3OldFallback()
85 ASSERT_ANY_THROW(a.addcmul(b, c)); in TestOut3MismatchedSizes()
|
| /aosp_15_r20/external/pytorch/tools/autograd/ |
| H A D | deprecated.yaml | 40 - name: addcmul(Tensor self, Scalar value, Tensor tensor1, Tensor tensor2) -> Tensor
41 aten: addcmul(self, tensor1, tensor2, value)
46 - name: addcmul(Tensor self, Scalar value, Tensor tensor1, Tensor tensor2, *, Tensor(a!) out) -> Te…
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ForeachOpsKernels.cpp | 404 FOREACH_POINTWISE_OP_SCALAR(addcmul);
407 FOREACH_POINTWISE_OP_SCALARLIST(addcmul);
410 FOREACH_POINTWISE_OP_TENSOR(addcmul);
|
| H A D | PointwiseOps.cpp | 17 TORCH_META_FUNC(addcmul) in TORCH_META_FUNC() argument
|
| H A D | ts_native_functions.yaml | 13 - addcmul
|
| H A D | group_norm.cpp | 245 .addcmul(out, weight.view(affine_param_shape), 1); in math_group_norm()
|
| H A D | layer_norm.cpp | 246 out = bias.addcmul(out, weight, 1); in math_native_layer_norm()
|
| /aosp_15_r20/external/pytorch/test/lazy/ |
| H A D | test_extract_compiled_graph.py | 33 …addcmul function takes a at::Scalar which results in a special TSData containing a Scalar rather t…
37 return torch.addcmul(a, b, c, value=5)
|
| /aosp_15_r20/external/pytorch/functorch/dim/ |
| H A D | op_properties.py | 75 torch.Tensor.addcmul,
76 torch.addcmul,
|
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_builtins.py | 269 return x.addcmul(tensor, tensor, value=tensor)
272 return x.addcmul(tensor, tensor, value=tensor)
|
| /aosp_15_r20/external/pytorch/test/typing/pass/ |
| H A D | math_ops.py | 31 # addcmul
32 torch.addcmul(torch.randn(1, 3), torch.randn(3, 1), torch.randn(1, 3), value=0.1)
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 85 aten.addcmul.default,
86 aten.addcmul.out,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesViews.cpp | 72 // func: addcmul(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1)
84 // m.impl("addcmul", addcmul_decomp);
|
| /aosp_15_r20/external/pytorch/functorch/op_analysis/ |
| H A D | public_api | 621 addcmul
|
| H A D | annotated_ops | 392 addcmul, composite pointwise
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 68 - aten::addcmul
744 aten::addcmul: 2
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | rmsprop.cpp | 122 avg = square_avg.addcmul(grad_avg, grad_avg, -1) in step()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | NamedRegistrations.cpp | 44 m.impl("addcmul", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
45 m.impl("addcmul.out", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit_autocast.py | 167 f = torch.addcmul(e, c, d, value=0.1)
178 return torch.addcmul(a, a, b, value=0.1)
|
| H A D | test_jit_fuser_te.py | 777 return t.addcmul(t + 1, t2, value=0.1)
783 FileCheck().check("aten::add(").check("aten::addcmul(").run(
1898 torch.addcmul,
2717 "addcmul",
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/timm_train/ |
| H A D | resmlp_12_224_training.txt | 14 Operator: aten.addcmul.default
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_shape_registry_util.cpp | 118 …{"aten::addcmul(Tensor self, Tensor tensor1, Tensor tensor2, *, Scalar value=1) -> Tensor", "broad… in get_tensorexpr_elementwise_set()
|
| /aosp_15_r20/external/pytorch/docs/source/ |
| H A D | torch.rst | 321 addcmul
|