Searched refs:src_indices (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | data_format_ops.cc | 70 xla::XlaOp src_indices = in Compile() local
73 xla::TorchIndexSelect(dst_indices, src_indices, /*dim=*/0); in Compile()
|
| /external/libgav1/libgav1/src/dsp/arm/ |
| D | convolve_neon.cc | 872 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local
877 VQTbl1U8(src_vals, src_indices), in ConvolveKernelHorizontal2Tap()
878 VQTbl1U8(src_vals, vadd_u8(src_indices, vdup_n_u8(1)))}; in ConvolveKernelHorizontal2Tap()
912 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local
917 vtbl3_u8(src_vals, src_indices), in ConvolveKernelHorizontal2Tap()
918 vtbl3_u8(src_vals, vadd_u8(src_indices, vdup_n_u8(1)))}; in ConvolveKernelHorizontal2Tap()
973 const uint8x8_t src_indices = in ConvolveKernelHorizontalPositive4Tap() local
984 VQTbl1U8(src_vals, src_indices), in ConvolveKernelHorizontalPositive4Tap()
985 VQTbl1U8(src_vals, vadd_u8(src_indices, vdup_n_u8(1))), in ConvolveKernelHorizontalPositive4Tap()
986 VQTbl1U8(src_vals, vadd_u8(src_indices, vdup_n_u8(2))), in ConvolveKernelHorizontalPositive4Tap()
[all …]
|
| /external/libgav1/libgav1/src/dsp/x86/ |
| D | convolve_sse4.cc | 838 inline void PrepareSourceVectors(const uint8_t* src, const __m128i src_indices, in PrepareSourceVectors() argument
843 const int used_bytes = _mm_extract_epi8(src_indices, 15) + 1 + num_taps - 2; in PrepareSourceVectors()
845 source[0] = _mm_shuffle_epi8(src_vals, src_indices); in PrepareSourceVectors()
848 source[1] = _mm_shuffle_epi8(_mm_srli_si128(src_vals, 2), src_indices); in PrepareSourceVectors()
851 source[2] = _mm_shuffle_epi8(_mm_srli_si128(src_vals, 4), src_indices); in PrepareSourceVectors()
854 source[3] = _mm_shuffle_epi8(_mm_srli_si128(src_vals, 6), src_indices); in PrepareSourceVectors()
863 src_indices); in PrepareSourceVectors()
865 src_indices); in PrepareSourceVectors()
869 src_indices); in PrepareSourceVectors()
897 const __m128i src_indices = _mm_packus_epi16(src_indices16, src_indices16); in HorizontalScaleIndices() local
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 244 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
248 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
272 tensor_out.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
277 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
419 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
423 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
438 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
|
| /external/mesa3d/src/compiler/nir/ |
| D | nir_algebraic.py | 1035 for src_indices in itertools.product(range(len(rep)), repeat=num_srcs):
1036 if all(src_idx < op_worklist_index for src_idx in src_indices):
1039 srcs = tuple(rep[src_idx] for src_idx in src_indices)
1050 table[src_indices] = self.states.add(frozenset(parent))
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_sharding_util.cc | 286 std::vector<int64> src_indices(indices.size(), -1); in TransposeSharding() local
288 src_indices[perm_dimensions[i]] = indices[i]; in TransposeSharding()
290 *value = sharding.tile_assignment()(src_indices); in TransposeSharding()
|