Home
last modified time | relevance | path

Searched defs:weight_ptr (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/libtextclassifier/native/lang_id/common/
H A Dembedding-network.cc103 const float *weight_ptr = in SparseReluProductPlusBias() local
129 const float16 *weight_ptr = in SparseReluProductPlusBias() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEQLSTMLayerNormalizationKernel.cpp230 const int16_t *weight_ptr, in normalize_qasymm16()
294 const auto weight_ptr = reinterpret_cast<const InputDataType *>(weight_iterator.ptr()); in compute_qsymm16() local
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_linear_scratch_example.cpp96 const scalar_t* const weight_ptr = in linear_scratch_example() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dlsh_projection.cc126 const float* weight_ptr = GetTensorData<float>(weight); in RunningSignBit() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DQuantizedLinear.cpp308 const int8_t* weight_ptr = weight_contig.const_data_ptr<int8_t>(); in fbgemm_pack_quantized_matrix() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DLinear.cpp388 auto weight_ptr = (float*)(w.get_data_handle()); in mkl_linear() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqconv_prepack.cpp459 std::unique_ptr<ideep::tensor> weight_ptr(packed_weight_p); in prepack() local
H A Dqlinear_prepack.cpp266 std::unique_ptr<ideep::tensor> weight_ptr(packed_weight_p); in prepack() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dscale_and_translate_op.cc506 for (const float* weight_ptr = weights_start; weight_ptr != weights_end; in GatherColumns() local