Home
last modified time | relevance | path

Searched defs:input_weights (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLFuseBatchNormalizationKernel.cpp42 Status validate_arguments(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITens… in validate_arguments()
109 void CLFuseBatchNormalizationKernel::configure(const ICLTensor *input_weights, const ICLTensor *bn_… in configure()
117 …onKernel::configure(const CLCompileContext &compile_context, const ICLTensor *input_weights, const… in configure()
182 Status CLFuseBatchNormalizationKernel::validate(const ITensorInfo *input_weights, const ITensorInfo… in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLFuseBatchNormalization.cpp44 void CLFuseBatchNormalization::configure(const ICLTensor *input_weights, const ICLTensor *bn_mean, … in configure()
52 …lization::configure(const CLCompileContext &compile_context, const ICLTensor *input_weights, const… in configure()
61 Status CLFuseBatchNormalization::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_m… in validate()
H A DCLLSTMLayerQuantized.cpp336 …const TensorInfo input_weights(TensorShape(input_size, 4 * output_size), 1, DataType::QASYMM… in validate() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbasic_rnn.cc65 const TfLiteTensor* input_weights; in Prepare() local
202 const TfLiteTensor* input_weights, in EvalFloat()
230 const TfLiteTensor* input_weights, in EvalHybrid()
286 const TfLiteTensor* input_weights; in Eval() local
H A Dunidirectional_sequence_rnn.cc66 const TfLiteTensor* input_weights; in Prepare() local
208 const TfLiteTensor* input_weights, in EvalFloat()
269 const TfLiteTensor* input, const TfLiteTensor* input_weights, in EvalHybrid()
360 const TfLiteTensor* input_weights; in Eval() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEFuseBatchNormalizationKernel.cpp153 Status validate_arguments(const ITensorInfo *input_weights, const ITensorInfo *bn_mean, const ITens… in validate_arguments()
220 void NEFuseBatchNormalizationKernel::configure(const ITensor *input_weights, const ITensor *bn_mean… in configure()
271 Status NEFuseBatchNormalizationKernel::validate(const ITensorInfo *input_weights, const ITensorInfo… in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEFuseBatchNormalization.cpp43 void NEFuseBatchNormalization::configure(const ITensor *input_weights, const ITensor *bn_mean, cons… in configure()
55 Status NEFuseBatchNormalization::validate(const ITensorInfo *input_weights, const ITensorInfo *bn_m… in validate()
H A DNELSTMLayerQuantized.cpp303 …const TensorInfo input_weights(TensorShape(input_size, 4 * output_size), 1, DataType::QASYMM… in validate() local
/aosp_15_r20/external/libopus/src/
H A Dmlp.h38 const opus_int8 *input_weights; member
46 const opus_int8 *input_weights; member
/aosp_15_r20/external/rnnoise/src/
H A Drnn.h46 const rnn_weight *input_weights; member
54 const rnn_weight *input_weights; member
/aosp_15_r20/external/ComputeLibrary/tests/datasets/
H A DLSTMLayerDataset.h126 …void add_config(TensorShape src, TensorShape input_weights, TensorShape recurrent_weights, TensorS… in add_config()
/aosp_15_r20/external/libopus/dnn/
H A Dnnet.c72 void compute_generic_gru(const LinearLayer *input_weights, const LinearLayer *recurrent_weights, fl… in compute_generic_gru()
H A Dlossgen.c83 void compute_generic_gru_lossgen(const LinearLayer *input_weights, const LinearLayer *recurrent_wei… in compute_generic_gru_lossgen()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DRNN.cpp54 const RunTimeOperandInfo* input_weights = GetInput(operation, operands, kWeightsTensor); in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dquantize.cc267 const auto& input_weights = model->GetArray(op.inputs[weights_input_index]); in ChooseQuantizationForOperatorInput() local