Searched refs:fw_output_ (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceLSTM.cpp | 191 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
516 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
568 params_.merge_outputs ? GetBuffer<float>(fw_output_) + n_fw_output_elements in Eval()
573 mergeThirdDimension(GetBuffer<float>(fw_output_), fw_output_dims, in Eval()
574 GetBuffer<float>(fw_output_) + n_fw_output_elements, in Eval()
577 GetBuffer<float>(fw_output_)); in Eval()
640 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
693 params_.merge_outputs ? GetBuffer<_Float16>(fw_output_) + n_fw_output_elements in Eval()
698 mergeThirdDimension(GetBuffer<_Float16>(fw_output_), fw_output_dims, in Eval()
699 GetBuffer<_Float16>(fw_output_) + n_fw_output_elements, in Eval()
[all …]
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | BidirectionalSequenceLSTM.h | 201 RunTimeOperandInfo* fw_output_; variable
|