Home
last modified time | relevance | path

Searched refs:pow_tensor_scalar (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DBinaryOp.cpp548 static Tensor pow_tensor_scalar(const Tensor& self, const Scalar& other) { in pow_tensor_scalar() function
550 self, other, std::optional<Scalar>(), VK_KERNEL(pow_tensor_scalar)); in pow_tensor_scalar()
620 TORCH_FN(pow_tensor_scalar)); in TORCH_LIBRARY_IMPL()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp3934 TEST_F(VulkanAPITest, pow_tensor_scalar) { in TEST_F() argument