Home
last modified time | relevance | path

Searched defs:expand_as (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DExpand.cpp70 Tensor expand_as(const at::Tensor& self, const at::Tensor& other) { in expand_as() function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py2997 def expand_as(a: Tensor, b: Tensor) -> Tensor: function
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py963 def expand_as(x, y): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1155 Tensor expand_as(const Tensor& self, const Tensor& other) { in expand_as() function
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py884 def expand_as(g: jit_utils.GraphContext, self, other): function
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp2494 TEST_F(VulkanAPITest, expand_as) { in TEST_F() argument