Searched refs:batch_3d_4x5_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 42 batch_3d_4x5_ = Array3D<float>{ in SetUp()
131 Array3D<float> batch_3d_4x5_; member in xla::SVDTest
156 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
160 ComputeAndCompareR3<float>(&builder, batch_3d_4x5_, {a_data.get()}, in XLA_TEST_F()
168 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
181 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
198 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
|