Searched refs:seq_dim_ (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/ |
| D | reverse_sequence_gpu_kernel.h | 38 seq_dim_(0), in ReverseSequenceGpuFwdKernel()
63 …CalReverseSequence(input_size_, input, seq_len, batch_dim_, seq_dim_, cur_pos_arr, input_shape_ptr, in Launch()
70 seq_dim_ = GetAttr<int64_t>(kernel_node, "seq_dim"); in Init()
122 int64_t seq_dim_; variable
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | array_ops.py | 5627 self.seq_dim_ = seq_dim
5632 validator.check("seq_dim", self.seq_dim_, "x rank", len(x), Rel.LE, self.name)
5634 validator.check("batch_dim", self.batch_dim_, "seq_dim", self.seq_dim_, Rel.NE, self.name)
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 1049 reverse_sequence_grad = P.ReverseSequence(batch_dim=self.batch_dim_, seq_dim=self.seq_dim_)
|