Searched refs:target_axes (Results 1 – 1 of 1) sorted by relevance
2016 std::set<int> target_axes; in RemoveReverse() local2020 target_axes.insert((axis.vec<int64>()(j) + shape.dim_size()) % in RemoveReverse()2023 target_axes.insert((axis.vec<int>()(j) + shape.dim_size()) % in RemoveReverse()2035 target_axes.find(j) == target_axes.end(); in RemoveReverse()