Searched defs:src_indices (Results 1 – 9 of 9) sorted by relevance
34 tflite::gpu::Tensor<BHWC, DataType::INT32> src_indices; in GatherWidthIntTest() local74 TensorFloat32 src_indices; in GatherWidthTest() local
888 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local927 const uint8x8_t src_indices = in ConvolveKernelHorizontal2Tap() local989 const uint8x8_t src_indices = in ConvolveKernelHorizontalPositive4Tap() local1062 const uint8x8_t src_indices[4] = {src_indices_base, in ConvolveKernelHorizontalSigned4Tap() local1132 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned6Tap() local1229 const uint8x8_t src_indices = in ConvolveKernelHorizontalMixed6Tap() local1318 const uint8x8_t src_indices = in ConvolveKernelHorizontalSigned8Tap() local
1964 const uint8x8_t src_indices = InterleaveLow8(src_indices0, src_indices1); in ConvolveKernelHorizontal2Tap() local2015 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
70 xla::XlaOp src_indices = in Compile() 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
813 const __m128i src_indices, in PrepareSourceVectors()872 const __m128i src_indices = _mm_packus_epi16(src_indices16, src_indices16); in HorizontalScaleIndices() local
228 std::vector<int64_t> src_indices(tile_assignment.num_dimensions(), 0); in Subgroup() local
455 auto src_indices = src._indices(); in add_out_sparse_contiguous() local1107 Tensor src_indices = src._indices(); in mul_out_sparse_cpu() local