Home
last modified time | relevance | path

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

/external/libtextclassifier/native/lang_id/common/
Dembedding-network.cc68 void SparseReluProductPlusBias(bool apply_relu, in SparseReluProductPlusBias()
284 const bool apply_relu = i > 0; in ComputeFinalScores() local
/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
Dqnnpack_utils.h52 at::Tensor apply_relu( in apply_relu() function
Dqlinear.cpp222 at::Tensor PackedLinearWeight::apply_relu( in apply_relu() function in ao::sparse::PackedLinearWeight
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dfbgemm_utils.h127 at::Tensor apply_relu( in apply_relu() function
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
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
/external/pytorch/aten/src/ATen/native/quantized/cudnn/
DConv.cpp329 at::Tensor PackedConvWeightCudnn<kSpatialDim>::apply_relu( in apply_relu() function in PackedConvWeightCudnn
DLinear.cpp331 at::Tensor PackedLinearWeightCudnn::apply_relu( in apply_relu() function in PackedLinearWeightCudnn