Searched refs:ReverseDimLookUp (Results 1 – 1 of 1) sorted by relevance
269 int64_t ReverseDimLookUp(absl::Span<const int64_t> permute_dims, int64_t id) { in ReverseDimLookUp() function in xla::__anon5e1d62920111::ConvolutionVisitor2079 auto dim = ReverseDimLookUp(permute_dims, i); in Propagate()2790 const int64_t old_dim = ReverseDimLookUp(permute_dims, i); in PropagateOnPad()2963 reversed_transpose_dims[i] = ReverseDimLookUp(prod_transpose_dims, i); in PropagateOnConstant()3077 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()3115 reversed_transpose_dims[i] = ReverseDimLookUp(retval.second, i); in PropagateOnBackpropFilterConv()