Searched refs:compute_shape_stack (Results 1 – 2 of 2) sorted by relevance
82 TORCH_API std::vector<torch::lazy::Shape> compute_shape_stack(at::TensorList tensors, int64_t dim);
1096 std::vector<Shape> compute_shape_stack(at::TensorList tensors, int64_t dim) { in compute_shape_stack() function