/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/aten/src/ATen/functorch/ |
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 | BatchRulesBinaryOps.cpp | 316 BINARY_RANDOM_POINTWISE2(normal, Tensor_Tensor); in TORCH_LIBRARY_IMPL() 423 BINARY_SCALAR_3(pow, Tensor_Tensor, Tensor_Scalar, Scalar); in TORCH_LIBRARY_IMPL()
|
H A D | BatchRulesDecompositions.cpp | 130 OP_DECOMPOSE2(float_power, Tensor_Tensor); in TORCH_LIBRARY_IMPL()
|
/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/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/executorch/backends/apple/mps/operators/ |
H A D | binary_ops.py | 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
|
H A D | TensorCompare.cpp | 207 TORCH_META_FUNC2(isin, Tensor_Tensor) ( in TORCH_META_FUNC2() argument
|
/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/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/mediatek/quantizer/ |
H A D | annotator.py | 239 torch.ops.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/vulkan/ |
H A D | op_registry.py | 251 exir_ops.edge.aten.pow.Tensor_Tensor,
|
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
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/_inductor/ |
H A D | lowering.py | 5321 aten.pow.Tensor_Tensor, add_to_fallback_set=False
|