Searched refs:perm0 (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
D | input_data.py | 209 perm0 = numpy.arange(self._num_examples) 210 numpy.random.shuffle(perm0) 211 self._images = self.images[perm0] 212 self._labels = self.labels[perm0]
|
/external/mesa3d/src/gallium/drivers/swr/rasterizer/core/ |
D | pa_avx.cpp | 754 const simdscalari perm0 = _simd_set_epi32(5, 2, 7, 4, 1, 6, 3, 0); in PaTriList2() 809 v0[i] = _simd_permute_ps(temp0, perm0); in PaTriList2() 837 …const simd16scalari perm0 = _simd16_set_epi32(13, 10, 7, 4, 1, 14, 11, 8, 5, 2, 15, 12, 9, 6, 3,… in PaTriList2_simd16() local 841 simd16scalar perm0 = _simd16_setzero_ps(); in PaTriList2_simd16() 874 v0[i] = _simd16_permute_ps(temp0, perm0); in PaTriList2_simd16() 882 perm0 = _simd16_permute2f128_ps(temp0, temp0, 0xB1); // (1, 0, 3, 2) => 01 00 11 10 => 0xB1 in PaTriList2_simd16() 883 temp0 = _simd16_blend_ps(temp0, perm0, 0x4444); // 0010 0010 0010 0010 in PaTriList2_simd16() 884 perm0 = _simd16_permute2f128_ps(temp0, temp0, 0x4E); // (2, 3, 0, 1) => 10 11 00 01 => 0x4E in PaTriList2_simd16() 885 v0[i] = _simd16_blend_ps(temp0, perm0, 0x3838); // 0001 1100 0001 1100 in PaTriList2_simd16() 1168 …simd16scalar perm0 = _simd16_permute2f128_ps(tempa, tempa, 0x39); // (0 3 2 1) = 00 11 10 01 // a4… in PaTriStrip1_simd16() local [all …]
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_ops_helpers.inc | 344 static bool AreCancellablePermutations(DenseIntElementsAttr perm0, 346 if (perm0.getNumElements() == 0 || perm1.getNumElements() == 0) return false; 347 if (perm0.getNumElements() != perm1.getNumElements()) return false; 350 for (const auto &value : perm0.getIntValues())
|
D | tf_ops_n_z.cc | 2513 DenseIntElementsAttr perm0; in FoldCancellableTranspose() local 2515 if (!matchPattern(op.perm(), m_Constant(&perm0)) || in FoldCancellableTranspose() 2520 if (!AreCancellablePermutations(perm0, perm1)) return {}; in FoldCancellableTranspose()
|