Searched refs:fw_output_dims (Results 1 – 1 of 1) sorted by relevance
443 std::vector<uint32_t> fw_output_dims = input_->shape().dimensions; in Eval() local444 fw_output_dims[2] = n_fw_output; in Eval()445 std::vector<uint32_t> bw_output_dims = fw_output_dims; in Eval()447 const uint32_t n_fw_output_elements = fw_output_dims[0] * fw_output_dims[1] * fw_output_dims[2]; in Eval()449 fw_output_dims[0] * fw_output_dims[1] * (fw_output_dims[2] + bw_output_dims[2]); in Eval()574 mergeThirdDimension(GetBuffer<float>(fw_output_), fw_output_dims, in Eval()699 mergeThirdDimension(GetBuffer<_Float16>(fw_output_), fw_output_dims, in Eval()