Searched refs:kNCHW (Results 1 – 7 of 7) sorted by relevance
75 …std::shared_ptr<CutMixBatchOp> op = std::make_shared<CutMixBatchOp>(ImageBatchFormat::kNCHW, 1.0, … in TEST_F()115 …std::shared_ptr<CutMixBatchOp> op = std::make_shared<CutMixBatchOp>(ImageBatchFormat::kNCHW, 1.0, … in TEST_F()
398 std::make_shared<vision::CutMixBatch>(mindspore::dataset::ImageBatchFormat::kNCHW, 1.0, 1.0); in TEST_F()
1312 std::make_shared<vision::CutMixBatch>(mindspore::dataset::ImageBatchFormat::kNCHW, 1.0, 1.0); in TEST_F()
41 …image_batch_format_ != ImageBatchFormat::kNHWC && image_batch_format_ != ImageBatchFormat::kNCHW) { in ValidateParams()
112 kNCHW = 1 ///< Indicate the input batch is of NCHW format. enumerator
130 .value("DE_IMAGE_BATCH_FORMAT_NCHW", ImageBatchFormat::kNCHW) in __anon39cb0f2d0e02()
94 image_batch_format_ == ImageBatchFormat::kNCHW) { in ValidateCutMixBatch()