Home
last modified time | relevance | path

Searched defs:reverse_dims (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.h61 bool ConvIsLowerable(HloInstruction* reverse_dims) const { in ConvIsLowerable()
H A Ddynamic_padder.cc831 auto reverse_dims = reverse->dimensions(); in RewriteReverse() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Darray_grad.cc284 auto reverse_dims = op.input(1); in ReverseGrad() local