Home
last modified time | relevance | path

Searched refs:ratio_ (Results 1 – 22 of 22) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dsoftdvpp_decode_random_crop_resize_jpeg_ir.cc34 : size_(size), scale_(scale), ratio_(ratio), max_attempts_(max_attempts) {} in SoftDvppDecodeRandomCropResizeJpegOperation()
58 RETURN_IF_NOT_OK(ValidateVectorRatio("SoftDvppDecodeRandomCropResizeJpeg", ratio_)); in ValidateParams()
82 … scale_[dimension_one], ratio_[dimension_zero], in Build()
83ratio_[dimension_one], max_attempts_); in Build()
91 args["ratio"] = ratio_; in to_json()
Drandom_resized_crop_with_bbox_ir.cc36 …: size_(size), scale_(scale), ratio_(ratio), interpolation_(interpolation), max_attempts_(max_atte… in RandomResizedCropWithBBoxOperation()
48 RETURN_IF_NOT_OK(ValidateVectorRatio("RandomResizedCropWithBBox", ratio_)); in ValidateParams()
79 …height, width, scale_[dimension_zero], scale_[dimension_one], ratio_[dimension_zero], ratio_[dimen… in Build()
88 args["ratio"] = ratio_; in to_json()
Dbounding_box_augment_ir.cc32 : transform_(transform), ratio_(ratio) {} in BoundingBoxAugmentOperation()
40 RETURN_IF_NOT_OK(ValidateScalar("BoundingBoxAugment", "ratio", ratio_, {0.0, 1.0}, false, false)); in ValidateParams()
45 …ndingBoxAugmentOp> tensor_op = std::make_shared<BoundingBoxAugmentOp>(transform_->Build(), ratio_); in Build()
56 args["ratio"] = ratio_; in to_json()
Drandom_resized_crop_ir.cc39 ratio_(ratio), in RandomResizedCropOperation()
53 RETURN_IF_NOT_OK(ValidateVectorRatio(Name(), ratio_)); in ValidateParams()
84 …height, width, scale_[dimension_zero], scale_[dimension_one], ratio_[dimension_zero], ratio_[dimen… in Build()
93 args["ratio"] = ratio_; in to_json()
Drandom_crop_decode_resize_ir.cc57 float aspect_lower_bound = ratio_[dimension_zero]; in Build()
58 float aspect_upper_bound = ratio_[dimension_one]; in Build()
73 args["ratio"] = ratio_; in to_json()
Dbounding_box_augment_ir.h56 float ratio_; variable
Dsoftdvpp_decode_random_crop_resize_jpeg_ir.h59 std::vector<float> ratio_; variable
Drandom_resized_crop_with_bbox_ir.h60 std::vector<float> ratio_; variable
Drandom_resized_crop_ir.h61 std::vector<float> ratio_; variable
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Ddropout.cc40 if ((param->ratio_ > 1.0f) || (param->ratio_ < 0.0f)) { in Init()
45 if (param->ratio_ >= 1.0f) { in Init()
48 scale_ = 1. / (1. - param->ratio_); in Init()
80 std::bernoulli_distribution distribution(param->ratio_); in Execute()
Ddropout_grad.cc46 if ((param->ratio_ > 1.0f) || (param->ratio_ < 0.0f)) { in Init()
51 if (param->ratio_ >= 1.0f) { in Init()
54 scale_ = 1. / (1. - param->ratio_); in Init()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dvision.cc153 float ratio_; member
158 data_->ratio_ = ratio; in BoundingBoxAugment()
164 data_->ratio_ = ratio; in BoundingBoxAugment()
170 data_->ratio_ = ratio; in BoundingBoxAugment()
174 return std::make_shared<BoundingBoxAugmentOperation>(data_->transform_, data_->ratio_); in Parse()
544 …: size_(size), scale_(scale), ratio_(ratio), interpolation_(interpolation), max_attempts_(max_atte… in Data()
547 std::vector<float> ratio_; member
558 …eturn std::make_shared<RandomCropDecodeResizeOperation>(data_->size_, data_->scale_, data_->ratio_, in Parse()
649 …: size_(size), scale_(scale), ratio_(ratio), interpolation_(interpolation), max_attempts_(max_atte… in Data()
652 std::vector<float> ratio_; member
[all …]
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc412 dropout_parameter->ratio_ = value->keep_prob(); in PopulateDropoutParameter()
413 if (dropout_parameter->ratio_ < 0.f || dropout_parameter->ratio_ > 1.f) { in PopulateDropoutParameter()
414 MS_LOG(ERROR) << "Dropout ratio must be between 0 to 1, got " << dropout_parameter->ratio_; in PopulateDropoutParameter()
432 dropoutgrad_parameter->ratio_ = value->keep_prob(); in PopulateDropoutGradParameter()
433 if (dropoutgrad_parameter->ratio_ < 0.f || dropoutgrad_parameter->ratio_ > 1.f) { in PopulateDropoutGradParameter()
434 …S_LOG(ERROR) << "Dropout Grad ratio must be between 0 to 1, got " << dropoutgrad_parameter->ratio_; in PopulateDropoutGradParameter()
Dtrain_populate_parameter_v0.cc563 dropout_parameter->ratio_ = dropout_prim->ratio(); in PopulateDropoutParameter()
564 if (dropout_parameter->ratio_ < 0.f || dropout_parameter->ratio_ > 1.f) { in PopulateDropoutParameter()
565 MS_LOG(ERROR) << "Dropout ratio must be between 0 to 1, got " << dropout_parameter->ratio_; in PopulateDropoutParameter()
583 dropoutGrad_parameter->ratio_ = dropoutGrad_prim->ratio(); in PopulateDropoutGradParameter()
584 if (dropoutGrad_parameter->ratio_ < 0.f || dropoutGrad_parameter->ratio_ > 1.f) { in PopulateDropoutGradParameter()
585 …S_LOG(ERROR) << "Dropout Grad ratio must be between 0 to 1, got " << dropoutGrad_parameter->ratio_; in PopulateDropoutGradParameter()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dbounding_box_augment_op.cc28 : ratio_(ratio), uniform_(0, 1), transform_(std::move(transform)) { in BoundingBoxAugmentOp()
41 if (uniform_(rnd_) < ratio_) { in Compute()
Dbounding_box_augment_op.h55 float ratio_;
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Ddropout_parameter.h24 float ratio_; member
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/
Dsplit_with_over_lap_base.cc36 total_block_count += param_->ratio_[i]; in CalculateSplitedShapes()
45 visited_block += param_->ratio_[i]; in CalculateSplitedShapes()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dsplit_parameter.h52 int ratio_[SPLIT_MAX_SLICE_NUM]; member
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dsplit_with_overlap_populate.cc65 param->ratio_[i] = (*ratio)[i]; in PopulateSplitWithOverlapParameter()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Ddropout_fp16_grad.cc40 auto ratio = param->ratio_; in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dsplit_with_over_lap_infer.c44 ratio[i] = param->ratio_[i]; in SplitWithOverlapInferShape()