| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | BinaryOp.cpp | 16 const Tensor& self_arg, in binary_op_scalar()
103 Tensor& self_arg, in binary_op_scalar_()
153 const Tensor& self_arg, in binary_op_tensor()
226 const Tensor& self_arg, in quantized_binary_op_tensor()
314 Tensor& self_arg, in binary_op_tensor_()
388 const Tensor& self_arg, in add_scalar()
404 const Tensor& self_arg, in quantized_add()
413 const Tensor& self_arg, in quantized_sub()
422 const Tensor& self_arg, in quantized_mul()
431 const Tensor& self_arg, in quantized_div()
[all …]
|
| H A D | UnaryOp.cpp | 15 const Tensor& self_arg, in unary_op()
62 Tensor& unary_op_(Tensor& self_arg, const api::ShaderInfo& shader_descriptor) { in unary_op_()
104 Tensor exp(const Tensor& self_arg) { in exp()
108 Tensor& exp_(Tensor& self_arg) { in exp_()
112 Tensor sqrt(const Tensor& self_arg) { in sqrt()
116 Tensor& sqrt_(Tensor& self_arg) { in sqrt_()
120 Tensor log(const Tensor& self_arg) { in log()
124 Tensor& log_(Tensor& self_arg) { in log_()
|
| H A D | Clamp.cpp | 13 const Tensor& self_arg, in _clamp()
98 const Tensor& self_arg, in clamp()
105 Tensor& self_arg, in _clamp_()
195 const Tensor& self_arg, in activation()
243 Tensor& self_arg, in activation_()
333 const Tensor& self_arg, in activation_scalar()
424 Tensor& self_arg, in activation_scalar_()
544 Tensor hardshrink(const Tensor& self_arg, const Scalar& lambd) { in hardshrink()
558 Tensor leaky_relu(const Tensor& self_arg, const Scalar& negative_slope) { in leaky_relu()
|
| H A D | Pool.cpp | 14 const at::Tensor& self_arg, in adaptive_avg_pool2d()
88 const Tensor& self_arg, in pool2d()
229 const Tensor& self_arg, in avg_pool2d()
247 const Tensor& self_arg, in max_pool2d()
|
| H A D | Shape.cpp | 11 static Tensor view_internal(const Tensor& self_arg, const IntArrayRef shape) { in view_internal()
51 inline Tensor view(const Tensor& self_arg, IntArrayRef shape) { in view()
56 const Tensor& self_arg, in _reshape_alias()
|
| H A D | Padding.cpp | 14 const Tensor& self_arg, in pad2d()
97 Tensor reflection_pad2d(const Tensor& self_arg, IntArrayRef padding) { in reflection_pad2d()
101 Tensor replication_pad2d(const Tensor& self_arg, IntArrayRef padding) { in replication_pad2d()
|
| H A D | MaskedFill.cpp | 15 const Tensor& self_arg, in masked_fill_scalar()
124 const Tensor& self_arg, in masked_fill_tensor()
|
| H A D | Lerp.cpp | 110 Tensor& self_arg, in _lerp_scalar_()
239 Tensor& self_arg, in _lerp_tensor_()
|
| H A D | Var.cpp | 14 const at::Tensor& self_arg, in var_dim_IntList()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | MaxUnpooling.cu | 131 TensorArg output_arg{output, "output", 1}, self_arg{self_, "self_", 2}, in max_unpooling2d_forward_out_cuda() local
309 TensorArg output_arg{output, "output", 1}, self_arg{self_, "self_", 2}, in max_unpooling3d_forward_out_cuda() local
413 self_arg{self_, "self_", 3}, indices_arg{indices_, "indices_", 4}; in max_unpooling2d_backward_out_cuda() local
521 TensorArg self_arg{self_, "self_", 1}, indices_arg{indices_, "indices_", 2}, in max_unpooling3d_backward_out_cuda() local
|
| H A D | RreluWithNoise.cu | 154 TensorArg self_arg{self, "self", 1}, noise_arg{noise, "noise", 2}, in rrelu_with_noise_out_cuda() local
|
| H A D | Embedding.cu | 349 auto self_arg = TensorArg(self, "self", 1); in embedding_renorm_cuda_() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Embedding.cpp | 183 auto self_arg = TensorArg(self, "self", 1); in embedding_renorm_cpu_() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_shape_registry.cpp | 93 Argument self_arg = schema->arguments()[0]; in getInplaceVariant() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_observers.cpp | 1011 auto self_arg = c10::Argument("self", module.type()); in insertObserverResetMinMax() local
|