Home
last modified time | relevance | path

Searched refs:GetUnitMatrix3D (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dsvd_test.cc59 Array3D<float> GetUnitMatrix3D(int32 batch_dim, int32 mat_dim) { in GetUnitMatrix3D() function in xla::SVDTest
173 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 4), {a_data.get()}, in XLA_TEST_F()
185 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(2, 5), {a_data.get()}, in XLA_TEST_F()
Dself_adjoint_eig_test.cc74 Array3D<float> GetUnitMatrix3D(const Array3D<float>& matrix) { in GetUnitMatrix3D() function in xla::SelfAdjointEigTest
191 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(batch_3d_4x4_), in XLA_TEST_F()