| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | reverse_sequence.cc | 47 void ReverseSequence::Init(const int64_t seq_dim, const int64_t batch_dim) { in Init()
52 void ReverseSequence::set_batch_dim(const int64_t batch_dim) { in set_batch_dim()
81 auto batch_dim = GetValue<int64_t>(batch_dim_ptr); in ReverseSequenceInferShape() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_auto_parallel_rec_batch_gather.py | 29 def __init__(self, param_size, mul_size1, mul_size2, batch_dim=0, axis=-1, strategy=None): argument
50 def __init__(self, param_size, mul_size1, mul_size2, batch_dim=0, axis=-1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/ |
| D | gather_fusion.cc | 29 ValuePtr batch_dim = MakeValue(static_cast<int64_t>(0)); in GatherFusionProcess() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | reverse_sequence_impl.cu | 61 … void ReverseSequence(const size_t size, const T *input, const S *seq_len, const int64_t batch_dim, in ReverseSequence()
99 …_t CalReverseSequence(const size_t size, const T *input, const S *seq_len, const int64_t batch_dim, in CalReverseSequence()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | reverse_sequence.cc | 42 uint32_t CalReverseSequence(size_t seq_dim, size_t batch_dim, const std::vector<void *> &ioAddrs, in CalReverseSequence()
112 AttrValue *batch_dim = ctx.GetAttr("batch_dim"); in GetInputAndCheck() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_reverse_sequence.py | 28 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_reverse_sequence.py | 13 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_reverse_sequence.py | 13 def __init__(self, seq_dim, batch_dim): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | combined_non_max_suppression_proto.cc | 159 int64_t batch_dim = boxes.GetDim(0); in IMPLEMT_INFERFUNC() local
|
| D | array_ops_proto.cc | 434 int64_t batch_dim = input_shape.GetDim(0); in CUST_IMPLEMT_INFERFUNC() local
635 int64_t batch_dim = op.GetInputDesc(0).GetShape().GetDim(0); in CUST_IMPLEMT_INFERFUNC() local
|
| D | image_ops_proto.cc | 101 int64_t batch_dim; in IMPLEMT_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | image_ops_shape_fns.cc | 77 graphStatus SetOutputToSizedImage(Operator &op, const int64_t batch_dim, const std::string &size_in… in SetOutputToSizedImage()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/ |
| D | config_file_parser.cc | 136 size_t batch_dim = 0; in CombineDynamicBatchList() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_vmap/ |
| D | vmap_array_ops.py | 1384 def _update_dim(dim, x_rank, batch_dim): argument
|
| /third_party/mesa3d/include/tensorflow/lite/core/c/ |
| D | builtin_op_data.h | 476 int batch_dim; member
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/mindir/src/ |
| D | mindir_1.cc | 1096 PrimitivePtr MindIR_ReverseSequence_CreatePrimitive(int64_t seq_dim, int64_t batch_dim) { in MindIR_ReverseSequence_CreatePrimitive()
1151 void MindIR_ReverseSequence_SetBatchDim(PrimitivePtr *primitive, int64_t batch_dim) { in MindIR_ReverseSequence_SetBatchDim()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | array_ops.py | 3994 def __init__(self, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/ |
| D | array_func.py | 1572 def reverse_sequence(x, seq_lengths, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 646 def reverse_sequence(x, seq_lengths, seq_dim, batch_dim=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | tensor.py | 1879 def reverse_sequence(self, seq_lengths, seq_dim=0, batch_dim=0): argument
|