Lines Matching refs:GetBuffer
38 inline T* GetBuffer(RunTimeOperandInfo* operand) { in GetBuffer() function
43 inline const T* GetBuffer(const RunTimeOperandInfo* operand) { in GetBuffer() function
459 const float* bwInput = GetBuffer<const float>(input_); in Eval()
463 bwInput = GetBuffer<const float>(aux_input_); in Eval()
473 fw_output_activation_state_buffer = GetBuffer<float>(fw_output_activation_state_); in Eval()
474 fw_output_cell_state_buffer = GetBuffer<float>(fw_output_cell_state_); in Eval()
486 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
487 GetBuffer<const float>(fw_input_to_input_weights_), in Eval()
488 GetBuffer<const float>(fw_input_to_forget_weights_), in Eval()
489 GetBuffer<const float>(fw_input_to_cell_weights_), in Eval()
490 GetBuffer<const float>(fw_input_to_output_weights_), in Eval()
492 GetBuffer<const float>(fw_recurrent_to_input_weights_), in Eval()
493 GetBuffer<const float>(fw_recurrent_to_forget_weights_), in Eval()
494 GetBuffer<const float>(fw_recurrent_to_cell_weights_), in Eval()
495 GetBuffer<const float>(fw_recurrent_to_output_weights_), in Eval()
497 GetBuffer<const float>(fw_cell_to_input_weights_), in Eval()
498 GetBuffer<const float>(fw_cell_to_forget_weights_), in Eval()
499 GetBuffer<const float>(fw_cell_to_output_weights_), auxInput, in Eval()
504 GetBuffer<const float>(fw_input_gate_bias_), in Eval()
505 GetBuffer<const float>(fw_forget_gate_bias_), in Eval()
506 GetBuffer<const float>(fw_cell_bias_), in Eval()
507 GetBuffer<const float>(fw_output_gate_bias_), in Eval()
508 GetBuffer<const float>(fw_projection_weights_), in Eval()
509 GetBuffer<const float>(fw_projection_bias_), in Eval()
510 GetBuffer<const float>(fw_activation_state_), in Eval()
511 GetBuffer<const float>(fw_cell_state_), in Eval()
517 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
525 bw_output_activation_state_buffer = GetBuffer<float>(bw_output_activation_state_); in Eval()
526 bw_output_cell_state_buffer = GetBuffer<float>(bw_output_cell_state_); in Eval()
539 GetBuffer<const float>(bw_input_to_input_weights_), in Eval()
540 GetBuffer<const float>(bw_input_to_forget_weights_), in Eval()
541 GetBuffer<const float>(bw_input_to_cell_weights_), in Eval()
542 GetBuffer<const float>(bw_input_to_output_weights_), in Eval()
544 GetBuffer<const float>(bw_recurrent_to_input_weights_), in Eval()
545 GetBuffer<const float>(bw_recurrent_to_forget_weights_), in Eval()
546 GetBuffer<const float>(bw_recurrent_to_cell_weights_), in Eval()
547 GetBuffer<const float>(bw_recurrent_to_output_weights_), in Eval()
549 GetBuffer<const float>(bw_cell_to_input_weights_), in Eval()
550 GetBuffer<const float>(bw_cell_to_forget_weights_), in Eval()
551 GetBuffer<const float>(bw_cell_to_output_weights_), auxInput, in Eval()
556 GetBuffer<const float>(bw_input_gate_bias_), in Eval()
557 GetBuffer<const float>(bw_forget_gate_bias_), in Eval()
558 GetBuffer<const float>(bw_cell_bias_), in Eval()
559 GetBuffer<const float>(bw_output_gate_bias_), in Eval()
560 GetBuffer<const float>(bw_projection_weights_), in Eval()
561 GetBuffer<const float>(bw_projection_bias_), in Eval()
562 GetBuffer<const float>(bw_activation_state_), in Eval()
563 GetBuffer<const float>(bw_cell_state_), in Eval()
569 params_.merge_outputs ? GetBuffer<float>(fw_output_) + n_fw_output_elements in Eval()
570 : GetBuffer<float>(bw_output_), in Eval()
574 mergeThirdDimension(GetBuffer<float>(fw_output_), fw_output_dims, in Eval()
575 GetBuffer<float>(fw_output_) + n_fw_output_elements, in Eval()
578 GetBuffer<float>(fw_output_)); in Eval()
582 const _Float16* bwInput = GetBuffer<const _Float16>(input_); in Eval()
586 bwInput = GetBuffer<const _Float16>(aux_input_); in Eval()
597 GetBuffer<_Float16>(fw_output_activation_state_); in Eval()
598 fw_output_cell_state_buffer = GetBuffer<_Float16>(fw_output_cell_state_); in Eval()
610 params_, GetBuffer<const _Float16>(input_), input_->shape(), in Eval()
612 GetBuffer<const _Float16>(fw_input_to_forget_weights_), in Eval()
613 GetBuffer<const _Float16>(fw_input_to_cell_weights_), in Eval()
614 GetBuffer<const _Float16>(fw_input_to_output_weights_), in Eval()
617 GetBuffer<const _Float16>(fw_recurrent_to_forget_weights_), in Eval()
618 GetBuffer<const _Float16>(fw_recurrent_to_cell_weights_), in Eval()
619 GetBuffer<const _Float16>(fw_recurrent_to_output_weights_), in Eval()
629 GetBuffer<const _Float16>(fw_forget_gate_bias_), in Eval()
630 GetBuffer<const _Float16>(fw_cell_bias_), in Eval()
631 GetBuffer<const _Float16>(fw_output_gate_bias_), in Eval()
634 GetBuffer<const _Float16>(fw_activation_state_), in Eval()
635 GetBuffer<const _Float16>(fw_cell_state_), in Eval()
641 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
650 GetBuffer<_Float16>(bw_output_activation_state_); in Eval()
651 bw_output_cell_state_buffer = GetBuffer<_Float16>(bw_output_cell_state_); in Eval()
665 GetBuffer<const _Float16>(bw_input_to_forget_weights_), in Eval()
666 GetBuffer<const _Float16>(bw_input_to_cell_weights_), in Eval()
667 GetBuffer<const _Float16>(bw_input_to_output_weights_), in Eval()
670 GetBuffer<const _Float16>(bw_recurrent_to_forget_weights_), in Eval()
671 GetBuffer<const _Float16>(bw_recurrent_to_cell_weights_), in Eval()
672 GetBuffer<const _Float16>(bw_recurrent_to_output_weights_), in Eval()
682 GetBuffer<const _Float16>(bw_forget_gate_bias_), in Eval()
683 GetBuffer<const _Float16>(bw_cell_bias_), in Eval()
684 GetBuffer<const _Float16>(bw_output_gate_bias_), in Eval()
687 GetBuffer<const _Float16>(bw_activation_state_), in Eval()
688 GetBuffer<const _Float16>(bw_cell_state_), in Eval()
694 params_.merge_outputs ? GetBuffer<_Float16>(fw_output_) + n_fw_output_elements in Eval()
695 : GetBuffer<_Float16>(bw_output_), in Eval()
699 mergeThirdDimension(GetBuffer<_Float16>(fw_output_), fw_output_dims, in Eval()
700 GetBuffer<_Float16>(fw_output_) + n_fw_output_elements, in Eval()
703 GetBuffer<_Float16>(fw_output_)); in Eval()