| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesBinaryOps.cpp | 316 BINARY_RANDOM_POINTWISE2(normal, Tensor_Tensor); in TORCH_LIBRARY_IMPL()
332 #define BINARY_SCALAR_2(op, tensor_tensor, tensor_scalar) \ in TORCH_LIBRARY_IMPL() argument
333 BINARY_POINTWISE2(op, tensor_tensor);\ in TORCH_LIBRARY_IMPL()
337 #define BINARY_SCALAR_3(op, tensor_tensor, tensor_scalar, scalar_tensor) \ in TORCH_LIBRARY_IMPL() argument
338 BINARY_POINTWISE2(op, tensor_tensor);\ in TORCH_LIBRARY_IMPL()
423 BINARY_SCALAR_3(pow, Tensor_Tensor, Tensor_Scalar, Scalar); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesFactory.cpp | 237 VMAP_SUPPORT2(linspace, Tensor_Tensor, linspace_Tensor_Tensor_batch_rule); in TORCH_LIBRARY_IMPL()
240 VMAP_SUPPORT2(logspace, Tensor_Tensor, logspace_Tensor_Tensor_batch_rule); in TORCH_LIBRARY_IMPL()
|
| H A D | BatchRulesDecompositions.cpp | 130 OP_DECOMPOSE2(float_power, Tensor_Tensor); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_block_io.cpp | 251 TensorMap<Tensor<float, 3, Layout> > tensor_tensor(tensor_data, tensor_dims); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze() local
257 float tensor_value = tensor_tensor(d0, d1, d2); in test_block_io_copy_using_reordered_dimensions_do_not_squeeze()
299 TensorMap<Tensor<float, 4, Layout> > tensor_tensor(tensor_data, tensor_dims); in test_block_io_copy_using_reordered_dimensions_squeeze() local
306 float tensor_value = tensor_tensor(d0, d1, d2, d3); in test_block_io_copy_using_reordered_dimensions_squeeze()
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | binary_ops.py | 46 "aten.pow.Tensor_Tensor",
73 exir_ops.edge.aten.pow.Tensor_Tensor: MPSPow,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Pow.cpp | 19 TORCH_META_FUNC2(pow, Tensor_Tensor) (const Tensor& base, const Tensor& exp) { in TORCH_META_FUNC2() argument
35 // but otherwise redispatches to the Tensor_Tensor overload. in TORCH_META_FUNC2()
|
| H A D | ts_native_functions.yaml | 103 - pow.Tensor_Tensor
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_decomp.py | 277 (torch.int8, torch.ops.aten.linspace.Tensor_Tensor): (0, 1),
278 (torch.uint8, torch.ops.aten.linspace.Tensor_Tensor): (0, 1),
279 (torch.int16, torch.ops.aten.linspace.Tensor_Tensor): (0, 1),
280 (torch.int32, torch.ops.aten.linspace.Tensor_Tensor): (0, 1),
281 (torch.int64, torch.ops.aten.linspace.Tensor_Tensor): (0, 1),
|
| /aosp_15_r20/external/pytorch/test/expect/ |
| H A D | HasDecompTest.test_aten_core_operators.expect | 385 aten::normal.Tensor_Tensor
402 aten::pow.Tensor_Tensor
|
| /aosp_15_r20/external/pytorch/test/mobile/model_test/ |
| H A D | coverage.yaml | 238 - aten::float_power.Tensor_Tensor
314 - aten::isin.Tensor_Tensor
474 - aten::pow.Tensor_Tensor
|
| H A D | model_ops.yaml | 282 aten::pow.Tensor_Tensor: 1
|
| /aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/ |
| H A D | _pointwise_ops.py | 205 aten.float_power.Tensor_Tensor,
327 aten.pow.Tensor_Tensor,
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | BinaryOp.cpp | 128 VK_REGISTER_OP(aten.pow.Tensor_Tensor, pow);
|
| /aosp_15_r20/external/pytorch/torchgen/aoti/ |
| H A D | fallback_ops.py | 99 "aten.pow.Tensor_Tensor",
|
| /aosp_15_r20/external/executorch/backends/vulkan/partitioner/ |
| H A D | supported_ops.py | 57 exir_ops.edge.aten.pow.Tensor_Tensor,
|
| /aosp_15_r20/external/pytorch/test/export/ |
| H A D | testing.py | 190 aten.float_power.Tensor_Tensor,
|
| /aosp_15_r20/external/executorch/backends/mediatek/quantizer/ |
| H A D | annotator.py | 239 torch.ops.aten.pow.Tensor_Tensor,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | VmapModeRegistrations.cpp | 62 …m.impl("normal.Tensor_Tensor", unsupportedRandomOp<const Tensor&, const Tensor&, std::optional<Gen… in TORCH_LIBRARY_IMPL()
|
| H A D | autocast_mode.cpp | 256 KERNEL_MPS2(pow, Tensor_Tensor, fp32) in TORCH_LIBRARY_IMPL()
|
| H A D | autocast_mode.h | 826 _(pow, Tensor_Tensor) \
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | symbolic_shape_registry_util.cpp | 43 // "aten::pow.Tensor_Tensor(Tensor self, Tensor exponent) -> Tensor", in get_tensorexpr_elementwise_set()
|
| /aosp_15_r20/external/executorch/backends/vulkan/ |
| H A D | op_registry.py | 251 exir_ops.edge.aten.pow.Tensor_Tensor,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | BinaryOp.cpp | 616 m.impl(TORCH_SELECTIVE_NAME("aten::pow.Tensor_Tensor"), TORCH_FN(pow)); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | cpu_rng_test.cpp | 145 m.impl("normal.Tensor_Tensor", normal_Tensor_Tensor); in TORCH_LIBRARY_IMPL()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | NamedRegistrations.cpp | 372 m.impl("pow.Tensor_Tensor", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
|