Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: bcaceb57a1
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'bcaceb57a1'
${ noResults }
mindspore/mindspore/lite/nnacl/reverse_sequence.h

49 lines
1.4 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2020 Huawei Technologies Co., Ltd

  3.  *

  4.  * Licensed under the Apache License, Version 2.0 (the "License");

  5.  * you may not use this file except in compliance with the License.

  6.  * You may obtain a copy of the License at

  7.  *

  8.  * http://www.apache.org/licenses/LICENSE-2.0

  9.  *

  10.  * Unless required by applicable law or agreed to in writing, software

  11.  * distributed under the License is distributed on an "AS IS" BASIS,

  12.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  13.  * See the License for the specific language governing permissions and

  14.  * limitations under the License.

  15.  */

  16. 

  17. #ifndef MINDSPORE_LITE_NNACL_REVERSE_SEQUENCE_H_

  18. #define MINDSPORE_LITE_NNACL_REVERSE_SEQUENCE_H_

  19. 

  20. #include "nnacl/op_base.h"

  21. 

  22. typedef struct ReverseSequenceParameter {

  23.   OpParameter op_parameter_;

  24.   int ndim_;

  25.   int input_shape0_[5];

  26.   int output_shape_[5];

  27.   int input_stride_[5];

  28.   int output_stride_[5];

  29.   int seq_axis_;

  30.   int batch_axis_;

  31.   int outer_count_;

  32.   int outer_stride_;

  33.   int inner_count_;

  34.   int inner_stride_;

  35.   int copy_byte_size_;

  36.   int total_data_size_;

  37.   bool is_seq_length_int32_;

  38. } ReverseSequenceParameter;

  39. 

  40. #ifdef __cplusplus

  41. extern "C" {

  42. #endif

  43. void ReverseSequence(float *input0, void *input1, float *output, ReverseSequenceParameter *para);

  44. #ifdef __cplusplus

  45. }

  46. #endif

  47. 

  48. #endif  // MINDSPORE_LITE_NNACL_REVERSE_SEQUENCE_H_