Searched defs:reverse_dims (Results 1 – 2 of 2) sorted by relevance
284 auto reverse_dims = op.input(1); in ReverseGrad() local
775 auto reverse_dims = reverse->dimensions(); in RewriteReverse() local