| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | remove_successive_transpose.cc | 27 bool TransformsToIdentity(std::vector<int> const& perm1, in TransformsToIdentity() argument
29 if (perm2.size() != perm1.size() || perm1.empty()) { in TransformsToIdentity()
35 for (size_t i = 0; i < perm1.size(); ++i) { in TransformsToIdentity()
36 if (perm1[i] < 0 || perm1[i] >= static_cast<int>(perm1.size()) || in TransformsToIdentity()
37 perm2[i] < 0 || perm2[i] >= static_cast<int>(perm1.size())) { in TransformsToIdentity()
40 if (perm1[perm2[i]] != static_cast<int>(i)) { in TransformsToIdentity()
|
| /external/webp/src/dsp/ |
| D | lossless_sse41.c | 34 const __m128i perm1 = _mm_setr_epi8(-1, 1, -1, 1, -1, 5, -1, 5, in TransformColorInverse_SSE41() local
41 const __m128i B = _mm_shuffle_epi8(A, perm1); // argb -> g0g0 in TransformColorInverse_SSE41()
65 const __m128i a1 = _mm_shuffle_epi8(in1, perm1); \
86 const __m128i perm1 = _mm_shuffle_epi32(perm0, 0x39); in ConvertBGRAToRGB_SSE41() local
104 const __m128i perm1 = _mm_shuffle_epi32(perm0, 0x39); in ConvertBGRAToBGR_SSE41() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_layout_helper.cc | 78 DenseIntElementsAttr perm1) { in AreCancellablePermutations() argument
79 if (perm0.getNumElements() == 0 || perm1.getNumElements() == 0) return false; in AreCancellablePermutations()
80 if (perm0.getNumElements() != perm1.getNumElements()) return false; in AreCancellablePermutations()
87 for (const auto &value : perm1.getValues<APInt>()) in AreCancellablePermutations()
|
| D | tf_ops_layout_helper.h | 48 DenseIntElementsAttr perm1);
|
| D | tf_ops_n_z.cc | 2707 DenseIntElementsAttr perm1; in FoldCancellableTranspose() local
2709 !matchPattern(transpose.perm(), m_Constant(&perm1))) in FoldCancellableTranspose()
2713 if (!AreCancellablePermutations(perm0, perm1)) return {}; in FoldCancellableTranspose()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/python_tests/ |
| D | tf_transpose_test.py | 230 perm1 = np.array([0, 1]).astype(np.int32)
235 [res1] = jitrt.execute(compiled, [tensor, perm1])
237 np.testing.assert_allclose(res1, np.transpose(tensor, perm1), atol=0.0)
255 perm1 = np.array([0, 1]).astype(np.int64)
257 [res] = jitrt.execute(compiled, [tensor, perm0, perm1])
259 res, np.transpose(np.transpose(tensor, perm0), perm1), atol=0.0)
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/tests/ |
| D | remove_successive_transpose_test.cc | 59 void CreateGraph(const std::vector<int>& perm1, in CreateGraph() argument
75 transpose_op->perm = perm1; in CreateGraph()
|
| /external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
| D | pa_avx.cpp | 755 const simdscalari perm1 = _simd_set_epi32(6, 3, 0, 5, 2, 7, 4, 1); in PaTriList2()
810 v1[i] = _simd_permute_ps(temp1, perm1); in PaTriList2()
838 …const simd16scalari perm1 = _simd16_set_epi32(14, 11, 8, 5, 2, 15, 12, 9, 6, 3, 0, 13, 10, 7, 4,… in PaTriList2_simd16() local
842 simd16scalar perm1 = _simd16_setzero_ps(); in PaTriList2_simd16()
875 v1[i] = _simd16_permute_ps(temp1, perm1); in PaTriList2_simd16()
888 perm1 = _simd16_permute2f128_ps(temp1, temp1, 0xB1); // (1, 0, 3, 2) => 01 00 11 10 => 0xB1 in PaTriList2_simd16()
889 temp1 = _simd16_blend_ps(temp1, perm1, 0x6666); // 0010 0010 0010 0010 in PaTriList2_simd16()
890 perm1 = _simd16_permute2f128_ps(temp1, temp1, 0x4E); // (2, 3, 0, 1) => 10 11 00 01 => 0x4E in PaTriList2_simd16()
891 v1[i] = _simd16_blend_ps(temp1, perm1, 0x1818); // 0001 1000 0001 1000 in PaTriList2_simd16()
1169 …simd16scalar perm1 = _simd16_permute2f128_ps(tempb, tempb, 0x39); // (0 3 2 1) = 00 11 10 01 // b4… in PaTriStrip1_simd16() local
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 1805 Output perm1 = ops::Const(s.WithOpName("perm1"), {0, 2, 3, 1}, {4}); in TEST_F() local
1808 Output transpose1 = ops::Transpose(s.WithOpName("transpose1"), inputs, perm1); in TEST_F()
1870 Output perm1 = ops::Const(s, {0, 2, 3, 1}, {4}); in TEST_F() local
1873 Output branch1 = ops::Transpose(s, ops::Transpose(s, split[1], perm1), perm2); in TEST_F()
1971 Output perm1 = ops::Const(s.WithOpName("perm1"), {0, 2, 3, 1}, {4}); in TEST_F() local
1974 s.WithOpName("transpose1").WithControlDependencies(perm2), inputs, perm1); in TEST_F()
|