Searched defs:original_indices (Results 1 – 2 of 2) sorted by relevance
81 auto original_indices = Reshape( in DynamicPartitionGrad() local
420 std::vector<int64> original_indices(indices.begin(), indices.end()); in ReverseSharding() local