Home
last modified time | relevance | path

Searched refs:NumInput (Results 1 – 24 of 24) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/data/
Drandom_choice_op.cc25 uint32_t RandomChoiceOp::NumInput() { in NumInput() function in mindspore::dataset::RandomChoiceOp
26 uint32_t num_input = ops_.front()->NumInput(); in NumInput()
28 uint32_t cur_num = op->NumInput(); in NumInput()
Drandom_apply_op.h45 uint32_t NumInput() override { return compose_->NumInput(); } in NumInput() function
Dcompose_op.h41 uint32_t NumInput() override { return ops_.front()->NumInput(); } in NumInput() function
Dconcatenate_op.h55 uint32_t NumInput() override { return 0; } in NumInput() function
Drandom_choice_op.h44 uint32_t NumInput() override;
Drandom_apply_op.cc26 if (compose_->NumOutput() != NumInput()) { in NumOutput()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Drandom_select_subpolicy_op.cc42 uint32_t RandomSelectSubpolicyOp::NumInput() { in NumInput() function in mindspore::dataset::RandomSelectSubpolicyOp
43 uint32_t num_in = policy_.front().front().first->NumInput(); in NumInput()
46 if (num_in != p.first->NumInput()) { in NumInput()
Drotate_op.cc84 if (inputs.size() != NumInput()) in OutputShape()
87 ", does not match the number of inputs: " + std::to_string(NumInput())); in OutputShape()
Drandom_vertical_flip_op.h46 uint32_t NumInput() override { return -1; } in NumInput() function
Drandom_horizontal_flip_op.h52 uint32_t NumInput() override { return -1; } in NumInput() function
Drandom_resize_op.h50 uint32_t NumInput() override { return -1; } in NumInput() function
Dresize_with_bbox_op.h52 uint32_t NumInput() override { return 2; } in NumInput() function
Drandom_select_subpolicy_op.h45 uint32_t NumInput() override;
Drandom_crop_and_resize_op.h65 uint32_t NumInput() override { return -1; } in NumInput() function
Drandom_crop_op.h82 uint32_t NumInput() override { return -1; } in NumInput() function
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/
Dtensor_op.cc58 if (inputs.size() != NumInput()) in OutputShape()
66 if (inputs.size() != NumInput()) in OutputType()
Dc_func_op.h36 uint32_t NumInput() override { return 0; } in NumInput() function
Dtensor_op.h225 bool OneToOne() { return NumInput() == 1 && NumOutput() == 1; } in OneToOne()
233 virtual uint32_t NumInput() { return 1; } in NumInput() function
Dpy_func_op.h38 uint32_t NumInput() override { return 0; } in NumInput() function
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/text/kernels/
Dtruncate_sequence_pair_op.h42 uint32_t NumInput() override { return -1; } in NumInput() function
Dsliding_window_op.cc35 …CHECK_FAIL_RETURN_UNEXPECTED(inputs.size() == NumInput(), "SlidingWindow: incorrect number of inpu… in OutputShape()
Dlookup_op.cc54 …CHECK_FAIL_RETURN_UNEXPECTED(inputs.size() == NumInput() && outputs.size() == NumOutput(), "size d… in OutputType()
Dngram_op.cc82 CHECK_FAIL_RETURN_UNEXPECTED(inputs.size() == NumInput(), "Ngram: incorrect num of inputs\n"); in OutputShape()
/third_party/mindspore/tests/ut/cpp/dataset/
Dmap_op_test.cc60 uint32_t NumInput() override { return 3; } in NumInput() function in mindspore::dataset::test::ThreeToOneOp