Searched refs:SlicePatchesOp (Results 1 – 3 of 3) sorted by relevance
22 const int32_t SlicePatchesOp::kDefNumH = 1;23 const int32_t SlicePatchesOp::kDefNumW = 1;24 const uint8_t SlicePatchesOp::kDefFillV = 0;25 const SliceMode SlicePatchesOp::kDefSliceMode = SliceMode::kPad;27 SlicePatchesOp::SlicePatchesOp(int32_t num_height, int32_t num_width, SliceMode slice_mode, uint8_t… in SlicePatchesOp() function in mindspore::dataset::SlicePatchesOp30 Status SlicePatchesOp::Compute(const TensorRow &input, TensorRow *output) { in Compute()
30 class SlicePatchesOp : public TensorOp {38 …SlicePatchesOp(int32_t num_height = kDefNumH, int32_t num_width = kDefNumW, SliceMode slice_mode =…41 ~SlicePatchesOp() override = default;
45 …auto tensor_op = std::make_shared<SlicePatchesOp>(num_height_, num_width_, slice_mode_, fill_value… in Build()