Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dtriangular_solve_test.cc108 CreateR2Parameter<float>(Array2D<float>(0, 0), 0, "a", &builder, &a); in XLA_TEST_F()
110 CreateR2Parameter<float>(Array2D<float>(0, 10), 1, "b", &builder, &b); in XLA_TEST_F()
124 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F()
125 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
145 auto a_data = CreateR2Parameter<float>(AValsLower(), 0, "a", &builder, &a); in XLA_TEST_F()
146 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
166 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F()
167 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
187 auto a_data = CreateR2Parameter<float>(AValsUpper(), 0, "a", &builder, &a); in XLA_TEST_F()
188 auto b_data = CreateR2Parameter<float>(BValsRight(), 1, "b", &builder, &b); in XLA_TEST_F()
[all …]
Dcholesky_test.cc50 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
114 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
140 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
165 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
Dclient_library_test_base.h364 std::unique_ptr<GlobalData> CreateR2Parameter(
633 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.cc63 auto a_data = CreateR2Parameter<float>(BValsRight(), 0, "a", &builder, &a); in XLA_TEST_F()
92 auto a_data = CreateR2Parameter<float>(AValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
93 auto b_data = CreateR2Parameter<float>({{9, 1, -10}}, 1, "b", &builder, &b); in XLA_TEST_F()
112 CreateR2Parameter<int>({{1, 2}, {3, 4}}, 0, "input", &builder, &input); in XLA_TEST_F()
114 CreateR2Parameter<int>({{0, 0}, {1, 0}}, 1, "index", &builder, &index); in XLA_TEST_F()
126 CreateR2Parameter<int>({{1, 2}, {3, 4}}, 0, "input", &builder, &input); in XLA_TEST_F()
128 CreateR2Parameter<int>({{0, 0}, {1, 0}}, 1, "index", &builder, &index); in XLA_TEST_F()
139 auto input_data = CreateR2Parameter<int>({{0, 0, 0}, {0, 0, 0}}, 0, "input", in XLA_TEST_F()
142 CreateR2Parameter<int>({{1, 0}, {1, 2}}, 1, "index", &builder, &index); in XLA_TEST_F()
144 CreateR2Parameter<int>({{1, 2}, {3, 4}}, 2, "src", &builder, &src); in XLA_TEST_F()
[all …]
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.cc129 auto a_data = CreateR2Parameter<float>(simple_2d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
197 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
209 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
221 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
235 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
249 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
263 auto a_data = CreateR2Parameter<float>(a_val, 0, "a", &builder, &a); in XLA_TEST_F()
Dlogdet_test.cc47 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
67 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
Dqr_test.cc49 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
78 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
Dmatrix_test.cc100 CreateR2Parameter<T>(Array2D<T>{kv.second}, 1, "d", &builder, &b); in TestSetMatrixDiagonal()