Home
last modified time | relevance | path

Searched refs:GetBuffer (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceLSTM.cpp37 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
42 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
458 const float* bwInput = GetBuffer<const float>(input_); in Eval()
462 bwInput = GetBuffer<const float>(aux_input_); in Eval()
472 fw_output_activation_state_buffer = GetBuffer<float>(fw_output_activation_state_); in Eval()
473 fw_output_cell_state_buffer = GetBuffer<float>(fw_output_cell_state_); in Eval()
485 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
486 GetBuffer<const float>(fw_input_to_input_weights_), in Eval()
487 GetBuffer<const float>(fw_input_to_forget_weights_), in Eval()
488 GetBuffer<const float>(fw_input_to_cell_weights_), in Eval()
[all …]
DLSTM.cpp38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
995 LSTMEvalFloat32(params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
996 GetBuffer<const float>(input_to_input_weights_), in Eval()
997 GetBuffer<const float>(input_to_forget_weights_), in Eval()
998 GetBuffer<const float>(input_to_cell_weights_), in Eval()
999 GetBuffer<const float>(input_to_output_weights_), in Eval()
1001 GetBuffer<const float>(recurrent_to_input_weights_), in Eval()
1002 GetBuffer<const float>(recurrent_to_forget_weights_), in Eval()
1003 GetBuffer<const float>(recurrent_to_cell_weights_), in Eval()
[all …]
DMultinomial.cpp49 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
54 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
95 convertFloat16ToFloat32(GetBuffer<_Float16>(input_), &inputDataFloat32); in Eval()
100 EvalFloat32(GetBuffer<float>(input_)); in Eval()
116 int32_t* seeds = GetBuffer<int32_t>(random_seeds_); in EvalFloat32()
146 auto* output_ptr_batch = GetBuffer<int32_t>(output_) + b * sample_count_; in EvalFloat32()
DQuantizedLSTM.cpp41 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
46 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
213 const uint8_t* submatrixValues = GetBuffer<uint8_t>(submatrix); in assignWeightsSubmatrix()
370 memcpy(bias + 0 * outputSize, GetBuffer<int32_t>(inputGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
371 memcpy(bias + 1 * outputSize, GetBuffer<int32_t>(cellGateBias_), sizeof(int32_t) * outputSize); in concatenateBiases()
372 memcpy(bias + 2 * outputSize, GetBuffer<int32_t>(forgetGateBias_), in concatenateBiases()
374 memcpy(bias + 3 * outputSize, GetBuffer<int32_t>(outputGateBias_), in concatenateBiases()
416 GetBuffer<const uint8_t>(input_), convertShapeToDims(input_->shape()), in eval()
417 GetBuffer<const uint8_t>(prevOutput_), convertShapeToDims(prevOutput_->shape()), in eval()
419 convertShapeToDims(biasShape), GetBuffer<const int16_t>(prevCellState_), in eval()
[all …]