Searched refs:rhs0 (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | banded_triangular_solve_op_test.py | 106 rhs0 = np.array([[1.]])
107 self._verifySolveAllWaysReal(matrix, rhs0)
114 rhs0 = np.array([[1.], [1.]])
115 self._verifySolveAllWaysReal(matrix, rhs0)
131 rhs0 = np.array([[1., 0., 1.], [0., 1., 1.], [-1., 2., 1.], [0., -1., -1.]])
132 self._verifySolveAllWaysReal(matrix, rhs0)
135 rhs0 = np.random.randn(6, 4)
139 self._verifySolveAllWaysReal(matrix, rhs0)
143 self._verifySolveAllWaysReal(matrix, rhs0)
151 rhs0 = np.array([[1. + 1j]])
[all …]
|
| D | matrix_triangular_solve_op_test.py | 101 rhs0 = np.array([[1.]])
102 self._verifySolveAllWaysReal(matrix, rhs0)
105 rhs0 = np.array([[1.], [1.]])
106 self._verifySolveAllWaysReal(matrix, rhs0)
117 rhs0 = np.array([[1. + 1j]])
118 self._verifySolveAllWaysComplex(matrix, rhs0)
122 rhs0 = np.array([[1.], [1.]]).astype(np.complex64)
123 rhs0 += 1j * rhs0
124 self._verifySolveAllWaysComplex(matrix, rhs0)
|
| /external/ruy/ruy/ |
| D | kernel_avx2_fma.cc | 292 __m256i rhs0 = _mm256_lddqu_si256(reinterpret_cast<const __m256i*>(
297 _mm256_permute2f128_si256(rhs0, rhs0, 0); // [0 1 2 3 0 1 2 3]
299 _mm256_permute2f128_si256(rhs0, rhs0, 0x11); // [4 5 6 7 4 5 6 7]
|
| D | kernel_avx.cc | 645 __m256i rhs0 = _mm256_lddqu_si256(reinterpret_cast<const __m256i*>(
650 _mm256_permute2f128_si256(rhs0, rhs0, 0); // [0 1 2 3 0 1 2 3]
652 _mm256_permute2f128_si256(rhs0, rhs0, 0x11); // [4 5 6 7 4 5 6 7]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier_test.cc | 5337 HloInstruction* rhs0 = builder.AddInstruction( in TEST_P() local
5346 HloInstruction::CreateConcatenate(rhs_shape, {rhs0, rhs1, rhs2}, 0)); in TEST_P()
|