Function cinn::frontend::decomposer::GetReduceDimsForX

Defined in File broadcast.cc

Function Documentation

void cinn::frontend::decomposer::GetReduceDimsForX(const std::vector<int> &dx_shape, const std::vector<int> &dout_shape, std::vector<int> *reduce_dims)