Searched refs:MixUpBatchOperation (Results 1 – 7 of 7) sorted by relevance
31 MixUpBatchOperation::MixUpBatchOperation(float alpha) : alpha_(alpha) {} in MixUpBatchOperation() function in mindspore::dataset::vision::MixUpBatchOperation33 MixUpBatchOperation::~MixUpBatchOperation() = default;35 std::string MixUpBatchOperation::Name() const { return kMixUpBatchOperation; } in Name()37 Status MixUpBatchOperation::ValidateParams() { in ValidateParams()42 std::shared_ptr<TensorOp> MixUpBatchOperation::Build() { return std::make_shared<MixUpBatchOp>(alph… in Build()44 Status MixUpBatchOperation::to_json(nlohmann::json *out_json) { in to_json()49 Status MixUpBatchOperation::from_json(nlohmann::json op_params, std::shared_ptr<TensorOperation> *o… in from_json()52 *operation = std::make_shared<vision::MixUpBatchOperation>(alpha); in from_json()
38 class MixUpBatchOperation : public TensorOperation {40 explicit MixUpBatchOperation(float alpha);42 ~MixUpBatchOperation();
232 MixUpBatchOperation, 1, ([](const py::module *m) { in __anon682838b21e02()233 …(void)py::class_<vision::MixUpBatchOperation, TensorOperation, std::shared_ptr<vision::MixUpBatchO… in __anon682838b21e02()236 auto mix_up_batch = std::make_shared<vision::MixUpBatchOperation>(alpha); in __anon682838b21e02()
283 ops_ptr[vision::kMixUpBatchOperation] = &(vision::MixUpBatchOperation::from_json); in InitializeFuncPtr()
564 return cde.MixUpBatchOperation(self.alpha)
117 std::shared_ptr<TensorOperation> operation7 = std::make_shared<vision::MixUpBatchOperation>(1.0); in TEST_F()
389 std::shared_ptr<TensorOperation> MixUpBatch::Parse() { return std::make_shared<MixUpBatchOperation>… in Parse()