Function cinn::frontend::decomposer::GetReduceDimsForY
Defined in File broadcast.cc
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)