Function Array4DLinearIndex

tensorflow/compiler/xla/array4d_test.cc:30–34

// array idx represents.
template <typename T>
int64 Array4DLinearIndex(const Array4D<T>& arr, absl::Span<const int64> idx) {
  EXPECT_EQ(4, idx.size());
  return (idx[3] + idx[2] * arr.n4() + idx[1] * arr.n3() * arr.n4() +
          idx[0] * arr.n2() * arr.n3() * arr.n4());
}

TEST(Array4dTest, UninitializedDimsCtor) {
  Array4D<int> empty(2, 3, 4, 5);