Searched defs:fused_block_size (Results 1 – 1 of 1) sorted by relevance
28 const int fused_block_size = block_size + 0; in EmbeddingLookupIdx_int32_t_float_float__avx2_fma() local477 const int64_t fused_block_size = block_size + 0; in EmbeddingLookupIdx_int64_t_float_float__avx2_fma() local926 const int fused_block_size = block_size + 0; in EmbeddingLookupIdx_int32_t_half_float__avx2_fma() local1495 const int64_t fused_block_size = block_size + 0; in EmbeddingLookupIdx_int64_t_half_float__avx2_fma() local2064 const int fused_block_size = block_size + 0; in EmbeddingLookupIdx_int32_t_bfloat16_float__avx2_fma() local2696 const int64_t fused_block_size = block_size + 0; in EmbeddingLookupIdx_int64_t_bfloat16_float__avx2_fma() local3328 const int fused_block_size = block_size + 0; in EmbeddingLookupIdx_int32_t_uint8_t_float__avx2_fma() local3915 const int64_t fused_block_size = block_size + 0; in EmbeddingLookupIdx_int64_t_uint8_t_float__avx2_fma() local