| /third_party/mindspore/tests/ut/cpp/dataset/ |
| D | slice_op_test.cc | 38 std::unique_ptr<SliceOp> op(new SliceOp(SliceOption(slice))); in TEST_F()
93 std::vector<SliceOption> slices_ = {SliceOption(slice1_), SliceOption(slice2_)}; in TEST_F()
122 …std::vector<SliceOption> slices_ = {SliceOption(slice1_), SliceOption(slice2_), SliceOption(slice3… in TEST_F()
151 …std::vector<SliceOption> slices_ = {SliceOption(slice1_), SliceOption(slice2_), SliceOption(slice3… in TEST_F()
203 std::unique_ptr<SliceOp> op(new SliceOp(SliceOption(true))); in TEST_F()
252 std::vector<SliceOption> indices; in TEST_F()
255 indices.emplace_back(SliceOption(index1)); in TEST_F()
256 indices.emplace_back(SliceOption(index2)); in TEST_F()
313 std::vector<SliceOption> slice_options = {SliceOption(indices), SliceOption(slice)}; in TEST_F()
342 std::vector<SliceOption> slice_options = {SliceOption(slice1_), SliceOption(slice2_)}; in TEST_F()
[all …]
|
| D | c_api_transforms_test.cc | 2009 SliceOption slice_option = SliceOption(Slice(0, 3, 2)); in TEST_F()
2073 SliceOption slice_option = SliceOption(true); in TEST_F()
2139 SliceOption slice_option = SliceOption(indices); in TEST_F()
2198 SliceOption slice_option = SliceOption(indices); in TEST_F()
2252 SliceOption slice_option1 = SliceOption(indices1); in TEST_F()
2256 SliceOption slice_option2 = SliceOption(indices2); in TEST_F()
2309 SliceOption slice_option = SliceOption(indices); in TEST_F()
2357 SliceOption slice_option = SliceOption(false); in TEST_F()
|
| D | tensor_test.cc | 428 std::vector<SliceOption> slice_options = {SliceOption(x)}; in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/ |
| D | slice_op.h | 34 explicit SliceOp(std::vector<SliceOption> slice_input) : slice_options_(slice_input) {} in SliceOp()
35 explicit SliceOp(SliceOption slice_option) { slice_options_.push_back(slice_option); } in SliceOp()
37 explicit SliceOp(Slice slice) { slice_options_.push_back(SliceOption(slice)); } in SliceOp()
38 explicit SliceOp(bool all) { slice_options_.push_back(SliceOption(all)); } in SliceOp()
39 explicit SliceOp(std::vector<dsize_t> indices) { slice_options_.push_back(SliceOption(indices)); } in SliceOp()
48 std::vector<SliceOption> slice_options_ = {};
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | transforms.h | 107 class SliceOption {
110 explicit SliceOption(bool all) : all_(all) {} in SliceOption() function
112 explicit SliceOption(std::vector<dsize_t> indices) : indices_(indices) {} in SliceOption() function
114 explicit SliceOption(Slice slice) : slice_(slice) {} in SliceOption() function
115 SliceOption(SliceOption const &slice) = default;
117 ~SliceOption() = default;
348 explicit Slice(const std::vector<SliceOption> &slice_input);
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
| D | audio_utils.cc | 284 input->Slice(&spec_time0, std::vector<SliceOption>({SliceOption(true), SliceOption(true), in TimeStretch()
285 … SliceOption(std::vector<dsize_t>{0}), SliceOption(true)}))); in TimeStretch()
293 …RETURN_IF_NOT_OK(input->Slice(&spec_0, std::vector<SliceOption>({SliceOption(true), SliceOption(tr… in TimeStretch()
294 … SliceOption(time_steps_0), SliceOption(true)}))); in TimeStretch()
296 …RETURN_IF_NOT_OK(input->Slice(&spec_1, std::vector<SliceOption>({SliceOption(true), SliceOption(tr… in TimeStretch()
297 … SliceOption(time_steps_1), SliceOption(true)}))); in TimeStretch()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/core/ |
| D | tensor_helpers.h | 39 …atorHelper(int8_t depth, std::vector<dsize_t> *numbers, const std::vector<SliceOption> &slice_list,
48 std::vector<std::vector<dsize_t>> IndexGenerator(const std::vector<SliceOption> &slice_list);
|
| D | tensor_helpers.cc | 24 const std::vector<mindspore::dataset::SliceOption> &slice_list, in IndexGeneratorHelper()
69 std::vector<std::vector<dsize_t>> IndexGenerator(const std::vector<mindspore::dataset::SliceOption>… in IndexGenerator()
|
| D | tensor.cc | 1010 Status Tensor::GetSliceOption(const SliceOption &slice_option, const int32_t &slice_index, in GetSliceOption()
1011 SliceOption *slice_option_ptr) { in GetSliceOption()
1033 *slice_option_ptr = SliceOption(slice_copy); in GetSliceOption()
1043 *slice_option_ptr = SliceOption(indices_copy); in GetSliceOption()
1048 Status Tensor::Slice(std::shared_ptr<Tensor> *out, const std::vector<SliceOption> slice_options_) { in Slice()
1050 std::vector<SliceOption> converted_slice_objects; in Slice()
1055 SliceOption slice_option = slice_options_[k]; in Slice()
1059 converted_slice_objects.push_back(SliceOption(slice)); in Slice()
1066 SliceOption slice_option_item(false); in Slice()
1075 converted_slice_objects.emplace_back(SliceOption(slice)); in Slice()
|
| D | tensor.h | 433 Status Slice(TensorPtr *out, const std::vector<mindspore::dataset::SliceOption> slice_options);
440 …Status GetSliceOption(const SliceOption &slice_option, const int32_t &slice_index, SliceOption *sl…
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/ |
| D | bindings.cc | 174 .def(py::init([](std::vector<SliceOption> slice_input) { in __anon360f34601602()
181 PYBIND_REGISTER(SliceOption, 0, ([](const py::module *m) { in __anon360f34601802()
182 (void)py::class_<SliceOption>(*m, "SliceOption") in __anon360f34601802()
201 return SliceOption(c_slice); in __anon360f34601802()
208 return SliceOption(indices); in __anon360f34601802()
211 .def(py::init<SliceOption>()); in __anon360f34601802()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/ |
| D | transforms.cc | 254 explicit Data(const std::vector<SliceOption> &slice_input) : slice_input_(slice_input) {} in Data()
255 std::vector<SliceOption> slice_input_;
258 Slice::Slice(const std::vector<SliceOption> &slice_input) : data_(std::make_shared<Data>(slice_inpu… in Slice()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/data/ |
| D | transforms_ir.h | 227 explicit SliceOperation(const std::vector<SliceOption> &slice_input);
238 std::vector<SliceOption> slice_input_;
|
| D | transforms_ir.cc | 254 SliceOperation::SliceOperation(const std::vector<SliceOption> &slice_input) : slice_input_(slice_in… in SliceOperation()
|
| /third_party/mindspore/mindspore/dataset/transforms/ |
| D | c_transforms.py | 142 class _SliceOption(cde.SliceOption):
|