Searched refs:ReverseDimLookUp (Results 1 – 1 of 1) sorted by relevance
203 int64 ReverseDimLookUp(absl::Span<const int64> permute_dims, int64 id) { in ReverseDimLookUp() function in xla::__anon6418faad0111::ConvolutionVisitor1442 auto dim = ReverseDimLookUp(permute_dims, i); in Propagate()2120 reversed_transpose_dims[i] = ReverseDimLookUp(prod_transpose_dims, i); in PropagateOnConstant()2212 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()2257 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()