Home
last modified time | relevance | path

Searched refs:NumOutput (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Drandom_apply_op.cc25 uint32_t RandomApplyOp::NumOutput() { in NumOutput() function in mindspore::dataset::RandomApplyOp
26 if (compose_->NumOutput() != NumInput()) { in NumOutput()
30 return compose_->NumOutput(); in NumOutput()
38 outputs.resize(NumOutput(), TensorShape::CreateUnknownRankShape()); in OutputShape()
46 outputs.resize(NumOutput(), DataType(DataType::DE_UNKNOWN)); in OutputType()
Drandom_choice_op.cc37 uint32_t RandomChoiceOp::NumOutput() { in NumOutput() function in mindspore::dataset::RandomChoiceOp
38 uint32_t num_output = ops_.front()->NumOutput(); in NumOutput()
40 uint32_t cur_num = op->NumOutput(); in NumOutput()
57 outputs.resize(NumOutput(), TensorShape::CreateUnknownRankShape()); in OutputShape()
72 outputs.resize(NumOutput(), DataType(DataType::DE_UNKNOWN)); in OutputType()
Dcompose_op.h45 uint32_t NumOutput() override { return ops_.back()->NumOutput(); } in NumOutput() function
Dunique_op.h39 uint32_t NumOutput() override { return 0; } in NumOutput() function
Dduplicate_op.h37 uint32_t NumOutput() override { return 2; } in NumOutput() function
Drandom_choice_op.h48 uint32_t NumOutput() override;
Drandom_apply_op.h49 uint32_t NumOutput() override;
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Drandom_select_subpolicy_op.cc55 uint32_t RandomSelectSubpolicyOp::NumOutput() { in NumOutput() function in mindspore::dataset::RandomSelectSubpolicyOp
56 uint32_t num_out = policy_.front().front().first->NumOutput(); in NumOutput()
59 if (num_out != p.first->NumOutput()) { in NumOutput()
70 outputs.resize(NumOutput(), TensorShape::CreateUnknownRankShape()); in OutputShape()
82 outputs.resize(NumOutput(), DataType(DataType::DE_UNKNOWN)); in OutputType()
Drandom_vertical_flip_op.h48 uint32_t NumOutput() override { return -1; } in NumOutput() function
Drandom_horizontal_flip_op.h54 uint32_t NumOutput() override { return -1; } in NumOutput() function
Drandom_resize_op.h52 uint32_t NumOutput() override { return -1; } in NumOutput() function
Dresize_with_bbox_op.h54 uint32_t NumOutput() override { return 2; } in NumOutput() function
Drandom_select_subpolicy_op.h49 uint32_t NumOutput() override;
Drandom_crop_and_resize_op.h67 uint32_t NumOutput() override { return -1; } in NumOutput() function
Drandom_crop_op.h84 uint32_t NumOutput() override { return -1; } in NumOutput() function
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dtruncate_sequence_pair_op.h45 uint32_t NumOutput() override { return -1; } in NumOutput() function
Dsliding_window_op.cc54 …CHECK_FAIL_RETURN_UNEXPECTED(outputs.size() == NumOutput(), "SlidingWindow: incorrect number of ou… in OutputShape()
Dlookup_op.cc54 …CHECK_FAIL_RETURN_UNEXPECTED(inputs.size() == NumInput() && outputs.size() == NumOutput(), "size d… in OutputType()
Dngram_op.cc92 CHECK_FAIL_RETURN_UNEXPECTED(outputs.size() == NumOutput(), "Ngram: incorrect num of outputs\n"); in OutputShape()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/
Dc_func_op.h37 uint32_t NumOutput() override { return 0; } in NumOutput() function
Dtensor_op.h225 bool OneToOne() { return NumInput() == 1 && NumOutput() == 1; } in OneToOne()
237 virtual uint32_t NumOutput() { return 1; } in NumOutput() function
Dpy_func_op.h39 uint32_t NumOutput() override { return 0; } in NumOutput() function
/third_party/mindspore/tests/ut/cpp/dataset/
Dmap_op_test.cc78 uint32_t NumOutput() override { return 3; } in NumOutput() function in mindspore::dataset::test::OneToThreeOp