| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | matrix_test.cc | 43 auto a_data = CreateR3Parameter<int32>(input, 0, "a", &builder, &a); in XLA_TEST_F()
69 auto a_data = CreateR3Parameter<T>(input, 0, "a", &builder, &a); in TestMatrixDiagonal()
104 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
105 auto row_data = CreateR3Parameter<float>({{{9, 1, 0, 0}}, {{2, 4, 0, 0}}}, 1, in XLA_TEST_F()
125 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
126 auto row_data = CreateR3Parameter<float>({{{9, 1, 0, 0}}, {{2, 4, 0, 0}}}, 1, in XLA_TEST_F()
|
| D | svd_test.cc | 156 auto a_data = CreateR3Parameter<float>(batch_3d_4x5_, 0, "a", &builder, &a); 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()
|
| 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 | qr_test.cc | 82 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
|
| D | slicing_test.cc | 76 CreateR3Parameter<float>(BatchedAValsFull(), 0, "a", &builder, &a); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | triangular_solve_test.cc | 421 auto a_data = CreateR3Parameter<float>(avals, 0, "a", &builder, &a); in XLA_TEST_F()
422 auto b_data = CreateR3Parameter<float>(bvals, 1, "b", &builder, &b); in XLA_TEST_F()
|
| D | cholesky_test.cc | 133 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); 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 | 484 auto p0 = CreateR3Parameter<float>(arr0, /*parameter_number=*/0, "p0", in XLA_TEST_F()
487 auto p1 = CreateR3Parameter<float>(arr1, /*parameter_number=*/1, "p1", in XLA_TEST_F()
|
| D | client_library_test_base.h | 370 std::unique_ptr<GlobalData> CreateR3Parameter(
591 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR3Parameter( in CreateR3Parameter() function
|
| D | dynamic_ops_test.cc | 546 CreateR3Parameter<T>(input_values, 0, "input_values", &builder, &input); in RunR3Contiguous()
549 std::unique_ptr<GlobalData> update_data = CreateR3Parameter<T>( in RunR3Contiguous()
|
| D | batch_normalization_test.cc | 284 auto operand = CreateR3Parameter<float>(Array3D<float>(260, 2, 2, 1.0f), in XLA_TEST_P()
315 auto operand = CreateR3Parameter<float>({{{0.0f}, {10.0f}, {20.0f}, {30.0f}}}, in XLA_TEST_P()
|