Home
last modified time | relevance | path

Searched refs:TensorTransform (Results 1 – 25 of 42) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/include/dataset/
Dvision.h42 class AdjustGamma final : public TensorTransform {
64 class AutoContrast final : public TensorTransform {
86 class BoundingBoxAugment final : public TensorTransform {
91 explicit BoundingBoxAugment(TensorTransform *transform, float ratio = 0.3);
96 explicit BoundingBoxAugment(const std::shared_ptr<TensorTransform> &transform, float ratio = 0.3);
101 …explicit BoundingBoxAugment(const std::reference_wrapper<TensorTransform> transform, float ratio =…
117 class ConvertColor final : public TensorTransform {
138 class CutMixBatch final : public TensorTransform {
160 class CutOut final : public TensorTransform {
181 class Equalize final : public TensorTransform {
[all …]
Dtransforms.h51 class TensorTransform : public std::enable_shared_from_this<TensorTransform> {
63 TensorTransform() {} in TensorTransform() function
66 ~TensorTransform() = default;
131 class Compose final : public TensorTransform {
135 explicit Compose(const std::vector<TensorTransform *> &transforms);
138 explicit Compose(const std::vector<std::shared_ptr<TensorTransform>> &transforms);
141 explicit Compose(const std::vector<std::reference_wrapper<TensorTransform>> &transforms);
157 class Concatenate final : public TensorTransform {
180 class Duplicate final : public TensorTransform {
196 class Fill final : public TensorTransform {
[all …]
Daudio.h40 class Angle final : public TensorTransform {
54 class BandBiquad final : public TensorTransform {
77 class AllpassBiquad final : public TensorTransform {
99 class AmplitudeToDB final : public TensorTransform {
124 class BandpassBiquad final : public TensorTransform {
148 class BandrejectBiquad final : public TensorTransform {
170 class BassBiquad final : public TensorTransform {
193 class Biquad final : public TensorTransform {
219 class ComplexNorm final : public TensorTransform {
239 class Contrast final : public TensorTransform {
[all …]
Dvision_lite.h39 class Affine final : public TensorTransform {
73 class CenterCrop final : public TensorTransform {
97 class Crop final : public TensorTransform {
120 class Decode final : public TensorTransform {
142 class GaussianBlur final : public TensorTransform {
164 class Normalize final : public TensorTransform {
189 class RandomAffine final : public TensorTransform {
230 class Resize final : public TensorTransform {
260 class ResizePreserveAR final : public TensorTransform {
291 class RGB2BGR final : public TensorTransform {
[all …]
Dtext.h44 class BasicTokenizer final : public TensorTransform {
75 class BertTokenizer final : public TensorTransform {
135 class CaseFold final : public TensorTransform {
152 class JiebaTokenizer final : public TensorTransform {
233 class Lookup final : public TensorTransform {
275 class Ngram final : public TensorTransform {
317 class NormalizeUTF8 final : public TensorTransform {
344 class RegexReplace final : public TensorTransform {
375 class RegexTokenizer final : public TensorTransform {
404 class SentencePieceTokenizer final : public TensorTransform {
[all …]
Dexecute.h46 …explicit Execute(std::shared_ptr<TensorTransform> op, MapTargetDevice device_type = MapTargetDevic…
53 …explicit Execute(std::reference_wrapper<TensorTransform> op, MapTargetDevice device_type = MapTarg…
60 …explicit Execute(TensorTransform *op, MapTargetDevice device_type = MapTargetDevice::kCpu, uint32_…
75 explicit Execute(std::vector<std::shared_ptr<TensorTransform>> ops,
83 explicit Execute(const std::vector<std::reference_wrapper<TensorTransform>> ops,
91 …explicit Execute(const std::vector<TensorTransform *> &ops, MapTargetDevice device_type = MapTarge…
129 std::vector<std::shared_ptr<TensorTransform>> transforms_;
Dvision_ascend.h39 class DvppDecodeResizeJpeg final : public TensorTransform {
62 class DvppDecodeResizeCropJpeg final : public TensorTransform {
86 class DvppDecodePng final : public TensorTransform {
/third_party/mindspore/tests/ut/cpp/dataset/
Dc_api_vision_random_subselect_policy_test.cc40 std::shared_ptr<TensorTransform> invert_op(new vision::Invert()); in TEST_F()
41 std::shared_ptr<TensorTransform> equalize_op(new vision::Equalize()); in TEST_F()
42 std::shared_ptr<TensorTransform> resize_op(new vision::Resize({15, 15})); in TEST_F()
50 …std::vector<std::pair<std::shared_ptr<TensorTransform>, double>> policy = {invert_pair, equalize_p… in TEST_F()
51 …std::shared_ptr<TensorTransform> random_select_subpolicy_op(new vision::RandomSelectSubpolicy({pol… in TEST_F()
101 …std::vector<std::pair<TensorTransform *, double>> policy = {invert_pair, equalize_pair, resize_pai… in TEST_F()
158 …std::vector<std::pair<std::reference_wrapper<TensorTransform>, double>> policy = {invert_pair, equ… in TEST_F()
201 std::shared_ptr<TensorTransform> invert_op(new vision::Invert()); in TEST_F()
202 std::shared_ptr<TensorTransform> equalize_op(new vision::Equalize()); in TEST_F()
203 std::shared_ptr<TensorTransform> resize_op(new vision::Resize({15, 15})); in TEST_F()
[all …]
Dexecute_test.cc49 …std::shared_ptr<TensorTransform> allpass_biquad_01 = std::make_shared<audio::AllpassBiquad>(44100,… in TEST_F()
69 …std::shared_ptr<TensorTransform> allpass_biquad_op = std::make_shared<audio::AllpassBiquad>(44100,… in TEST_F()
115 std::shared_ptr<TensorTransform> amplitude_to_db_op = std::make_shared<audio::AmplitudeToDB>(); in TEST_F()
134 std::shared_ptr<TensorTransform> amplitude_to_db_op = in TEST_F()
154 std::shared_ptr<TensorTransform> amplitude_to_db_op = std::make_shared<audio::AmplitudeToDB>(); in TEST_F()
168 std::shared_ptr<TensorTransform> decode = std::make_shared<vision::Decode>(); in TEST_F()
169 std::shared_ptr<TensorTransform> center_crop(new vision::CenterCrop({30})); in TEST_F()
170 std::shared_ptr<TensorTransform> rescale = std::make_shared<vision::Rescale>(1. / 3, 0.5); in TEST_F()
205 …std::shared_ptr<TensorTransform> frequency_masking_op = std::make_shared<audio::FrequencyMasking>(… in TEST_F()
218 std::shared_ptr<TensorTransform> time_masking_op = std::make_shared<audio::TimeMasking>(true, 2); in TEST_F()
[all …]
Dc_api_vision_random_test.cc40 std::shared_ptr<TensorTransform> affine1(new vision::RandomAffine({0.0, 0.0}, {})); in TEST_F()
50 std::shared_ptr<TensorTransform> affine2(new vision::RandomAffine({0.0, 0.0}, {1, 1, 1, 1, 1})); in TEST_F()
60 …std::shared_ptr<TensorTransform> affine3(new vision::RandomAffine({30.0, 30.0}, {0.0, 0.0}, {2.0, … in TEST_F()
83 std::shared_ptr<TensorTransform> affine( in TEST_F()
132 std::shared_ptr<TensorTransform> affine(new vision::RandomAffine({0.0, 0.0})); in TEST_F()
181 …std::shared_ptr<TensorTransform> random_color_op_1 = std::make_shared<vision::RandomColor>(0.0, 0.… in TEST_F()
184 …std::shared_ptr<TensorTransform> random_color_op_2 = std::make_shared<vision::RandomColor>(0.0, 1.… in TEST_F()
233 …std::shared_ptr<TensorTransform> random_color_adjust1(new vision::RandomColorAdjust({1.0}, {0.0}, … in TEST_F()
236 std::shared_ptr<TensorTransform> random_color_adjust2( in TEST_F()
240 std::shared_ptr<TensorTransform> random_color_adjust3( in TEST_F()
[all …]
Dc_api_vision_uniform_aug_test.cc45 std::shared_ptr<TensorTransform> resize_op(new vision::Resize({30, 30})); in TEST_F()
46 std::shared_ptr<TensorTransform> random_crop_op(new vision::RandomCrop({28, 28})); in TEST_F()
47 std::shared_ptr<TensorTransform> center_crop_op(new vision::CenterCrop({16, 16})); in TEST_F()
48 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
187 std::shared_ptr<TensorTransform> random_crop_op(new vision::RandomCrop({28, 28})); in TEST_F()
188 std::shared_ptr<TensorTransform> center_crop_op(new vision::CenterCrop({16, 16})); in TEST_F()
191 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
212 std::shared_ptr<TensorTransform> random_crop_op(new vision::RandomCrop({28, 28})); in TEST_F()
213 std::shared_ptr<TensorTransform> center_crop_op(new vision::CenterCrop({16, 16})); in TEST_F()
216 …std::shared_ptr<TensorTransform> uniform_aug_op(new vision::UniformAugment({random_crop_op, center… in TEST_F()
[all …]
Dc_api_vision_a_to_q_test.cc114 std::shared_ptr<TensorTransform> adjust_gamma(new vision::AdjustGamma(-1, 1.0)); in TEST_F()
137 std::shared_ptr<TensorTransform> auto_contrast(new vision::AutoContrast()); in TEST_F()
186 std::shared_ptr<TensorTransform> auto_contrast(new vision::AutoContrast(10, {10, 20})); in TEST_F()
235 std::shared_ptr<TensorTransform> centre_out1(new vision::CenterCrop({30})); in TEST_F()
281 std::shared_ptr<TensorTransform> crop(new vision::Crop({0, 0}, {height, width})); in TEST_F()
327 std::shared_ptr<TensorTransform> crop1(new vision::Crop({-1, -1}, {20})); in TEST_F()
337 std::shared_ptr<TensorTransform> crop2(new vision::Crop({5}, {10})); in TEST_F()
347 std::shared_ptr<TensorTransform> crop3(new vision::Crop({0, 0}, {-10, -5})); in TEST_F()
357 std::shared_ptr<TensorTransform> crop4(new vision::Crop({0, 0}, {10, 10, 10})); in TEST_F()
377 std::shared_ptr<TensorTransform> hwc_to_chw = std::make_shared<vision::HWC2CHW>(); in TEST_F()
[all …]
Dc_api_vision_bounding_box_augment_test.cc39 std::shared_ptr<TensorTransform> random_rotation_op(new vision::RandomRotation({90.0})); in TEST_F()
40 …std::shared_ptr<TensorTransform> bound_box_augment_op(new vision::BoundingBoxAugment({random_rotat… in TEST_F()
161 std::shared_ptr<TensorTransform> random_rotation_op(new vision::RandomRotation({90.0})); in TEST_F()
164 …std::shared_ptr<TensorTransform> bound_box_augment_op(new vision::BoundingBoxAugment({random_rotat… in TEST_F()
186 std::shared_ptr<TensorTransform> random_rotation_op(new vision::RandomRotation({90.0})); in TEST_F()
189 …std::shared_ptr<TensorTransform> bound_box_augment_op(new vision::BoundingBoxAugment({random_rotat… in TEST_F()
211 …std::shared_ptr<TensorTransform> bound_box_augment_op(new vision::BoundingBoxAugment(nullptr, 0.5)… in TEST_F()
234 std::shared_ptr<TensorTransform> random_rotation_op(new vision::RandomRotation({50.0, -50.0})); in TEST_F()
237 …std::shared_ptr<TensorTransform> bound_box_augment_op(new vision::BoundingBoxAugment({random_rotat… in TEST_F()
Dc_api_vision_r_to_z_test.cc48 std::shared_ptr<TensorTransform> rescale(new mindspore::dataset::vision::Rescale(1.0, 0.0)); in TEST_F()
52 …std::shared_ptr<TensorTransform> type_cast(new transforms::TypeCast(mindspore::DataType::kNumberTy… in TEST_F()
83 std::shared_ptr<TensorTransform> rescale(new mindspore::dataset::vision::Rescale(1.0 / 255, 1.0)); in TEST_F()
125 std::shared_ptr<TensorTransform> resize_op(new vision::Resize({30})); in TEST_F()
169 std::shared_ptr<TensorTransform> resize_with_bbox_op(new vision::ResizeWithBBox({30})); in TEST_F()
170 std::shared_ptr<TensorTransform> resize_with_bbox_op1(new vision::ResizeWithBBox({30, 30})); in TEST_F()
207 std::shared_ptr<TensorTransform> convert(new mindspore::dataset::vision::RGB2GRAY()); in TEST_F()
244 …std::shared_ptr<TensorTransform> rotate1(new vision::Rotate(90.0, InterpolationMode::kNearestNeigh… in TEST_F()
254 std::shared_ptr<TensorTransform> rotate2( in TEST_F()
273 std::shared_ptr<TensorTransform> resize(new vision::Resize({50, 25})); in TEST_F()
[all …]
Dc_api_vision_affine_test.cc42 std::shared_ptr<TensorTransform> crop(new vision::RandomCrop({256, 256})); in TEST_F()
43 std::shared_ptr<TensorTransform> affine( in TEST_F()
85 std::shared_ptr<TensorTransform> crop(new vision::RandomCrop({256, 256})); in TEST_F()
86 std::shared_ptr<TensorTransform> affine( in TEST_F()
Dir_tensor_op_fusion_pass_test.cc42 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
43 std::shared_ptr<TensorTransform> random_resized_crop(new vision::RandomResizedCrop({5})); in TEST_F()
73 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
74 std::shared_ptr<TensorTransform> random_resized_crop(new vision::RandomResizedCrop({5})); in TEST_F()
Dc_api_vision_gaussian_blur_test.cc37 std::shared_ptr<TensorTransform> gaussian_blur1(new vision::GaussianBlur({-1})); in TEST_F()
47 std::shared_ptr<TensorTransform> gaussian_blur2(new vision::GaussianBlur({2, 2}, {3, 3})); in TEST_F()
57 std::shared_ptr<TensorTransform> gaussian_blur3(new vision::GaussianBlur({3}, {-3})); in TEST_F()
75 std::shared_ptr<TensorTransform> gaussian_blur(new vision::GaussianBlur({3, 3}, {5, 5})); in TEST_F()
Dc_api_text_test.cc52 std::shared_ptr<TensorTransform> basic_tokenizer = std::make_shared<text::BasicTokenizer>(); in TEST_F()
113 std::shared_ptr<TensorTransform> basic_tokenizer = std::make_shared<text::BasicTokenizer>(true); in TEST_F()
163 std::shared_ptr<TensorTransform> basic_tokenizer = in TEST_F()
246 std::shared_ptr<TensorTransform> bert_tokenizer = std::make_shared<text::BertTokenizer>(vocab); in TEST_F()
309 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
370 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
432 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
492 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
552 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
612 std::shared_ptr<TensorTransform> bert_tokenizer = in TEST_F()
[all …]
Drandom_posterize_op_test.cc50 std::shared_ptr<TensorTransform> decode_op = std::make_shared<vision::Decode>(); in TEST_F()
51 std::shared_ptr<TensorTransform> randomposterize_op(new vision::RandomPosterize({3, 5})); in TEST_F()
66 std::shared_ptr<TensorTransform> randomsolarize_op(new vision::RandomSolarize({12, 25})); in TEST_F()
Dc_api_text_sentence_piece_vocab_test.cc56 std::shared_ptr<TensorTransform> sentencepiece_tokenizer = in TEST_F()
119 std::shared_ptr<TensorTransform> sentencepiece_tokenizer = in TEST_F()
198 std::shared_ptr<TensorTransform> sentencepiece_tokenizer = in TEST_F()
222 std::shared_ptr<TensorTransform> sentencepiece_tokenizer = in TEST_F()
246 std::shared_ptr<TensorTransform> sentencepiece_tokenizer = in TEST_F()
270 std::shared_ptr<TensorTransform> sentencepiece_tokenizer4 = in TEST_F()
Dc_api_vision_soft_dvpp_test.cc39 std::shared_ptr<TensorTransform> soft_dvpp_decode_random_crop_resize_jpeg( in TEST_F()
81 std::shared_ptr<TensorTransform> soft_dvpp_decode_random_crop_resize_jpeg( in TEST_F()
126 …std::shared_ptr<TensorTransform> soft_dvpp_decode_resize_jpeg_op(new vision::SoftDvppDecodeResizeJ… in TEST_F()
164 …std::shared_ptr<TensorTransform> soft_dvpp_decode_resize_jpeg_op(new vision::SoftDvppDecodeResizeJ… in TEST_F()
Dc_api_vision_slice_patches_test.cc38 std::shared_ptr<TensorTransform> slice_patches_1(new vision::SlicePatches(-1)); in TEST_F()
48 std::shared_ptr<TensorTransform> slice_patches_2(new vision::SlicePatches(1, 0)); in TEST_F()
67 std::shared_ptr<TensorTransform> slice_patches(new vision::SlicePatches(2, 2)); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dtransforms.cc40 Compose::Compose(const std::vector<TensorTransform *> &transforms) : data_(std::make_shared<Data>()… in Compose()
42 [](TensorTransform *const op) -> std::shared_ptr<TensorOperation> { in Compose()
47 Compose::Compose(const std::vector<std::shared_ptr<TensorTransform>> &transforms) : data_(std::make… in Compose()
49 [](std::shared_ptr<TensorTransform> op) -> std::shared_ptr<TensorOperation> { in Compose()
54 Compose::Compose(const std::vector<std::reference_wrapper<TensorTransform>> &transforms) in Compose()
57 … [](TensorTransform &op) -> std::shared_ptr<TensorOperation> { return op.Parse(); }); in Compose()
193 RandomApply::RandomApply(const std::vector<TensorTransform *> &transforms, double prob) in RandomApply()
196 [](TensorTransform *const op) -> std::shared_ptr<TensorOperation> { in RandomApply()
202 RandomApply::RandomApply(const std::vector<std::shared_ptr<TensorTransform>> &transforms, double pr… in RandomApply()
205 [](std::shared_ptr<TensorTransform> op) -> std::shared_ptr<TensorOperation> { in RandomApply()
[all …]
/third_party/mindspore/tests/st/cpp/dataset/
Dtest_de.cc72 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
73 std::shared_ptr<TensorTransform> resize(new vision::Resize({224, 224})); in TEST_F()
74 std::shared_ptr<TensorTransform> normalize( in TEST_F()
76 std::shared_ptr<TensorTransform> hwc2chw(new vision::HWC2CHW()); in TEST_F()
113 …std::shared_ptr<TensorTransform> decode_resize_crop(new vision::DvppDecodeResizeCropJpeg(crop_para… in TEST_F()
167 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
168 std::shared_ptr<TensorTransform> resize(new vision::Resize(resize_paras)); in TEST_F()
169 std::shared_ptr<TensorTransform> centercrop(new vision::CenterCrop(crop_paras)); in TEST_F()
170 std::vector<std::shared_ptr<TensorTransform>> trans_list = {decode, resize, centercrop}; in TEST_F()
217 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
[all …]
/third_party/mindspore/tests/st/cpp/model/
Dtest_zero_copy.cc75 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
76 std::shared_ptr<TensorTransform> resize(new vision::Resize({256})); in TEST_F()
77 std::shared_ptr<TensorTransform> center_crop(new vision::CenterCrop({224, 224})); in TEST_F()
123 std::shared_ptr<TensorTransform> decode(new vision::Decode()); in TEST_F()
124 std::shared_ptr<TensorTransform> resize(new vision::Resize({256})); in TEST_F()
125 std::shared_ptr<TensorTransform> center_crop(new vision::CenterCrop({224, 224})); in TEST_F()

12