| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | reverse_sequence_gpu_kernel.cc | 23 ReverseSequence,
27 ReverseSequence,
32 ReverseSequence,
36 ReverseSequence,
41 ReverseSequence,
45 ReverseSequence,
50 ReverseSequence,
54 ReverseSequence,
59 ReverseSequence,
63 ReverseSequence,
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | reverse_sequence.h | 30 class MS_CORE_API ReverseSequence : public PrimitiveC {
33 ReverseSequence() : PrimitiveC(kNameReverseSequence) { InitIOName({"x", "seq_lengths"}, {"y"}); } in ReverseSequence() function
35 ~ReverseSequence() = default;
36 MS_DECLARE_PARENT(ReverseSequence, PrimitiveC);
54 using PrimReverseSequence = std::shared_ptr<ReverseSequence>;
|
| D | reverse_sequence.cc | 26 void ReverseSequence::Init(const int64_t seq_dim, const int64_t batch_dim) { in Init()
30 void ReverseSequence::set_seq_dim(const int64_t seq_dim) { (void)this->AddAttr(kSeqDim, MakeValue(s… in set_seq_dim()
31 void ReverseSequence::set_batch_dim(const int64_t batch_dim) { (void)this->AddAttr(kBatchDim, MakeV… in set_batch_dim()
33 int64_t ReverseSequence::get_seq_dim() const { return GetValue<int64_t>(GetAttr(kSeqDim)); } in get_seq_dim()
34 int64_t ReverseSequence::get_batch_dim() const { in get_batch_dim()
70 REGISTER_PRIMITIVE_C(kNameReverseSequence, ReverseSequence);
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | array_ops_declare.cc | 93 INPUT_MAP(ReverseSequence) = {{1, INPUT_DESC(x)}, {2, INPUT_DESC(seq_lengths)}};
94 ATTR_MAP(ReverseSequence) = {{"seq_dim", ATTR_DESC(seq_dim, AnyTraits<int64_t>())},
96 OUTPUT_MAP(ReverseSequence) = {{0, OUTPUT_DESC(y)}};
97 REG_ADPT_DESC(ReverseSequence, kNameReverseSequence, ADPT_DESC(ReverseSequence))
|
| D | array_ops_declare.h | 58 DECLARE_OP_ADAPTER(ReverseSequence)
59 DECLARE_OP_USE_OUTPUT(ReverseSequence)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | reverse_sequence_fp32.h | 28 void ReverseSequence(const float *input0, const void *input1, float *output, ReverseSequenceParamet…
|
| D | reverse_sequence_fp32.c | 19 void ReverseSequence(const float *input0, const void *input1, float *output, ReverseSequenceParamet… in ReverseSequence() function
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tflite/ |
| D | tflite_reverse_sequence_parser.cc | 32 auto prim = std::make_unique<ops::ReverseSequence>(); in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_reverse_sequence_parser.cc | 29 auto prim = std::make_unique<ops::ReverseSequence>(); in Parse()
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_reverse_sequence.py | 29 self.reverse_sequence = P.ReverseSequence(seq_dim=seq_dim, batch_dim=batch_dim)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | reverse_sequence_impl.cu | 56 __global__ void ReverseSequence(const size_t size, const T *input, const S *seq_len, const int64_t … in ReverseSequence() function
91 ReverseSequence<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>( in CalReverseSequence()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | reverse_sequence_fp32.cc | 104 ReverseSequence(input0, input1, output, param); in Run()
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_reverse_sequence.py | 14 self.reverse_sequence = P.ReverseSequence(
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | model_v0.fbs | 141 ReverseSequence,
|
| D | ops.fbs | 141 ReverseSequence,
842 table ReverseSequence {
|
| D | ops_v0.fbs | 817 table ReverseSequence {
|
| D | model_generated.h | 827 const mindspore::schema::ReverseSequence *value_as_ReverseSequence() const { in value_as_ReverseSequence()
828 …PrimitiveType_ReverseSequence ? static_cast<const mindspore::schema::ReverseSequence *>(value()) :… in value_as_ReverseSequence()
1556 …mplate<> inline const mindspore::schema::ReverseSequence *Primitive::value_as<mindspore::schema::R…
|
| D | model_v0_generated.h | 1041 template<> struct PrimitiveTypeTraits<mindspore::schema::v0::ReverseSequence> {
2145 const mindspore::schema::v0::ReverseSequence *value_as_ReverseSequence() const {
2146 …itiveType_ReverseSequence ? static_cast<const mindspore::schema::v0::ReverseSequence *>(value()) :…
2849 …<> inline const mindspore::schema::v0::ReverseSequence *Primitive::value_as<mindspore::schema::v0:…
4087 auto ptr = reinterpret_cast<const mindspore::schema::v0::ReverseSequence *>(obj);
|
| /third_party/mindspore/mindspore/lite/src/ops/ |
| D | ops_func_declare.h | 385 FUNC_MSOP2SCHEMAOP_DECLARE(ReverseSequence)
|
| D | ops_def.cc | 140 OP_TYPE(ReverseSequence)
842 OP_SCHEMA_DEF(ReverseSequence)
845 OP_SCHEMA_DEF_END(ReverseSequence)
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 35 … BatchToSpace, SpaceToBatchND, BatchToSpaceND, BroadcastTo, InplaceUpdate, ReverseSequence,
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | lstm.py | 164 self.reverse_seq = P.ReverseSequence(batch_dim=1, seq_dim=0)
|
| D | rnns.py | 235 self.reverse_sequence = P.ReverseSequence(0, 1)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 1046 @bprop_getters.register(P.ReverseSequence)
1049 reverse_sequence_grad = P.ReverseSequence(batch_dim=self.batch_dim_, seq_dim=self.seq_dim_)
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/ |
| D | common_infer.c | 474 REG_INFER(ReverseSequence, PrimType_ReverseSequence, CommonInferShape)
|