Home
last modified time | relevance | path

Searched refs:CreateR2Parameter (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc106 CreateR2Parameter<float>(Array2D<float>(0, 0), 0, "a", &builder, &a); in XLA_TEST_F()
108 CreateR2Parameter<float>(Array2D<float>(0, 10), 1, "b", &builder, &b); in XLA_TEST_F()
122 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F()
123 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
143 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F()
144 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
164 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F()
165 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
185 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F()
186 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
[all …]
Dcholesky_test.cc51 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
77 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
102 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
Dclient_library_test_base.h356 std::unique_ptr<GlobalData> CreateR2Parameter(
577 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR2Parameter( in CreateR2Parameter() function
Dbroadcast_simple_test.cc253 auto x_data = CreateR2Parameter<bool>(x_vals, 0, "x", &b, &x); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc61 auto a_data = CreateR2Parameter<float>(BValsRight(), 0, "a", &builder, &a); in XLA_TEST_F()
90 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
91 auto b_data = CreateR2Parameter<float>({{9, 1, -10}}, 1, "b", &builder, &b); in XLA_TEST_F()
110 CreateR2Parameter<int>({{1, 2}, {3, 4}}, 0, "input", &builder, &input); in XLA_TEST_F()
112 CreateR2Parameter<int>({{0, 0}, {1, 0}}, 1, "index", &builder, &index); in XLA_TEST_F()
124 CreateR2Parameter<float>({{0.1427, 0.0231, -0.5414, -1.0009}, in XLA_TEST_F()
143 CreateR2Parameter<float>({{0.1427, 0.0231, -0.5414, -1.0009}, in XLA_TEST_F()
Dself_adjoint_eig_test.cc199 auto a_data = CreateR2Parameter<float>(low_rank_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
215 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F()
229 auto a_data = CreateR2Parameter<float>(matrix2d_8x8_, 0, "a", &builder, &a); in XLA_TEST_F()
244 auto a_data = CreateR2Parameter<int>(wrong_type_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
255 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
268 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
281 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
294 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
307 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
Dsvd_test.cc144 auto a_data = CreateR2Parameter<float>(simple_2d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
210 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
222 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
234 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
246 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
258 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
270 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
Dqr_test.cc46 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()