Home
last modified time | relevance | path

Searched refs:TensorOperation (Results 1 – 25 of 180) sorted by relevance

12345678

/third_party/mindspore/tests/ut/cpp/dataset/
Dir_vision_test.cc78 std::shared_ptr<TensorOperation> auto_contrast1(new vision::AutoContrastOperation(-1.0, {})); in TEST_F()
83 …std::shared_ptr<TensorOperation> auto_contrast2(new vision::AutoContrastOperation(110.0, {10, 20})… in TEST_F()
94 std::shared_ptr<TensorOperation> center_crop1(new vision::CenterCropOperation({-32, 32})); in TEST_F()
99 std::shared_ptr<TensorOperation> center_crop2(new vision::CenterCropOperation({32, -32})); in TEST_F()
104 std::shared_ptr<TensorOperation> center_crop3(new vision::CenterCropOperation({0, 32})); in TEST_F()
109 std::shared_ptr<TensorOperation> center_crop4(new vision::CenterCropOperation({10, 20, 30})); in TEST_F()
120 std::shared_ptr<TensorOperation> crop1(new vision::CropOperation({0, 0}, {32, -32})); in TEST_F()
125 std::shared_ptr<TensorOperation> crop2(new vision::CropOperation({0, 0}, {-32, -32})); in TEST_F()
130 std::shared_ptr<TensorOperation> crop3(new vision::CropOperation({0, 0}, {0, 32})); in TEST_F()
135 std::shared_ptr<TensorOperation> crop4(new vision::CropOperation({-1, 0}, {32, 32})); in TEST_F()
[all …]
Dir_vision_random_test.cc50 std::shared_ptr<TensorOperation> random_color1(new vision::RandomColorOperation(1.0, 0.1)); in TEST_F()
55 std::shared_ptr<TensorOperation> random_color2(new vision::RandomColorOperation(-0.5, 0.5)); in TEST_F()
64 std::shared_ptr<TensorOperation> random_color_adjust1( in TEST_F()
70 std::shared_ptr<TensorOperation> random_color_adjust2( in TEST_F()
76 std::shared_ptr<TensorOperation> random_color_adjust3( in TEST_F()
82 std::shared_ptr<TensorOperation> random_color_adjust4( in TEST_F()
88 std::shared_ptr<TensorOperation> random_color_adjust5( in TEST_F()
94 std::shared_ptr<TensorOperation> random_color_adjust6( in TEST_F()
104 …std::shared_ptr<TensorOperation> random_horizontal_flip1(new vision::RandomHorizontalFlipOperation… in TEST_F()
109 …std::shared_ptr<TensorOperation> random_horizontal_flip2(new vision::RandomHorizontalFlipOperation… in TEST_F()
[all …]
Ddeserialize_test.cc60 std::shared_ptr<TensorOperation> operation0 = std::make_shared<vision::EqualizeOperation>(); in TEST_F()
61 …std::shared_ptr<TensorOperation> operation1 = std::make_shared<vision::BoundingBoxAugmentOperation… in TEST_F()
62 std::shared_ptr<TensorOperation> operation2 = std::make_shared<vision::HorizontalFlipOperation>(); in TEST_F()
63 std::shared_ptr<TensorOperation> operation3 = std::make_shared<vision::HwcToChwOperation>(); in TEST_F()
64 std::shared_ptr<TensorOperation> operation4 = std::make_shared<vision::RgbaToBgrOperation>(); in TEST_F()
65 std::shared_ptr<TensorOperation> operation5 = std::make_shared<vision::RgbaToRgbOperation>(); in TEST_F()
66 std::shared_ptr<TensorOperation> operation6 = std::make_shared<vision::SwapRedBlueOperation>(); in TEST_F()
67 std::vector<std::vector<std::pair<std::shared_ptr<TensorOperation>, double>>> policy; in TEST_F()
68 std::vector<std::pair<std::shared_ptr<TensorOperation>, double>> sub_policy; in TEST_F()
71 …std::shared_ptr<TensorOperation> operation7 = std::make_shared<vision::RandomSelectSubpolicyOperat… in TEST_F()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dvision.h35 class TensorOperation; variable
56 std::shared_ptr<TensorOperation> Parse() override;
77 std::shared_ptr<TensorOperation> Parse() override;
109 std::shared_ptr<TensorOperation> Parse() override;
129 std::shared_ptr<TensorOperation> Parse() override;
152 std::shared_ptr<TensorOperation> Parse() override;
173 std::shared_ptr<TensorOperation> Parse() override;
192 std::shared_ptr<TensorOperation> Parse() override;
207 std::shared_ptr<TensorOperation> Parse() override;
222 std::shared_ptr<TensorOperation> Parse() override;
[all …]
Daudio.h34 class TensorOperation; variable
50 std::shared_ptr<TensorOperation> Parse() override;
69 std::shared_ptr<TensorOperation> Parse() override;
90 std::shared_ptr<TensorOperation> Parse() override;
116 std::shared_ptr<TensorOperation> Parse() override;
140 std::shared_ptr<TensorOperation> Parse() override;
162 std::shared_ptr<TensorOperation> Parse() override;
185 std::shared_ptr<TensorOperation> Parse() override;
210 std::shared_ptr<TensorOperation> Parse() override;
231 std::shared_ptr<TensorOperation> Parse() override;
[all …]
Dvision_lite.h65 std::shared_ptr<TensorOperation> Parse() override;
87 std::shared_ptr<TensorOperation> Parse() override;
89 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
112 std::shared_ptr<TensorOperation> Parse() override;
132 std::shared_ptr<TensorOperation> Parse() override;
134 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
156 std::shared_ptr<TensorOperation> Parse() override;
179 std::shared_ptr<TensorOperation> Parse() override;
181 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
222 std::shared_ptr<TensorOperation> Parse() override;
[all …]
Dtransforms.h33 class TensorOperation; variable
71 virtual std::shared_ptr<TensorOperation> Parse() = 0;
76 virtual std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) { return nullptr; } in Parse()
149 std::shared_ptr<TensorOperation> Parse() override;
171 std::shared_ptr<TensorOperation> Parse() override;
191 std::shared_ptr<TensorOperation> Parse() override;
210 std::shared_ptr<TensorOperation> Parse() override;
236 std::shared_ptr<TensorOperation> Parse() override;
256 std::shared_ptr<TensorOperation> Parse() override;
279 std::shared_ptr<TensorOperation> Parse() override;
[all …]
Dtext.h36 class TensorOperation; variable
66 std::shared_ptr<TensorOperation> Parse() override;
127 std::shared_ptr<TensorOperation> Parse() override;
146 std::shared_ptr<TensorOperation> Parse() override;
211 std::shared_ptr<TensorOperation> Parse() override;
267 std::shared_ptr<TensorOperation> Parse() override;
308 std::shared_ptr<TensorOperation> Parse() override;
336 std::shared_ptr<TensorOperation> Parse() override;
367 std::shared_ptr<TensorOperation> Parse() override;
395 std::shared_ptr<TensorOperation> Parse() override;
[all …]
Dvision_ascend.h51 std::shared_ptr<TensorOperation> Parse() override;
53 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
75 std::shared_ptr<TensorOperation> Parse() override;
77 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
97 std::shared_ptr<TensorOperation> Parse() override;
99 std::shared_ptr<TensorOperation> Parse(const MapTargetDevice &env) override;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/data/
Dtransforms_ir.h51 class ComposeOperation : public TensorOperation {
53 explicit ComposeOperation(const std::vector<std::shared_ptr<TensorOperation>> &transforms);
64 std::vector<std::shared_ptr<TensorOperation>> transforms_;
67 class ConcatenateOperation : public TensorOperation {
86 class DuplicateOperation : public TensorOperation {
99 class FillOperation : public TensorOperation {
113 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
119 class MaskOperation : public TensorOperation {
137 class OneHotOperation : public TensorOperation {
151 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/image/
Dbindings.cc74 …(void)py::class_<vision::AdjustGammaOperation, TensorOperation, std::shared_ptr<vision::AdjustGamm… in __anon682838b20102()
85 …(void)py::class_<vision::AutoContrastOperation, TensorOperation, std::shared_ptr<vision::AutoContr… in __anon682838b20302()
95 (void)py::class_<vision::BoundingBoxAugmentOperation, TensorOperation, in __anon682838b20502()
108 …(void)py::class_<vision::CenterCropOperation, TensorOperation, std::shared_ptr<vision::CenterCropO… in __anon682838b20702()
119 …(void)py::class_<vision::ConvertColorOperation, TensorOperation, std::shared_ptr<vision::ConvertCo… in __anon682838b20902()
129 … (void)py::class_<vision::CropOperation, TensorOperation, std::shared_ptr<vision::CropOperation>>( in __anon682838b20b02()
143 …(void)py::class_<vision::CutMixBatchOperation, TensorOperation, std::shared_ptr<vision::CutMixBatc… in __anon682838b20d02()
153 …(void)py::class_<vision::CutOutOperation, TensorOperation, std::shared_ptr<vision::CutOutOperation… in __anon682838b20f02()
164 …(void)py::class_<vision::DecodeOperation, TensorOperation, std::shared_ptr<vision::DecodeOperation… in __anon682838b21102()
180 …py::class_<vision::EqualizeOperation, TensorOperation, std::shared_ptr<vision::EqualizeOperation>>( in __anon682838b21402()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dtransforms.cc37 std::vector<std::shared_ptr<TensorOperation>> transforms_;
42 [](TensorTransform *const op) -> std::shared_ptr<TensorOperation> { in Compose()
49 [](std::shared_ptr<TensorTransform> op) -> std::shared_ptr<TensorOperation> { in Compose()
57 … [](TensorTransform &op) -> std::shared_ptr<TensorOperation> { return op.Parse(); }); in Compose()
60 std::shared_ptr<TensorOperation> Compose::Parse() { return std::make_shared<ComposeOperation>(data_… in Parse()
74 std::shared_ptr<TensorOperation> Concatenate::Parse() { in Parse()
97 std::shared_ptr<TensorOperation> Duplicate::Parse() { return std::make_shared<DuplicateOperation>()… in Parse()
107 std::shared_ptr<TensorOperation> Fill::Parse() { in Parse()
134 std::shared_ptr<TensorOperation> Mask::Parse() { in Parse()
159 std::shared_ptr<TensorOperation> OneHot::Parse() { return std::make_shared<OneHotOperation>(data_->… in Parse()
[all …]
Dvision.cc117 std::shared_ptr<TensorOperation> Affine::Parse() { in Parse()
132 std::shared_ptr<TensorOperation> AdjustGamma::Parse() { in Parse()
146 std::shared_ptr<TensorOperation> AutoContrast::Parse() { in Parse()
152 std::shared_ptr<TensorOperation> transform_;
173 std::shared_ptr<TensorOperation> BoundingBoxAugment::Parse() { in Parse()
186 std::shared_ptr<TensorOperation> CenterCrop::Parse() { return std::make_shared<CenterCropOperation>… in Parse()
188 std::shared_ptr<TensorOperation> CenterCrop::Parse(const MapTargetDevice &env) { in Parse()
213 std::shared_ptr<TensorOperation> ConvertColor::Parse() { in Parse()
229 std::shared_ptr<TensorOperation> Crop::Parse() { in Parse()
246 std::shared_ptr<TensorOperation> CutMixBatch::Parse() { in Parse()
[all …]
Daudio.cc58 std::shared_ptr<TensorOperation> AllpassBiquad::Parse() { in Parse()
75 std::shared_ptr<TensorOperation> AmplitudeToDB::Parse() { in Parse()
82 std::shared_ptr<TensorOperation> Angle::Parse() { return std::make_shared<AngleOperation>(); } in Parse()
96 std::shared_ptr<TensorOperation> BandBiquad::Parse() { in Parse()
113 std::shared_ptr<TensorOperation> BandpassBiquad::Parse() { in Parse()
130 std::shared_ptr<TensorOperation> BandrejectBiquad::Parse() { in Parse()
147 std::shared_ptr<TensorOperation> BassBiquad::Parse() { in Parse()
166 std::shared_ptr<TensorOperation> Biquad::Parse() { in Parse()
178 std::shared_ptr<TensorOperation> ComplexNorm::Parse() { return std::make_shared<ComplexNormOperatio… in Parse()
188 std::shared_ptr<TensorOperation> Contrast::Parse() { in Parse()
[all …]
Dtext.cc63 std::shared_ptr<TensorOperation> BasicTokenizer::Parse() { in Parse()
100 std::shared_ptr<TensorOperation> BertTokenizer::Parse() { in Parse()
109 std::shared_ptr<TensorOperation> CaseFold::Parse() { return std::make_shared<CaseFoldOperation>(); } in Parse()
131 std::shared_ptr<TensorOperation> JiebaTokenizer::Parse() { in Parse()
238 std::shared_ptr<TensorOperation> Lookup::Parse() { in Parse()
260 std::shared_ptr<TensorOperation> Ngram::Parse() { in Parse()
273 std::shared_ptr<TensorOperation> NormalizeUTF8::Parse() { in Parse()
289 std::shared_ptr<TensorOperation> RegexReplace::Parse() { in Parse()
308 std::shared_ptr<TensorOperation> RegexTokenizer::Parse() { in Parse()
332 std::shared_ptr<TensorOperation> SentencePieceTokenizer::Parse() { in Parse()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/
Dbindings.cc29 Status PyListToTensorOperations(const py::list &py_ops, std::vector<std::shared_ptr<TensorOperation… in PyListToTensorOperations()
37 } else if (py::isinstance<TensorOperation>(op)) { in PyListToTensorOperations()
38 ops->emplace_back(op.cast<std::shared_ptr<TensorOperation>>()); in PyListToTensorOperations()
50 PYBIND_REGISTER(TensorOperation, 0, ([](const py::module *m) { in __anon360f34600102()
51 … (void)py::class_<TensorOperation, std::shared_ptr<TensorOperation>>(*m, "TensorOperation"); in __anon360f34600102()
57 …(void)py::class_<transforms::PluginOperation, TensorOperation, std::shared_ptr<transforms::PluginO… in __anon360f34600202()
68 …(void)py::class_<transforms::ComposeOperation, TensorOperation, std::shared_ptr<transforms::Compos… in __anon360f34600402()
71 std::vector<std::shared_ptr<TensorOperation>> t_ops; in __anon360f34600402()
82 …py::class_<transforms::ConcatenateOperation, TensorOperation, std::shared_ptr<transforms::Concaten… in __anon360f34600602()
93 …(void)py::class_<transforms::DuplicateOperation, TensorOperation, std::shared_ptr<transforms::Dupl… in __anon360f34600802()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dascend_vision_ir.h48 class DvppCropJpegOperation : public TensorOperation {
62 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
68 class DvppDecodeResizeOperation : public TensorOperation {
82 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
88 class DvppDecodeResizeCropOperation : public TensorOperation {
102 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
109 class DvppDecodeJpegOperation : public TensorOperation {
120 class DvppDecodePngOperation : public TensorOperation {
131 class DvppNormalizeOperation : public TensorOperation {
145 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
[all …]
Drandom_select_subpolicy_ir.cc33 const std::vector<std::vector<std::pair<std::shared_ptr<TensorOperation>, double>>> &policy) in RandomSelectSubpolicyOperation() argument
34 : TensorOperation(true), policy_(policy) {} in RandomSelectSubpolicyOperation()
106 std::shared_ptr<TensorOperation> *operation) { in from_json()
109 std::vector<std::vector<std::pair<std::shared_ptr<TensorOperation>, double>>> policy; in from_json()
110 std::vector<std::pair<std::shared_ptr<TensorOperation>, double>> policy_items; in from_json()
115 std::vector<std::shared_ptr<TensorOperation>> operations; in from_json()
116 std::pair<std::shared_ptr<TensorOperation>, double> policy_pair; in from_json()
117 std::shared_ptr<TensorOperation> operation; in from_json()
Dbounding_box_augment_ir.h38 class BoundingBoxAugmentOperation : public TensorOperation {
40 …explicit BoundingBoxAugmentOperation(const std::shared_ptr<TensorOperation> &transform, float rati…
52 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
55 std::shared_ptr<TensorOperation> transform_;
Duniform_aug_ir.h38 class UniformAugOperation : public TensorOperation {
40 …explicit UniformAugOperation(const std::vector<std::shared_ptr<TensorOperation>> &transforms, int3…
52 static Status from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *operation);
55 std::vector<std::shared_ptr<TensorOperation>> transforms_;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/audio/kernels/ir/
Dbindings.cc52 …(void)py::class_<audio::AllpassBiquadOperation, TensorOperation, std::shared_ptr<audio::AllpassBiq… in __anonb7c3dee10102()
63 …(void)py::class_<audio::AmplitudeToDBOperation, TensorOperation, std::shared_ptr<audio::AmplitudeT… in __anonb7c3dee10302()
80 … (void)py::class_<audio::AngleOperation, TensorOperation, std::shared_ptr<audio::AngleOperation>>( in __anonb7c3dee10602()
91 …(void)py::class_<audio::BandBiquadOperation, TensorOperation, std::shared_ptr<audio::BandBiquadOpe… in __anonb7c3dee10802()
102 …(void)py::class_<audio::BandpassBiquadOperation, TensorOperation, std::shared_ptr<audio::BandpassB… in __anonb7c3dee10a02()
113 (void)py::class_<audio::BandrejectBiquadOperation, TensorOperation, in __anonb7c3dee10c02()
125 …(void)py::class_<audio::BassBiquadOperation, TensorOperation, std::shared_ptr<audio::BassBiquadOpe… in __anonb7c3dee10e02()
135 …(void)py::class_<audio::BiquadOperation, TensorOperation, std::shared_ptr<audio::BiquadOperation>>( in __anonb7c3dee11002()
146 …(void)py::class_<audio::ComplexNormOperation, TensorOperation, std::shared_ptr<audio::ComplexNormO… in __anonb7c3dee11202()
157 … py::class_<audio::ContrastOperation, TensorOperation, std::shared_ptr<audio::ContrastOperation>>( in __anonb7c3dee11402()
[all …]
/third_party/mindspore/mindspore/dataset/transforms/
Dc_transforms.py29 class TensorOperation: class
57 class OneHot(TensorOperation):
82 class Fill(TensorOperation):
112 class TypeCast(TensorOperation):
168 class Slice(TensorOperation):
243 class Mask(TensorOperation):
282 class PadEnd(TensorOperation):
318 class Concatenate(TensorOperation):
349 class Duplicate(TensorOperation):
376 class Unique(TensorOperation):
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/ir/kernels/
Dtext_ir.h57 class BasicTokenizerOperation : public TensorOperation {
78 class BertTokenizerOperation : public TensorOperation {
105 class CaseFoldOperation : public TensorOperation {
119 class JiebaTokenizerOperation : public TensorOperation {
142 class LookupOperation : public TensorOperation {
164 class NgramOperation : public TensorOperation {
185 class NormalizeUTF8Operation : public TensorOperation {
201 class RegexReplaceOperation : public TensorOperation {
219 class RegexTokenizerOperation : public TensorOperation {
238 class SentencePieceTokenizerOperation : public TensorOperation {
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/text/kernels/ir/
Dbindings.cc30 …(void)py::class_<text::BasicTokenizerOperation, TensorOperation, std::shared_ptr<text::BasicTokeni… in __anonf0f064540102()
43 …(void)py::class_<text::BertTokenizerOperation, TensorOperation, std::shared_ptr<text::BertTokenize… in __anonf0f064540302()
58 …(void)py::class_<text::CaseFoldOperation, TensorOperation, std::shared_ptr<text::CaseFoldOperation… in __anonf0f064540502()
69 …(void)py::class_<text::NormalizeUTF8Operation, TensorOperation, std::shared_ptr<text::NormalizeUTF… in __anonf0f064540702()
80 …(void)py::class_<text::RegexReplaceOperation, TensorOperation, std::shared_ptr<text::RegexReplaceO… in __anonf0f064540902()
91 …(void)py::class_<text::RegexTokenizerOperation, TensorOperation, std::shared_ptr<text::RegexTokeni… in __anonf0f064540b02()
103 (void)py::class_<text::UnicodeScriptTokenizerOperation, TensorOperation, in __anonf0f064540d02()
115 (void)py::class_<text::WhitespaceTokenizerOperation, TensorOperation, in __anonf0f064540f02()
138 …(void)py::class_<text::JiebaTokenizerOperation, TensorOperation, std::shared_ptr<text::JiebaTokeni… in __anonf0f064541202()
152 … (void)py::class_<text::LookupOperation, TensorOperation, std::shared_ptr<text::LookupOperation>>( in __anonf0f064541502()
[all …]
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/
Dtensor_operation.h29 class TensorOperation : public std::enable_shared_from_this<TensorOperation> {
32 TensorOperation() : random_op_(false) {} in TensorOperation() function
35 explicit TensorOperation(bool random) : random_op_(random) {} in TensorOperation() function
38 ~TensorOperation() = default;

12345678