| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 68 auto a_data = CreateR3Parameter<int32>(input, 0, "a", &builder, &a); in XLA_TEST_F()
83 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal()
98 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestSetMatrixDiagonal()
184 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
185 auto row_data = CreateR3Parameter<float>({{{9, 1, 0, 0}}, {{2, 4, 0, 0}}}, 1, in XLA_TEST_F()
205 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
206 auto row_data = CreateR3Parameter<float>({{{9, 1, 0, 0}}, {{2, 4, 0, 0}}}, 1, in XLA_TEST_F()
|
| D | tridiagonal_test.cc | 64 auto lower_diagonal_data = CreateR3Parameter<float>( in XLA_TEST_P()
66 auto main_diagonal_data = CreateR3Parameter<float>( in XLA_TEST_P()
68 auto upper_diagonal_data = CreateR3Parameter<float>( in XLA_TEST_P()
70 auto rhs_data = CreateR3Parameter<float>(rhs, 3, "rhs", &builder, &rhs_xla); in XLA_TEST_P()
|
| D | svd_test.cc | 141 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
153 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
166 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
183 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); in XLA_TEST_F()
|
| D | self_adjoint_eig_test.cc | 149 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
161 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F()
174 auto a_data = CreateR3Parameter<float>( in XLA_TEST_F()
187 auto a_data = CreateR3Parameter<float>(batch_3d_4x4_, 0, "a", &builder, &a); in XLA_TEST_F()
|
| D | logdet_test.cc | 95 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
|
| D | slicing_test.cc | 78 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
240 CreateR3Parameter<int>({{{0, 1, 2, 3}, {4, 5, 6, 7}, {8, 9, 10, 11}}, in XLA_TEST_F()
|
| D | qr_test.cc | 116 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cholesky_test.cc | 81 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
196 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
|
| D | triangular_solve_test.cc | 431 auto a_data = CreateR3Parameter<float>(avals, 0, "a", &builder, &a); in XLA_TEST_F()
432 auto b_data = CreateR3Parameter<float>(bvals, 1, "b", &builder, &b); in XLA_TEST_F()
|
| D | broadcast_simple_test.cc | 254 auto y_data = CreateR3Parameter<bool>(y_vals, 1, "y", &b, &y); in XLA_TEST_F()
386 auto r1 = CreateR3Parameter(r1d, 1, "r1", &b, &r1h); in XLA_TEST_F()
387 auto r3 = CreateR3Parameter(r3d, 0, "r3", &b, &r3h); in XLA_TEST_F()
|
| D | concat_test.cc | 487 auto p0 = CreateR3Parameter<float>(arr0, /*parameter_number=*/0, "p0", in XLA_TEST_F()
490 auto p1 = CreateR3Parameter<float>(arr1, /*parameter_number=*/1, "p1", in XLA_TEST_F()
|
| D | dynamic_ops_test.cc | 547 CreateR3Parameter<T>(input_values, 0, "input_values", &builder, &input); in RunR3Contiguous()
550 std::unique_ptr<GlobalData> update_data = CreateR3Parameter<T>( in RunR3Contiguous()
|
| D | client_library_test_base.h | 378 std::unique_ptr<GlobalData> CreateR3Parameter(
647 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR3Parameter( in CreateR3Parameter() function
|
| D | batch_normalization_test.cc | 345 auto operand = CreateR3Parameter<float>(Array3D<float>(260, 2, 2, 1.0f), in XLA_TEST_P()
376 auto operand = CreateR3Parameter<float>({{{0.0f}, {10.0f}, {20.0f}, {30.0f}}}, in XLA_TEST_P()
|