Home
last modified time | relevance | path

Searched defs:diag_embed (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp495 at::Tensor LazyNativeFunctions::diag_embed( in diag_embed() function in torch::lazy::LazyNativeFunctions
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py4341 def diag_embed( function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp1122 Tensor diag_embed(const Tensor& self, int64_t offset, int64_t dim1_, int64_t dim2_) { in diag_embed() function