Searched refs:diagonal_scatter (Results 1 – 18 of 18) sorted by relevance
| /external/pytorch/test/ |
| D | test_fx_reinplace_pass.py | 315 z = torch.diagonal_scatter(x, y_updated)
|
| /external/pytorch/test/expect/ |
| D | HasDecompTest.test_aten_core_operators.expect | 179 aten::diagonal_scatter
180 aten::diagonal_scatter.out
|
| /external/pytorch/test/mobile/model_test/ |
| D | tensor_ops.py | 191 torch.diagonal_scatter(y, torch.ones(4)),
|
| /external/pytorch/torch/fx/passes/ |
| D | reinplace.py | 169 torch.ops.aten.diagonal_scatter.default: torch.ops.aten.diagonal.default,
|
| /external/pytorch/functorch/op_analysis/ |
| D | public_api | 544 diagonal_scatter
|
| /external/pytorch/torch/_inductor/fx_passes/ |
| D | reinplace.py | 37 torch.ops.aten.diagonal_scatter.default: torch.ops.aten.diagonal.default,
|
| /external/pytorch/aten/src/ATen/ |
| D | FunctionalInverses.cpp | 166 return base.diagonal_scatter(mutated_view, offset, dim1, dim2); in diagonal_inverse()
|
| /external/pytorch/docs/source/ |
| D | torch.rst | 120 diagonal_scatter
|
| D | tensors.rst | 402 Tensor.diagonal_scatter
|
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesScatterOps.cpp | 922 return std::make_tuple(at::diagonal_scatter(self_, src_, offset, dim1, dim2), 0); in diagonal_scatter_batch_rule()
1246 VMAP_SUPPORT(diagonal_scatter, diagonal_scatter_batch_rule); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/ |
| D | _torch_docs.py | 3631 torch.diagonal_scatter,
|
| D | overrides.py | 570 torch.diagonal_scatter: lambda input, src, offset=0, dim1=0, dim2=1: -1,
1415 Tensor.diagonal_scatter: lambda self, src, offset=0, dim1=0, dim2=1: -1,
|
| /external/pytorch/torch/csrc/lazy/core/ |
| D | shape_inference.cpp | 1313 auto out_meta = at::compositeexplicitautogradnonfunctional::diagonal_scatter( in compute_shape_diagonal_scatter()
|
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 4277 @register_decomposition(aten.diagonal_scatter)
4279 def diagonal_scatter( function
|
| /external/pytorch/torch/_inductor/ |
| D | lowering.py | 1585 @register_lowering(aten.diagonal_scatter, type_promotion_kind=None)
1586 def diagonal_scatter(input, src, offset: int = 0, dim1: int = 0, dim2: int = 1): function
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorShape.cpp | 4017 at::Tensor diagonal_scatter(const at::Tensor& self, const at::Tensor& src, int64_t offset, int64_t … in diagonal_scatter() function
|
| /external/pytorch/torch/_decomp/ |
| D | decompositions.py | 876 return torch.diagonal_scatter(grad_input, grad_output, offset, dim1, dim2)
|
| /external/pytorch/torch/csrc/autograd/ |
| D | FunctionsManual.cpp | 3792 ret = ret.diagonal_scatter(gL, 0, -2, -1); in linalg_eig_backward()
|