Home
last modified time | relevance | path

Searched defs:apply_relu (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/libtextclassifier/native/lang_id/common/
H A Dembedding-network.cc68 void SparseReluProductPlusBias(bool apply_relu, in SparseReluProductPlusBias()
284 const bool apply_relu = i > 0; in ComputeFinalScores() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp252 at::Tensor PackedLinearWeight::apply_relu( in apply_relu() function in PackedLinearWeight
774 at::Tensor PackedLinearWeightsQnnp::apply_relu( in apply_relu() function in PackedLinearWeightsQnnp
886 at::Tensor PackedLinearWeightsOnednn::apply_relu( in apply_relu() function in PackedLinearWeightsOnednn
H A Dqconv.cpp245 at::Tensor PackedConvWeight<kSpatialDim>::apply_relu( in apply_relu() function in PackedConvWeight
1069 at::Tensor PackedConvWeightsQnnp<kSpatialDim>::apply_relu( in apply_relu() function in PackedConvWeightsQnnp
1124 at::Tensor PackedConvWeightsOnednn<kSpatialDim>::apply_relu( in apply_relu() function in PackedConvWeightsOnednn
H A Dfbgemm_utils.h127 at::Tensor apply_relu( in apply_relu() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqnnpack_utils.h52 at::Tensor apply_relu( in apply_relu() function
H A Dqlinear.cpp222 at::Tensor PackedLinearWeight::apply_relu( in apply_relu() function in ao::sparse::PackedLinearWeight
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DConv.cpp329 at::Tensor PackedConvWeightCudnn<kSpatialDim>::apply_relu( in apply_relu() function in PackedConvWeightCudnn
H A DLinear.cpp331 at::Tensor PackedLinearWeightCudnn::apply_relu( in apply_relu() function in PackedLinearWeightCudnn