Function cinn::frontend::decomposer::GetReduceDimsForY

Function Documentation

void cinn::frontend::decomposer::GetReduceDimsForY(const std::vector<int> &dy_shape, const std::vector<int> &dout_shape, int axis, std::vector<int> *reduce_dims)