Home
last modified time | relevance | path

Searched defs:src_indices (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Ddata_format_ops.cc70 xla::XlaOp src_indices = in Compile() local
/external/libgav1/libgav1/src/dsp/arm/
Dconvolve_neon.cc872 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local
912 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local
973 const uint8x8_t src_indices = in ConvolveKernelHorizontalPositive4Tap() local
1045 const uint8x8_t src_indices[4] = {src_indices_base, in ConvolveKernelHorizontalSigned4Tap() local
1114 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned6Tap() local
1210 const uint8x8_t src_indices = in ConvolveKernelHorizontalMixed6Tap() local
1297 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned8Tap() local
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d.cc244 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
419 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
/external/libgav1/libgav1/src/dsp/x86/
Dconvolve_sse4.cc838 inline void PrepareSourceVectors(const uint8_t* src, const __m128i src_indices, in PrepareSourceVectors()
897 const __m128i src_indices = _mm_packus_epi16(src_indices16, src_indices16); in HorizontalScaleIndices() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc286 std::vector<int64> src_indices(indices.size(), -1); in TransposeSharding() local