Searched refs:a_vals (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | qr_test.cc | 38 xla::Array2D<float> a_vals({ in XLA_TEST_F() local
46 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
59 ComputeAndCompareR2<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
66 xla::Array3D<float> a_vals({ in XLA_TEST_F() local
82 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
89 ComputeAndCompareR3<float>(&builder, a_vals, {a_data.get()}, in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cholesky_test.cc | 43 Array2D<float> a_vals({ in XLA_TEST_F() local
51 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
69 Array2D<float> a_vals({ in XLA_TEST_F() local
77 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
94 Array2D<float> a_vals({ in XLA_TEST_F() local
102 auto a_data = CreateR2Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
117 Array3D<float> a_vals({ in XLA_TEST_F() local
133 auto a_data = CreateR3Parameter<float>(a_vals, 0, "a", &builder, &a); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client_test.py | 1190 a_vals = np.array(
1197 c.TriangularSolve(c.Constant(a_vals), c.Constant(b_vals), left_side=False,
|