Searched defs:reverse_dims (Results 1 – 3 of 3) sorted by relevance
61 bool ConvIsLowerable(HloInstruction* reverse_dims) const { in ConvIsLowerable()
831 auto reverse_dims = reverse->dimensions(); in RewriteReverse() local
284 auto reverse_dims = op.input(1); in ReverseGrad() local