Searched refs:compute_shape_native_dropout (Results 1 – 2 of 2) sorted by relevance
65 TORCH_API std::vector<torch::lazy::Shape> compute_shape_native_dropout(const at::Tensor & input, do…
677 std::vector<Shape> compute_shape_native_dropout( in compute_shape_native_dropout() function