Searched refs:compute_shape_diagonal_scatter (Results 1 – 2 of 2) sorted by relevance
119 TORCH_API std::vector<torch::lazy::Shape> compute_shape_diagonal_scatter(const at::Tensor & self, c…
1293 std::vector<Shape> compute_shape_diagonal_scatter( in compute_shape_diagonal_scatter() function