Home
last modified time | relevance | path

Searched defs:scale_bias (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/caffe2/perfkernels/
H A Dembedding_lookup_idx_avx2.cc23 const float* scale_bias, in EmbeddingLookupIdx_int32_t_float_float__avx2_fma()
420 const float* scale_bias, in EmbeddingLookupIdx_int32_t_float_float_false__avx2_fma()
445 const float* scale_bias, in EmbeddingLookupIdx_int32_t_float_float_true__avx2_fma()
472 const float* scale_bias, in EmbeddingLookupIdx_int64_t_float_float__avx2_fma()
869 const float* scale_bias, in EmbeddingLookupIdx_int64_t_float_float_false__avx2_fma()
894 const float* scale_bias, in EmbeddingLookupIdx_int64_t_float_float_true__avx2_fma()
921 const float* scale_bias, in EmbeddingLookupIdx_int32_t_half_float__avx2_fma()
1438 const float* scale_bias, in EmbeddingLookupIdx_int32_t_half_float_false__avx2_fma()
1463 const float* scale_bias, in EmbeddingLookupIdx_int32_t_half_float_true__avx2_fma()
1490 const float* scale_bias, in EmbeddingLookupIdx_int64_t_half_float__avx2_fma()
[all …]
H A Dembedding_lookup_idx.cc29 const float* scale_bias, // optional scale & bias params for uint8 input in EmbeddingLookupGenericSlowIdx()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu58 dequantize_intx(uint32_t packedVals, float2 scale_bias, uint8_t offset_bits) { in dequantize_intx()
71 accumulate_packed_intx(float4* acc, uint32_t packedVals, float2 scale_bias, float sample_weight) { in accumulate_packed_intx()
143 float2 scale_bias; in embedding_bag_nbits_rowwise_offsets_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag.cpp108 const uint8_t* scale_bias = in embedding_lookup_fallback_impl() local
147 const uint8_t* scale_bias = in embedding_lookup_fallback_impl() local