Searched defs:normalize_by_lengths (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/caffe2/perfkernels/ |
| H A D | embedding_lookup_idx_avx2.cc | 24 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_float_float__avx2_fma()
421 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_float_float_false__avx2_fma()
446 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_float_float_true__avx2_fma()
473 bool normalize_by_lengths, in EmbeddingLookupIdx_int64_t_float_float__avx2_fma()
870 bool normalize_by_lengths, in EmbeddingLookupIdx_int64_t_float_float_false__avx2_fma()
895 bool normalize_by_lengths, in EmbeddingLookupIdx_int64_t_float_float_true__avx2_fma()
922 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_half_float__avx2_fma()
1439 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_half_float_false__avx2_fma()
1464 bool normalize_by_lengths, in EmbeddingLookupIdx_int32_t_half_float_true__avx2_fma()
1491 bool normalize_by_lengths, in EmbeddingLookupIdx_int64_t_half_float__avx2_fma()
[all …]
|
| H A D | embedding_lookup_idx.cc | 30 bool normalize_by_lengths, in EmbeddingLookupGenericSlowIdx()
|