Searched defs:src_indices (Results 1 – 8 of 8) sorted by relevance
34 tflite::gpu::Tensor<BHWC, DataType::INT32> src_indices; in GatherWidthIntTest() local74 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
1960 const uint8x8_t src_indices = InterleaveLow8(src_indices0, src_indices1); in ConvolveKernelHorizontal2Tap() local2011 const uint8x16_t src_indices = in ConvolveKernelHorizontal2Tap() local
249 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local446 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
1700 std::vector<int> src_indices; in ReplaceResourceArgsWithVarHandleOps() local1752 std::vector<int> src_indices; in ReplaceResourceArgsWithVarHandleOps() local1951 std::vector<int> src_indices; in ReplaceAndPartitionXLAShardingVariable() local
228 std::vector<int64_t> src_indices(tile_assignment.num_dimensions(), 0); in Subgroup() local
804 const __m128i src_indices, in PrepareSourceVectors()863 const __m128i src_indices = _mm_packus_epi16(src_indices16, src_indices16); in HorizontalScaleIndices() local