Home
last modified time | relevance | path

Searched refs:Tensor_Tensor (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_decomp.py277 (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 DBatchRulesFactory.cpp237 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 DBatchRulesBinaryOps.cpp316 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 DBatchRulesDecompositions.cpp130 OP_DECOMPOSE2(float_power, Tensor_Tensor); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DBinaryOp.cpp128 VK_REGISTER_OP(aten.pow.Tensor_Tensor, pow);
/aosp_15_r20/external/pytorch/test/expect/
H A DHasDecompTest.test_aten_core_operators.expect385 aten::normal.Tensor_Tensor
402 aten::pow.Tensor_Tensor
/aosp_15_r20/external/executorch/backends/apple/mps/operators/
H A Dbinary_ops.py73 exir_ops.edge.aten.pow.Tensor_Tensor: MPSPow,
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DPow.cpp19 TORCH_META_FUNC2(pow, Tensor_Tensor) (const Tensor& base, const Tensor& exp) { in TORCH_META_FUNC2() argument
H A DTensorCompare.cpp207 TORCH_META_FUNC2(isin, Tensor_Tensor) ( in TORCH_META_FUNC2() argument
/aosp_15_r20/external/executorch/backends/vulkan/partitioner/
H A Dsupported_ops.py57 exir_ops.edge.aten.pow.Tensor_Tensor,
/aosp_15_r20/external/pytorch/torch/distributed/tensor/_ops/
H A D_pointwise_ops.py205 aten.float_power.Tensor_Tensor,
327 aten.pow.Tensor_Tensor,
/aosp_15_r20/external/executorch/backends/mediatek/quantizer/
H A Dannotator.py239 torch.ops.aten.pow.Tensor_Tensor,
/aosp_15_r20/external/pytorch/test/export/
H A Dtesting.py190 aten.float_power.Tensor_Tensor,
/aosp_15_r20/external/executorch/backends/vulkan/
H A Dop_registry.py251 exir_ops.edge.aten.pow.Tensor_Tensor,
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A Dautocast_mode.cpp256 KERNEL_MPS2(pow, Tensor_Tensor, fp32) in TORCH_LIBRARY_IMPL()
H A Dautocast_mode.h826 _(pow, Tensor_Tensor) \
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py5321 aten.pow.Tensor_Tensor, add_to_fallback_set=False