Searched refs:mat_a (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | matrix_functions_test.py | 43 mat_a = np.random.rand(size, size)
44 mat = np.dot(mat_a, mat_a.T)
54 mat_a = np.random.rand(size, size)
55 mat = np.dot(mat_a, mat_a.T)
|
| D | matrix_functions.py | 26 def matrix_square_root(mat_a, mat_a_size, iter_count=100, ridge_epsilon=1e-4): argument
57 residual = mat_a - mat_a_approx
62 mat_a = mat_a + ridge_epsilon * identity
63 norm = math_ops.sqrt(math_ops.reduce_sum(mat_a * mat_a))
64 mat_init_y = mat_a / norm
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session_test.cc | 235 auto mat_a = outputs[0].matrix<float>(); in TEST_F() local
237 EXPECT_FLOAT_EQ(3.0, mat_a(0, 0)); in TEST_F()
238 EXPECT_FLOAT_EQ(2.0, mat_a(0, 1)); in TEST_F()
239 EXPECT_FLOAT_EQ(-1.0, mat_a(1, 0)); in TEST_F()
240 EXPECT_FLOAT_EQ(0.0, mat_a(1, 1)); in TEST_F()
|