Searched refs:Concat3D (Results 1 – 2 of 2) sorted by relevance
630 auto concat0 = ReferenceUtil::Concat3D(y_bcast3d, x3d, 0); in XLA_TEST_F()631 auto concat1 = ReferenceUtil::Concat3D(*concat0, z_bcast3d, 1); in XLA_TEST_F()
257 static std::unique_ptr<Array3D<T>> Concat3D(const Array3D<T>& lhs, in Concat3D() function