Implement flatten one level with keys in C++ and use it for the prefix/equality error printing. #55953

	# Copyright 2023 The TensorFlow Authors. All Rights Reserved.
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	# See the License for the specific language governing permissions and
	# limitations under the License.
	# ============================================================================
	name: Buildifier
	permissions:
	contents: read
	on:
	pull_request:
	push:
	branches:
	- main

	env:
	# Have `go install` place binaries in $PATH
	GOBIN: "/usr/local/bin"

	jobs:
	buildifier-lint:
	runs-on: ubuntu-22.04
	defaults:
	run:
	shell: bash
	timeout-minutes: 6
	steps:
	- name: "Checking out repository"
	uses: actions/checkout@e2f20e631ae6d7dd3b768f56a5d2af784dd54791 # v2.5.0
	- name: "Install buildifier"
	run: parallel --ungroup --retries 3 --delay 15 --nonall -- go install github.com/bazelbuild/buildtools/buildifier@433ea85 # 6.4.0
	- name: "Run buildifier"
	run: buildifier --lint=warn --warnings=-out-of-order-load $(find xla/ -type f -name "BUILD" -or -name "*bzl" \| grep -v /tsl/)

Provide feedback