Home
last modified time | relevance | path

Searched defs:SelectInferShape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dselect_infer.c22 int SelectInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **outputs, size_t ou… in SelectInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dselection_ops_proto.cc153 IMPLEMT_COMMON_INFERFUNC(SelectInferShape) { in IMPLEMT_COMMON_INFERFUNC() argument