Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Dunsorted_segment_sum_infer.c20 int UnsortedSegmentSumInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **output… in UnsortedSegmentSumInferShape() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dselection_ops_proto.cc354 IMPLEMT_COMMON_INFERFUNC(UnsortedSegmentSumInferShape) { in IMPLEMT_COMMON_INFERFUNC() argument