Searched defs:src_indices (Results 1 – 8 of 8) sorted by relevance
33 TensorFloat32 src_indices; in GatherWidthTest() local
70 xla::XlaOp src_indices = in Compile() local
890 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local929 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local991 const uint8x8_t src_indices = in ConvolveKernelHorizontalPositive4Tap() local1064 const uint8x8_t src_indices[4] = {src_indices_base, in ConvolveKernelHorizontalSigned4Tap() local1134 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned6Tap() local1231 const uint8x8_t src_indices = in ConvolveKernelHorizontalMixed6Tap() local1320 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned8Tap() local
1968 const uint8x8_t src_indices = InterleaveLow8(src_indices0, src_indices1); in ConvolveKernelHorizontal2Tap() local2019 const uint8x16_t src_indices = in ConvolveKernelHorizontal2Tap() local
244 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local432 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
1647 std::vector<int> src_indices; in ReplaceResourceArgsWithVarHandleOps() local1701 std::vector<int> src_indices; in ReplaceResourceArgsWithVarHandleOps() local1873 std::vector<int> src_indices; in ReplaceAndPartitionXLAShardingVariable() local
834 const __m128i src_indices, in PrepareSourceVectors()893 const __m128i src_indices = _mm_packus_epi16(src_indices16, src_indices16); in HorizontalScaleIndices() local
306 std::vector<int64> src_indices(indices.size(), -1); in TransposeSharding() local