Searched refs:compute_shape_relu (Results 1 – 2 of 2) sorted by relevance
77 TORCH_API std::vector<torch::lazy::Shape> compute_shape_relu(const at::Tensor & self);
714 std::vector<Shape> compute_shape_relu(const at::Tensor& self) { in compute_shape_relu() function