Searched refs:batch_3d_4x4_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | self_adjoint_eig_test.cc | 39 batch_3d_4x4_ = Array3D<float>{ in SetUp()
139 Array3D<float> batch_3d_4x4_; member in xla::SelfAdjointEigTest
149 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
153 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
162 ExtractTriangularMatrix(batch_3d_4x4_, true), 0, "a", &builder, &a); in XLA_TEST_F()
166 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
175 ExtractTriangularMatrix(batch_3d_4x4_, false), 0, "a", &builder, &a); in XLA_TEST_F()
179 ComputeAndCompareR3<float>(&builder, batch_3d_4x4_, {a_data.get()}, in XLA_TEST_F()
187 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
191 ComputeAndCompareR3<float>(&builder, GetUnitMatrix3D(batch_3d_4x4_), in XLA_TEST_F()
|