| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 332 #define BINARY_SCALAR_2(op, tensor_tensor, tensor_scalar) \ in TORCH_LIBRARY_IMPL() argument
334 UNARY_POINTWISE2(op, tensor_scalar); in TORCH_LIBRARY_IMPL()
337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
339 UNARY_POINTWISE2(op, tensor_scalar);\ in TORCH_LIBRARY_IMPL()
342 #define BINARY_SCALAR_3_Tensor(op, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
344 UNARY_POINTWISE2(op, tensor_scalar);\ in TORCH_LIBRARY_IMPL()
371 BINARY_SCALAR_3(bitwise_left_shift, Tensor, Tensor_Scalar, Scalar_Tensor); in TORCH_LIBRARY_IMPL()
372 POINTWISE_BOXED(bitwise_left_shift_.Tensor_Scalar); in TORCH_LIBRARY_IMPL()
374 BINARY_SCALAR_3(bitwise_right_shift, Tensor, Tensor_Scalar, Scalar_Tensor); in TORCH_LIBRARY_IMPL()
375 POINTWISE_BOXED(bitwise_right_shift_.Tensor_Scalar); in TORCH_LIBRARY_IMPL()
[all …]
|
| H A D | BatchRulesFactory.cpp | 238 VMAP_SUPPORT2(linspace, Tensor_Scalar, linspace_Tensor_Scalar_batch_rule); in TORCH_LIBRARY_IMPL()
241 VMAP_SUPPORT2(logspace, Tensor_Scalar, logspace_Tensor_Scalar_batch_rule); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesDecompositions.cpp | 131 OP_DECOMPOSE2(float_power, Tensor_Scalar); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 114 aten.bitwise_left_shift.Tensor_Scalar,
118 aten.bitwise_left_shift_.Tensor_Scalar,
131 aten.bitwise_right_shift.Tensor_Scalar,
135 aten.bitwise_right_shift_.Tensor_Scalar,
203 aten.float_power.Tensor_Scalar,
325 aten.pow.Tensor_Scalar,
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_aten_core_operators.expect | 96 aten::bitwise_left_shift.Tensor_Scalar
100 aten::bitwise_left_shift_.Tensor_Scalar
115 aten::bitwise_right_shift.Tensor_Scalar
119 aten::bitwise_right_shift_.Tensor_Scalar
400 aten::pow.Tensor_Scalar
|
| /aosp_15_r20/external/executorch/backends/xnnpack/test/ops/ |
| H A D | pow.py | 27 .check_count({"torch.ops.aten.pow.Tensor_Scalar": 1})
56 .check_count({"torch.ops.aten.pow.Tensor_Scalar": 1})
|
| H A D | square.py | 24 Note that torch.square maps to aten.pow.Tensor_Scalar. The pow visitor has logic
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | binary_ops.py | 47 "aten.pow.Tensor_Scalar",
74 exir_ops.edge.aten.pow.Tensor_Scalar: MPSPow,
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 282 (torch.int8, torch.ops.aten.linspace.Tensor_Scalar): (0, 1),
283 (torch.uint8, torch.ops.aten.linspace.Tensor_Scalar): (0, 1),
284 (torch.int16, torch.ops.aten.linspace.Tensor_Scalar): (0, 1),
285 (torch.int32, torch.ops.aten.linspace.Tensor_Scalar): (0, 1),
286 (torch.int64, torch.ops.aten.linspace.Tensor_Scalar): (0, 1),
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 237 - aten::float_power.Tensor_Scalar
473 - aten::pow.Tensor_Scalar
925 aten::pow.Tensor_Scalar: 3
|
| /aosp_15_r20/external/executorch/backends/xnnpack/operators/ |
| H A D | op_square.py | 24 target = "aten.pow.Tensor_Scalar"
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | op_pow.py | 21 target = ["aten.pow.Tensor_Scalar"]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | ts_native_functions.yaml | 102 - pow.Tensor_Scalar
|
| H A D | Pow.cpp | 23 TORCH_META_FUNC2(pow, Tensor_Scalar) (const Tensor& base, const Scalar& exp) { in TORCH_META_FUNC2() argument
|
| /aosp_15_r20/external/pytorch/torchgen/aoti/ |
| H A D | fallback_ops.py | 98 "aten.pow.Tensor_Scalar",
|
| /aosp_15_r20/external/executorch/backends/xnnpack/partition/ |
| H A D | configs.py | 52 exir_ops.edge.aten.pow.Tensor_Scalar,
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 191 aten.float_power.Tensor_Scalar,
|
| /aosp_15_r20/external/executorch/backends/mediatek/quantizer/ |
| H A D | annotator.py | 238 torch.ops.aten.pow.Tensor_Scalar,
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | layout_transform.py | 63 exir_ops.edge.aten.pow.Tensor_Scalar,
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/hf_train/ |
| H A D | GoogleFnet_training.txt | 66 Operator: aten.pow.Tensor_Scalar
|
| H A D | DistillGPT2_training.txt | 74 Operator: aten.pow.Tensor_Scalar
|
| H A D | GPTNeoForCausalLM_training.txt | 82 Operator: aten.pow.Tensor_Scalar
|
| H A D | GPTNeoForSequenceClassification_training.txt | 86 Operator: aten.pow.Tensor_Scalar
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_shape_registry_util.cpp | 24 {"aten::pow.Tensor_Scalar(Tensor self, Scalar exponent) -> Tensor", "unary"}, in get_tensorexpr_elementwise_set()
|
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/microbenchmarks/operator_inp_logs/torchbench_train/ |
| H A D | hf_GPT2_training.txt | 71 Operator: aten.pow.Tensor_Scalar
|