Home
last modified time | relevance | path

Searched defs:block_diag (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/linalg/
H A Dlinear_operator_block_diag_test.py26 from tensorflow.python.ops.linalg import linear_operator_block_diag as block_diag unknown
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_native_functions.cpp402 at::Tensor LazyNativeFunctions::block_diag(at::TensorList tensors) { in block_diag() function in torch::lazy::LazyNativeFunctions
/aosp_15_r20/external/pytorch/torch/
H A Dfunctional.py1395 def block_diag(*tensors): function
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py4435 def block_diag(*tensors: List[TensorLikeType]) -> TensorLikeType: function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp862 Tensor block_diag(TensorList tensors) { in block_diag() function